~tj zdZddlmZddlZddlmZddlmZmZddlm Z m Z dd Z dd Z ddZ ddZddZddZdS) u@Small pure helpers for the Pipecat → LangSmith span processor.) annotationsN)datetime)AnyOptional)build_assistant_messagetry_parse_json_object input_datarreturnlist[dict[str, Any]]c tj|}n#tjtf$rgcYSwxYwt |t sgSd|DS)zHParse a Pipecat ``llm`` span's ``input`` (a JSON message list) to dicts.c<g|]}t|t|S) isinstancedict).0ms e/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/integrations/pipecat/_helpers.py z&parse_llm_messages..s' 2 2 2!jD11 2A 2 2 2)jsonloadsJSONDecodeError TypeErrorrlist)r raws rparse_llm_messagesrsoj$$  ) ,  c4  2 2s 2 2 22s 22valuec\t|ttfrt|ndS)zHCoerce a numeric span-attribute value to ``int``, or ``None`` otherwise.N)rintfloat)rs r_as_intr!s%#EC<88 B3u:::dBr attributesdict[str, Any]ci}t|dx}||d<t|dx}||d<i}t|dx} ||d<t|dx} ||d <|r||d <t|d x}d |i|d <|S)zFRead Pipecat's ``gen_ai.usage.*`` span keys into ``set_usage`` kwargs.zgen_ai.usage.input_tokensN input_tokenszgen_ai.usage.output_tokens output_tokensz$gen_ai.usage.cache_read.input_tokens cache_readz(gen_ai.usage.cache_creation.input_tokenscache_creationinput_token_detailszgen_ai.usage.reasoning_tokens reasoningoutput_token_details)r!get)r"usagev input_details rextract_llm_usager0s E Z^^$?@@ A AAN !n Z^^$@AA B BBO!"o#%L Z^^$JKK L LL &' \" Z^^$NOO P PP *+ %&4'3#$ Z^^$CDD E EER)4a(8$% Lr output_datact|}t|tr|drddi|St|tr|nt |}t |S)zBuild the assistant completion for a Pipecat ``llm`` span. A structured response carrying ``tool_calls`` is forwarded in its OpenAI shape; anything else becomes plain assistant text. tool_callsrole assistant)rrrr,strr)r1parsedcontents rbuild_completion_messager96st #; / /F&$/FJJ|$<$</ .v..' S99Okks;?O?OG "7 + ++r timestampr6rc ttj|dzS#tt f$rYdSwxYw)zDParse an ISO 8601 timestamp to epoch nanoseconds (0 if unparseable).geAr)rr fromisoformatr: ValueErrorr)r:s r iso_to_nsr>CsX8))44>>@@3FGGG  "qqs58A  A messageOptional[tuple]c|d}|rdtd|DfS|ddkr$dt|dfSdS) aDedup identity for a tool-round-trip message, or ``None`` otherwise. Only assistant tool-call messages and tool-result messages carry one; user turns and plain assistant replies (owned by other sources) return ``None`` so they are ignored here. r3assistant_tool_callc3ZK|]&}t|dV'dS)idN)r6r,)rcs r z#tool_message_key..Ts4,R,R!St-=-=,R,R,R,R,R,Rrr4tool tool_result tool_call_idN)r,tupler6)r?r3s rtool_message_keyrKKs{\**JT%u,R,Rz,R,R,R'R'RSS{{6f$$s7;;~#>#>??@@ 4r)r rr r )rrr r)r"r#r r#)r1rr r#)r:r6r r)r?r#r r@)__doc__ __future__rrrtypingrr"langsmith._internal.voice._helpersrrrr!r0r9r>rKrrrrPsFF""""""   3333CCCC . , , , ,      r