j xdZddlmZddlZddlmZddlmZmZddlm Z m Z d dZ ddZ dd Z dd Zdd Zdd Zy)u@Small pure helpers for the Pipecat → LangSmith span processor.) annotationsN)datetime)AnyOptional)build_assistant_messagetry_parse_json_objectc tj|}t |t sgS|Dcgc]}t |t s|c}S#tjtf$rgcYSwxYwcc}w)zHParse a Pipecat ``llm`` span's ``input`` (a JSON message list) to dicts.)jsonloadsJSONDecodeError TypeError isinstancelistdict) input_datarawms q/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langsmith/integrations/pipecat/_helpers.pyparse_llm_messagesrsijj$ c4   2s!jD1As 22  ) ,  3sA A+A+ A('A(cHt|ttfr t|SdS)zHCoerce a numeric span-attribute value to ``int``, or ``None`` otherwise.N)rintfloat)values r_as_intrs#EC<83u:BdBcni}t|jdx}||d<t|jdx}||d<i}t|jdx} ||d<t|jdx} ||d<|r||d <t|jd x}d |i|d <|S) zFRead Pipecat's ``gen_ai.usage.*`` span keys into ``set_usage`` kwargs.zgen_ai.usage.input_tokens 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)rget) attributesusagev input_details rextract_llm_usager)sE Z^^$?@ AAN !n Z^^$@A BBO!"o#%L Z^^$JK LL &' \" Z^^$NO PP *+ %&'3#$ Z^^$CD EER)4a(8$% Lrct|}t|tr|jdrddi|St|tr|n t |}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) output_dataparsedcontents rbuild_completion_messager26sS #; /F&$FJJ|$< .v..' S9ks;?OG "7 ++rc ttj|jdzS#tt f$rYywxYw)zDParse an ISO 8601 timestamp to epoch nanoseconds (0 if unparseable).geAr)rr fromisoformat timestamp ValueErrorr )r5s r iso_to_nsr7CsB8)))4>>@3FGG  "s.1AAc|jd}|rdtd|DfS|jddk(rdt|jdfSy) 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. r+assistant_tool_callc3PK|]}t|jd yw)idN)r.r$).0cs r z#tool_message_key..Ts,Rz!St-=zs$&r,tool tool_result tool_call_idN)r$tupler.)messager+s rtool_message_keyrDKsX\*J%u,Rz,R'RSS{{6f$s7;;~#>?@@ r)rrreturnzlist[dict[str, Any]])rrrEr)r%dict[str, Any]rErF)r/rrErF)r5r.rEr)rCrFrEzOptional[tuple])__doc__ __future__rr rtypingrr"langsmith._internal.voice._helpersrrrrr)r2r7rDrrrLs8F"   3C . , r