jdZddlmZddlZddlmZddlmZddlmZm Z ddl m Z m Z ddl mZd d ZeGd d Zy) u``TranslatedSpan`` — the mutable draft handlers rewrite before export. Split out of :mod:`langsmith._internal.voice.base_span_processor` so the base processor stays focused on span lifecycle and export wiring. The framework subclasses (Pipecat, LiveKit) rewrite one of these per span while translating ``lk.*`` / ``pipecat.*`` data into LangSmith's ``gen_ai.*`` / ``langsmith.*`` namespaces. ) annotationsN)copy) dataclass)AnyOptional)Event ReadableSpan)rebuild_readable_spanc |siS|jDcic](\}}t|ttfs|t|*c}}Scc}}w)zFKeep only the int-valued detail keys, dropping ``None`` / non-numeric.)items isinstanceintfloat)detailskvs s/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langsmith/_internal/voice/translated_span.py_clean_token_detailsrsA  ")--/ Q/$!QZC<5PAs1vI/ QQ Qs A A ceZdZUdZded<ded<ded<eddZdd Zdd Zdd Z dd Z dd Z d dZ d!dZ d"dZd#dZd$dZdddddd d%dZddd d&dZd'dZd(dZy))TranslatedSpanuA span being translated into LangSmith's namespaces before export. Wraps a shallow copy of the original read-only OTel ``ReadableSpan`` with mutable ``attributes`` and ``events`` seeded from it. Handlers rewrite the draft while translating; :meth:`finalize` builds a fresh ``ReadableSpan`` from it — so the original span is never mutated. Created per span in :meth:`BaseLangSmithSpanProcessor.on_end` and threaded through dispatch — no global state. A processor that defers a span (see :meth:`BaseLangSmithSpanProcessor._dispatch`) simply holds the ``TranslatedSpan`` itself until it exports it later, so the in-progress translation outlives the originating ``on_end`` call. r spanzdict[str, Any] attributesz list[Event]eventsc|t|t|jxsit|jxsgS)z>Seed a draft from a span's own fields, attributes, and events.)rdictrlistr)clsrs rofzTranslatedSpan.of1s: J &B ' " #  cZt|j|j|jS)zDBuild the export span from the draft and its rewritten attrs/events.)rr)r rrrselfs rfinalizezTranslatedSpan.finalize:s"$ II$//$++  rc&||j_y)z6Set the exported span's name (the LangSmith run name).N)r_name)r"names rset_namezTranslatedSpan.set_name@s  rc&||j_y)uSet the exported span's end time (epoch ns). Lets a span merged from two sources report a duration that runs past its own end — e.g. a tool span spanning from the call to the result. N)r _end_time)r" end_time_nss r set_end_timezTranslatedSpan.set_end_timeDs * rc"||jd<y)uCSet ``langsmith.span.kind`` (``llm`` / ``chain`` / ``tool`` / …).zlangsmith.span.kindNr)r"kinds rset_kindzTranslatedSpan.set_kindLs15-.rc"||jd<y)zBSet ``langsmith.metadata.thread_id`` (the conversation/thread id).zlangsmith.metadata.thread_idNr-)r" thread_ids r set_thread_idzTranslatedSpan.set_thread_idPs:C67rcF|r||jd<||jd<yy)uSSet ``provider`` on both gen_ai.* provider keys (no-op if empty). LangSmith maps either ``gen_ai.provider.name`` (newer) or the legacy ``gen_ai.system`` to ``ls_provider``; writing both is version-agnostic. The caller passes an already-resolved slug — normalization is the integration's concern. zgen_ai.provider.namez gen_ai.systemNr-)r"providers r set_providerzTranslatedSpan.set_providerTs) 6>DOO2 3/7DOOO , rcj|r1t||jd<t||jd<yy)zJSet the request model on ``gen_ai.request.model`` and its metadata mirror.zgen_ai.request.modelzlangsmith.metadata.model_nameN)strr)r"models r set_modelzTranslatedSpan.set_model`s0 69%jDOO2 3?B5zDOO; < rc"d|jd<y)u@Drop this span from the conversation Messages view (still in the tree). That view reconstructs the chat from ``llm``/``tool`` runs. STT/TTS spans are tagged ``llm``-kind for the tree but would otherwise add fake turns (raw transcripts, "Generated audio for: …"), so they opt out here. Tz*langsmith.metadata.ls_message_view_excludeNr-r!s rexclude_from_message_viewz(TranslatedSpan.exclude_from_message_viewfsIMDErc"||jd<y)z:Mark the span as the trace root (``langsmith.root_span``).zlangsmith.root_spanNr-)r"is_roots r set_root_spanzTranslatedSpan.set_root_spanos18-.rc(||jd|<y)u[Set ``langsmith.metadata.`` to the given value. LangSmith surfaces everything under ``langsmith.metadata.*`` as run metadata. Note ``langsmith.root_span`` and ``langsmith.span.kind`` are NOT metadata — they live in the top-level ``langsmith.*`` namespace and have their own setters / direct writes. zlangsmith.metadata.Nr-)r"keyvalues r set_metadatazTranslatedSpan.set_metadatass8=-cU34rN) input_tokens output_tokens total_tokensinput_token_detailsoutput_token_detailsci}|t||d<|t||d<|t||d<t|x}r||d<t|x}r||d<|r#tj||jd<yy)uWrite the given token usage as ``langsmith.usage_metadata`` (JSON). Callers pass the complete usage — it replaces the flat ``gen_ai.usage.*`` at ingest. No-op when nothing is given. NrCrDrErFrGzlangsmith.usage_metadata)rrjsondumpsr)r"rCrDrErFrGusagers r set_usagezTranslatedSpan.set_usage}s!#  #$' $5E. !  $%(%7E/ "  #$' $5E. !*+>? ?7 ?+2E' (*+?@ @7 @,3E( ) :>**U:KDOO6 7 r)prompt completionc|$tjd|i|jd<|%tjd|i|jd<yy)zIWrite ``gen_ai.prompt``/``gen_ai.completion`` as ``{"messages": [...]}``.Nmessages gen_ai.promptgen_ai.completion)rIrJr)r"rMrNs r set_messageszTranslatedSpan.set_messagessN  /3zz:v:N/ODOOO ,  !37::z:>V3WDOO/ 0 "rclt|tr|ntj||jd<y)zWrite a tool run's input to ``gen_ai.prompt`` as raw I/O (not messages). A ``str`` is passed through unchanged; any other value is JSON-encoded. rQNr r7rIrJr)r" tool_inputs rset_tool_inputzTranslatedSpan.set_tool_inputs) %Z5J4::j;Q (rclt|tr|ntj||jd<y)zWrite a tool run's output to ``gen_ai.completion`` as raw I/O (not messages). A ``str`` is passed through unchanged; any other value is JSON-encoded. rRNrU)r" tool_outputs rset_tool_outputzTranslatedSpan.set_tool_outputs* &k37KTZZ =T +,r)rr returnr)r[r )r&r7r[None)r*rr[r\)r.r7r[r\)r1r7r[r\)r4 Optional[str]r[r\)r8r]r[r\)r[r\)r=boolr[r\)r@r7rArr[r\) rC Optional[int]rDr_rEr_rFOptional[dict[str, int]]rGr`r[r\)rMOptional[list[dict]]rNrar[r\)rVrr[r\)rYrr[r\)__name__ __module__ __qualname____doc____annotations__ classmethodrr#r'r+r/r2r5r9r;r>rBrLrSrWrZrrrrs      *6D 8J M9='+'+&*8<9=L$L% L $ L 6 L7L L>(,+/ X% X) X  X  rr)rzOptional[dict[str, Any]]r[zdict[str, int])re __future__rrIr dataclassesrtypingrropentelemetry.sdk.tracerr $langsmith._internal.otel._span_utilsr rrrhrrrnsC# ! 7FR W W  W r