~tjdZddlmZddlZddlmZddlmZddlmZm Z ddl m Z m Z ddl mZdd ZeGddZdS)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_spandetailsOptional[dict[str, Any]]returndict[str, int]cF|siSd|DS)zFKeep only the int-valued detail keys, dropping ``None`` / non-numeric.cli|]1\}}t|ttf!|t|2S) isinstanceintfloat).0kvs g/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/_internal/voice/translated_span.py z(_clean_token_details..s7 Q Q Q$!QZC<5P5P QAs1vv Q Q Q)items)r s r_clean_token_detailsrs+  Q Q'--// Q Q QQrceZdZUdZded<ded<ded<ed6d Zd7d Zd8dZd9dZ d:dZ d;dZ ddZd?dZd@d"Zd#d#d#d#d#d$dAd,Zd#d#d-dBd1ZdCd3ZdDd5Zd#S)ETranslatedSpanuA 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]eventsr c|t|t|jpit|jpgS)z>Seed a draft from a span's own fields, attributes, and events.)rdictr listr!)clsrs rofzTranslatedSpan.of1sEs JJ &B ' ' " # #   rcDt|j|j|jS)zDBuild the export span from the draft and its rewritten attrs/events.)r r!)r rr r!selfs rfinalizezTranslatedSpan.finalize:s&$ I$/$+    rnamestrNonec||j_dS)z6Set the exported span's name (the LangSmith run name).N)r_name)r)r+s rset_namezTranslatedSpan.set_name@s r end_time_nsrc||j_dS)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)r1s r set_end_timezTranslatedSpan.set_end_timeDs * rkindc||jd<dS)uCSet ``langsmith.span.kind`` (``llm`` / ``chain`` / ``tool`` / …).zlangsmith.span.kindNr )r)r5s rset_kindzTranslatedSpan.set_kindLs15-...r thread_idc||jd<dS)zBSet ``langsmith.metadata.thread_id`` (the conversation/thread id).zlangsmith.metadata.thread_idNr7)r)r9s r set_thread_idzTranslatedSpan.set_thread_idPs:C6777rprovider Optional[str]c6|r||jd<||jd<dSdS)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.systemNr7)r)r<s r set_providerzTranslatedSpan.set_providerTs5  86>DO2 3/7DOO , , , 8 8rmodelcj|r0t||jd<t||jd<dSdS)zJSet the request model on ``gen_ai.request.model`` and its metadata mirror.zgen_ai.request.modelzlangsmith.metadata.model_nameN)r,r )r)r@s r set_modelzTranslatedSpan.set_model`sC  J69%jjDO2 3?B5zzDO; < < < J Jrcd|jd<dS)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_excludeNr7r(s rexclude_from_message_viewz(TranslatedSpan.exclude_from_message_viewfsIMDEEEris_rootboolc||jd<dS)z:Mark the span as the trace root (``langsmith.root_span``).zlangsmith.root_spanNr7)r)rEs r set_root_spanzTranslatedSpan.set_root_spanos18-...rkeyvaluerc ||jd|<dS)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.Nr7)r)rIrJs r set_metadatazTranslatedSpan.set_metadatass8=3c33444rN) input_tokens output_tokens total_tokensinput_token_detailsoutput_token_detailsrM Optional[int]rNrOrPOptional[dict[str, int]]rQci}|t||d<|t||d<|t||d<t|x}r||d<t|x}r||d<|rtj||jd<dSdS)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. NrMrNrOrPrQzlangsmith.usage_metadata)rrjsondumpsr )r)rMrNrOrPrQusager s r set_usagezTranslatedSpan.set_usage}s!#  #$' $5$5E. !  $%(%7%7E/ "  #$' $5$5E. !*+>?? ?7 3+2E' (*+?@@ @7 4,3E( )  L:>*U:K:KDO6 7 7 7 L Lr)prompt completionrYOptional[list[dict]]rZc|tjd|i|jd<| tjd|i|jd<dSdS)zIWrite ``gen_ai.prompt``/``gen_ai.completion`` as ``{"messages": [...]}``.Nmessages gen_ai.promptgen_ai.completion)rUrVr )r)rYrZs r set_messageszTranslatedSpan.set_messagessU  /3z:v:N/O/ODOO ,  !37:z:>V3W3WDO/ 0 0 0 " !r tool_inputclt|tr|ntj||jd<dS)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. r^Nrr,rUrVr )r)ras rset_tool_inputzTranslatedSpan.set_tool_inputs6 %Z55 QJJ4:j;Q;Q (((r tool_outputclt|tr|ntj||jd<dS)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. r_Nrc)r)res rset_tool_outputzTranslatedSpan.set_tool_outputs7 &k377 TKKTZ =T=T +,,,r)rr r r)r r )r+r,r r-)r1rr r-)r5r,r r-)r9r,r r-)r<r=r r-)r@r=r r-)r r-)rErFr r-)rIr,rJrr r-) rMrRrNrRrOrRrPrSrQrSr r-)rYr[rZr[r r-)rarr r-)rerr r-)__name__ __module__ __qualname____doc____annotations__ classmethodr&r*r0r4r8r;r?rBrDrHrLrXr`rdrgrrrrrs     [     ****6666DDDD 8 8 8 8JJJJ MMMM9999===='+'+&*8<9=LLLLLL>(,+/ X X X X X X          rr)r r r r)rk __future__rrUr dataclassesrtypingrropentelemetry.sdk.tracerr $langsmith._internal.otel._span_utilsr rrrrrrss#""""" !!!!!! 77777777FFFFFFRRRR W W W W W W W  W W W r