jw?+ddlZddlZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmcmZddlmZmZddlmZddlmZmZmZmZddlmZmZmZej>e Z!Gdd Z" d"d e#d e#d e#d ejHdee#dee#dee#dedeejdeejdeee#efdeee#efdeee#efdeee#deeee#efe jJfdee#deeee#fdeedee&deeejNde"f*dZ( d#d e#d e#dee#d e#d e#d ejHdeejdeee#efdeee#efdeee#efdeee#deeee#efe jJfdee#deeee#fdeedee&deeejNddf$d!Z)y)$N)AnyDictListOptionalUnion)messagesstreamer) converters)DistributedTraceHeadersDict ErrorInfoDict LLMProviderSpanType) constantsvalidation_helpershelpersc!eZdZ d!dedededej dedeef dZ d"d eejd ee ee fd ee ee fd ee ee fd ee edee e ee fe jfdeedee eefdeedeeddfdZ d"d eejd ee ee fd ee ee fd ee ee fd ee edee e ee fe jfdeedee eefdeedeeddfdZdddddddddddej(dddfdeedeededeejd eejd ee ee fd ee ee fd ee ee fd ee edee e ee fe jfdeededeedeedee ej.ddf dZ d#dededeedeeddf dZdefd Zy)$SpanNidtrace_id project_namemessage_streamer url_overrideparent_span_idcX||_||_||_||_||_||_y)z A Span object. This object should not be created directly, instead use the `span` method of a Trace (:func:`opik.Opik.span`) or another Span (:meth:`opik.Span.span`). N)rrr _streamer _project_name _url_override)selfrrrrrrs z/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/api_objects/span/span_client.py__init__z Span.__init__s1  ,)))end_timemetadatainputoutputtagsusagemodelprovider error_info total_costreturnc j||ntj}|j||||||||| |  y)a End the span and update its attributes. This method is similar to the `update` method, but it automatically computes the end time if not provided. Args: end_time: The end time of the span. If not provided, the current time will be used. metadata: Additional metadata to be associated with the span. input: The input data for the span. output: The output data for the span. tags: A list of tags to be associated with the span. usage: Usage data for the span. In order for input, output and total tokens to be visible in the UI, the usage must contain OpenAI-formatted keys (they can be passed additionaly to original usage on the top level of the dict): prompt_tokens, completion_tokens and total_tokens. If OpenAI-formatted keys were not found, Opik will try to calculate them automatically if the usage format is recognized (you can see which provider's formats are recognized in opik.LLMProvider enum), but it is not guaranteed. model: The name of LLM. provider: The provider of LLM. You can find providers officially supported by Opik for cost tracking in `opik.LLMProvider` enum. If your provider is not here, please open an issue in our github - https://github.com/comet-ml/opik. If your provider not in the list, you can still specify it but the cost tracking will not be available error_info: The dictionary with error information (typically used when the span function has failed). total_cost: The cost of the span in USD. This value takes priority over the cost calculated by Opik from the usage. Returns: None N) r#r$r%r&r'r(r)r*r+r,)datetime_helperslocal_timestampupdate rr#r$r%r&r'r(r)r*r+r,s r endzSpan.end,sOP!,H2B2R2R2T  !!  r"c tdid|jd|jd|jd|jd|j d|j d|d|d |d |d |d |d |d|d| d| y)aA Update the span attributes. Args: end_time: The end time of the span. metadata: Additional metadata to be associated with the span. input: The input data for the span. output: The output data for the span. tags: A list of tags to be associated with the span. usage: Usage data for the span. In order for input, output and total tokens to be visible in the UI, the usage must contain OpenAI-formatted keys (they can be passed additionaly to original usage on the top level of the dict): prompt_tokens, completion_tokens and total_tokens. If OpenAI-formatted keys were not found, Opik will try to calculate them automatically if the usage format is recognized (you can see which provider's formats are recognized in opik.LLMProvider enum), but it is not guaranteed. model: The name of LLM. provider: The provider of LLM. You can find providers officially supported by Opik for cost tracking in `opik.LLMProvider` enum. If your provider is not here, please open an issue in our github - https://github.com/comet-ml/opik. If your provider not in the list, you can still specify it but the cost tracking will not be available error_info: The dictionary with error information (typically used when the span function has failed). total_cost: The cost of the span in USD. This value takes priority over the cost calculated by Opik from the usage. Returns: None rrrrrrr#r$r%r&r'r(r)r*r+r,N) update_spanrrrrrrr2s r r1z Span.updatedsH  ww ]]  .. ++  "^^  ++          " "! r"generalnametype start_time attachmentsctdid|jd|jd|jd|jd|d|j d|d|d |d |d |d |d |d| d| d| d| d| d|d|S)a4 Create a new child span within the current span. Args: id: The ID of the span should be in UUIDv7 format. If not provided, a new ID will be generated. name: The name of the span. type: The type of the span. Defaults to "general". start_time: The start time of the span. If not provided, the current time will be used. end_time: The end time of the span. metadata: Additional metadata to be associated with the span. input: The input data for the span. output: The output data for the span. tags: A list of tags to be associated with the span. usage: Usage data for the span. In order for input, output and total tokens to be visible in the UI, the usage must contain OpenAI-formatted keys (they can be passed additionally to the original usage on the top level of the dict): prompt_tokens, completion_tokens and total_tokens. If OpenAI-formatted keys were not found, Opik will try to calculate them automatically if the usage format is recognized (you can see which provider's formats are recognized in opik.LLMProvider enum), but it is not guaranteed. model: The name of LLM (in this case `type` parameter should be == `llm`) provider: The provider of LLM. You can find providers officially supported by Opik for cost tracking in `opik.LLMProvider` enum. If your provider is not here, please open an issue in our GitHub - https://github.com/comet-ml/opik. If your provider is not in the list, you can still specify it, but the cost tracking will not be available error_info: The dictionary with error information (typically used when the span function has failed). total_cost: The cost of the span in USD. This value takes priority over the cost calculated by Opik from the usage. attachments: The list of attachments to be uploaded to the span. Returns: Span: The created child span object. rrrrspan_idrr8r9r:r#r$r%r&r'r(r)r*r+r,r;r5) create_spanrrrrr)rrr8r9r:r#r$r%r&r'r(r)r*r+r,r;s r spanz Span.spans\ ]] ++ ++ "^^     77    "       ! "# $"% &"' ($)  r"value category_namereasonc tjtj|j||||tj |j g}|jj|y)aH Log a feedback score for the span. Args: name: The name of the feedback score. value: The value of the feedback score. category_name: The category name for the feedback score. reason: The reason for the feedback score. Returns: None )rr8r@rArBsourcer)batchN) r!AddSpanFeedbackScoresBatchMessageFeedbackScoreMessagerrFEEDBACK_SCORE_SOURCE_SDKrrput)rr8r@rArBadd_span_feedback_batch_messages r log_feedback_scorezSpan.log_feedback_scoresb&+3*T*T--ww"/!$>>!%!3!3  + ' :;r"c4|j|jdS)zg Returns headers dictionary to be passed into tracked function on remote node. )opik_parent_span_id opik_trace_id)rr)rs r get_distributed_trace_headersz"Span.get_distributed_trace_headerss (,wwOOr")N) NNNNNNNNNN)NN)__name__ __module__ __qualname__strr Streamerrr!datetimerrrr llm_usage OpikUsagerr floatr3r1OPENAIr attachment Attachmentr?rKr rOr5r"r rrs#)-* ** * #++ *  *! **15-1*.+/$(FJ#6:.2&*6 8,,-6 4S>*6 S#X' 6 c3h( 6 tCy! 6 d38ni.A.AABC6 }6 5c!1236 ]+6 UO6  6 t15-1*.+/$(FJ#6:.2&*5 8,,-5 4S>*5 S#X' 5 c3h( 5 tCy! 5 d38ni.A.AABC5 }5 5c!1235 ]+5 UO5  5 r!""2604-1*.+/$(FJ# + 2 2.2&*=A!C SMC smC  C X../ C 8,,- C 4S>*C S#X'C c3h(C tCy!C d38ni.A.AABCC }C C ]+C UOC d:#8#89:!C " #C R(, $ !<!<!< } !<  !<  !r> s,!,g*2H2H2JG , 2B2R2R2T 2JJ   +00uxP"44"&     ' !"#$)88:%(,-*O(=(S(S /"!)) ) %  !: ; +  %)!!  r"rc>tj| t| }|tj| |}t j |||||||| | || | ||}|2|D]-}tj|d|||}|j|/|j|y)Nr]r_)r=rrrr#r$r%r&r'r(r)r*r+r,r?ra) rrgrhrrirUpdateSpanMessagerkrlrI)rrrrrrr#r$r%r&r'r(r)r*r+r,r;rmupdate_span_messagerbros r r6r6Zs& 2JJ   +00uxP"44%! &"*O(=(S(S /")) ) %  !: ; +,-r")NNNr7NNNNNNNNNNNN) NNNNNNNNNNN)*rUloggingtypingrrrrropik.datetime_helpersr/opik.id_helpersreopik.llm_usagerVopik.api_objects.attachment api_objectsrZopik.message_processingrr r rk opik.typesr r rrrrr getLoggerrPrhrrSrTrWrXr[r>r6r5r"r r~s|330$"006< 65   8 $pPpPp"$(.2,0)-&*'+ $BF26*."&9=)LLLL'' L c] L SM L 3-L L**+Lx(()LtCH~&L DcN #L T#s(^ $L 49 L E$sCx.)*=*==> ?L C=!L"u[#-./#L$'%L&'L($z4456)L* +Ll-1)-&*'+ $BF26*."&9=#8. 8.8.SM8. 8.  8. '' 8.x(()8.tCH~&8. DcN #8. T#s(^ $8. 49 8. E$sCx.)*=*==> ?8. C=8.u[#-./8.'8. !8."$z4456#8.$ %8.r"