Function make_llm_span

tests/adapter/test_triplet_trace_tree.py:117–143

(
    span_id: str,
    *,
    parent_id: str,
    start: float,
    end: float,
    prompt_ids: Optional[List[int]] = None,
    response_ids: Optional[List[int]] = None,
    response_id: Optional[str] = None,
    extra_attrs: Optional[Dict[str, Any]] = None,
)

def make_llm_span(
    span_id: str,
    *,
    parent_id: str,
    start: float,
    end: float,
    prompt_ids: Optional[List[int]] = None,
    response_ids: Optional[List[int]] = None,
    response_id: Optional[str] = None,
    extra_attrs: Optional[Dict[str, Any]] = None,
) -> Span:
    attrs: Dict[str, Any] = {
        "prompt_token_ids": prompt_ids or [],
        "response_token_ids": response_ids or [],
    }
    if response_id is not None:
        attrs["gen_ai.response.id"] = response_id
    if extra_attrs:
        attrs.update(extra_attrs)
    return make_span(
        span_id,
        "openai.chat.completion",
        parent_id=parent_id,
        start_time=start,
        end_time=end,
        attributes=attrs,
    )


def reward_attributes(value: float) -> Dict[str, Any]: