hjCddlmZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZmZmZe rdd lmZmZdd lmZmZmZdd l m!Z!dd l"m#Z#e dedde fZ$ejJe&Z'ejPd%dZ)d&dZ*d'dZ+d(dZ,d)dZ-d*dZ. d+ d,dZ/d-dZ0 d. d/dZ1 d0 d1dZ2d2dZ3Gdded Z4dd!dd" d3d#Z5d4d$Z6y)5) annotationsN) defaultdict)Mapping) TYPE_CHECKINGAnyCallableOptionalTypeVarUnion) TypedDict)client) run_helpers)InputTokenDetailsOutputTokenDetails UsageMetadata) AsyncOpenAIOpenAI)ChatCompletionChunkChoiceChoiceDeltaToolCall) Completion)ResponseStreamEventCrr)boundc2 ddlm}|S#t$rYywxYw)NrNotGiven) openai._typesr ImportErrorrs X/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langsmith/wrappers/_openai.py_get_not_givenr!%s!* s  c t}||S|jDcic]'\}}t||s|jdr|||)c}}Scc}}w#t$r$}t j d||cYd}~Sd}~wwxYw)Nextra_zError stripping NotGiven: )r!items isinstance startswith Exceptionloggererror)d not_givenkves r _strip_not_givenr//s "$  H  1q),h1GAI qD    1!56s3 AA,AAA B"B;BBct|}|jd}|rt|tr|g}|||jd|jd|jdxs$|jdxs|jd|dS)Nstopmodel temperature max_tokensmax_completion_tokensmax_output_tokens) ls_provider ls_model_type ls_model_namels_temperature ls_max_tokensls_stop)r/getr%str) model_typeproviderkwargsstrippedr1s r _infer_invocation_paramsrC>s'H << D 4%v #!g.",,}5!l3- <</ 0- <<+ ,  ctt|}ddd}|D]A}t|dst|jdds(|jj |d<nt t}|D]5}t|dst|jddr!|dxx|jjz cc<t|jddr|jdsddd|d<t|jjd d}|r|dd xx|z cc<t|jjd d}|r|dd xx|z cc<t|jd ds|jj}||D] }||jj|"8|rtt|jd zD cgc]} dc} |d <|j!D]\} } | t#d | Ddt#d| Dddddd|d | <| D]o} t| ddst| j$d d}|r|d | dd xx|z cc<t| j$d d}|sZ|d | dd xx|z cc<q|rt|dddndt#d|Dd|dScc} w)N assistantrolecontentdeltarIrJ function_call)name argumentsrMrN tool_callsc3NK|]}|js|jywN)id.0cs r z"_reduce_choices..ssCQaddADDC%%c3NK|]}|js|jywrR)typerTs r rWz"_reduce_choices..tsI!&&affIrX)indexrSrZfunctionr\rr[c3NK|]}t|ddr|jyw) finish_reasonN)getattrr^rTs r rWz"_reduce_choices..s( 1ot4 s#%)r[r^message)listreversedhasattrr_rKrIrrJr=rLrOr[appendrangemaxkeysr$nextr\) choicesreversed_choicesr`rVrOname_ arguments_tool_calls_list tool_call_r[tool_call_chunkschunks r _reduce_choicesrrQsHW-.G 1g 7177FD#AggllGFO ?J$>OJ F 1g qww 40 "aggoo5"qww6{{?38:/LGO, 5 5vtDO,V4=4$QWW%:%:KN O,[9ZG9qww d3"#''"4"4".%4F "9??3::9EF!F$/4S9J5Ka5O/P Q! Q '1'7'7'9  #E#C*:CTJI.>I4P%'b9 ,GL !% ( * 5*d3#ENNFDAE -e4Z@HEQH!(d!KJ! -e4Z@M&M  $5<Wa0 )      %!Rs K cvtt}|D]1}|jD] }||jj |"3|r>|dj }|j Dcgc] }t|c}|d<n dddddigi}|jdd}|r t|nd|d<|Scc}w) Nrir`rFrGrHusageusage_metadata) rrarir[rd model_dumpvaluesrrpop_create_usage_metadata) all_chunkschoices_by_indexrqchoicer*rioai_token_usages r _reduce_chatrs7B47H:mm :F V\\ * 1 1& 9 :: rN % % '4D4K4K4M )0OG $ ) )kb%IJK LeeGT*O3B/ H s-B6cg}|D]/}|jdj}||j|1dj|}|r|dj }d|ig|d<|Sdd|igi}|S)NrrGrttextri)rirrdjoinrw)r{ all_contentrqrJr*s r _reduce_completionsrsK(--"''     w '(ggk"G rN % % ')*)  H&'*+ , HrDc,|dvr|nd}|r|dnd}|jdxs|jdxsd}|jdxs|jdxsd}|jd xs||z}d |jd xs|jd xsijd |d|jd xs|jd xsijdi}d |jdxs|jdxsijd |d|jdxs|jdxsijdi}|r<||j|dxsdz ||<||j|dxsdz ||<t|||tdi|jD cic] \} } |  | |  c} } t di|jD cic] \} } |  | |  c} } Scc} } wcc} } w)N)priorityflexrorG prompt_tokens input_tokensrcompletion_tokens output_tokens total_tokensaudioprompt_tokens_detailsinput_tokens_details audio_tokens cache_read cached_tokenscompletion_tokens_detailsoutput_tokens_details reasoningreasoning_tokens)rrrinput_token_detailsoutput_token_details)r=rrr$r) r~ service_tierrecognized_service_tierservice_tier_prefixrrrrrr,r-s r rzrzs%(<< $*A " #1%b O,X0C0CN0SXWX /0    /  #&&~6V,:VL    7 8 ""#9:  #n   z*    7 8 ""#9:  #o  !     ; < ""#:;  #n   y)    ; < ""#:;  # ! "8D  # #':&;:$F G L18 349F $ $(;''""#sAA A7A2,A72A7c|xsitjfd}tjfd}tjr|S|S)Nc tjdd|jddurndtd}||i|SNllmstreamTrMrun_type reduce_fnprocess_inputs_invocation_params_fnprocess_outputsr)r traceabler=r/ argsrA decoratorinvocation_params_fnrMoriginal_createrrtextras r createz_get_wrapper..creates_)) #)::h#74#?iT+"6+   *y)4:6::rDc Kt|}tjdd|jddurndtd}||i|d{S7wr)r/rrr=rs r acreatez_get_wrapper..acreatess!&))) #)::h#74#?iT+"6+   0Y/@@@@@sAAAA functoolswrapsris_async) rrMr tracing_extrarrrrrs ``` `` @r _get_wrapperrsn bF___% ; ;& ;___% A A& A"**?;7GGrDc|xsitjfd}tjfd}tjr|S|S)Nc `tjdddtd}||i|SNrrr)rrr/rrArrrMoriginal_parserrs r parsez!_get_parse_wrapper..parse9sM)) +"6+   )y($9&99rDc Kt|}tjdddtd}||i|d{S7wr)r/rrrs r aparsez"_get_parse_wrapper..aparseFsa!&))) +"6+   /Y~.?????s=AAAr)rrMrrrrrrs``` ` @r _get_parse_wrapperr0sc bF__^$ :% :__^$ @% @!)).96DuDrDc`|D](}|jdk(st|jcSiS)Nzresponse.completed)rZ_process_responses_api_outputresponse)eventsevents r _reduce_response_eventsrWs6A ::- -0@ @A IrDc,eZdZUded<ded<ded<y) TracingExtrazOptional[Mapping[str, Any]]metadatazOptional[list[str]]tagszOptional[ls_client.Client]r N)__name__ __module__ __qualname____annotations__rrDr rr^s))  &&rDrF)total ChatOpenAI)r chat_namecompletions_namec x|xsi}d} ddlm}m}t||s t||rd}d}d}t |j jj|t|tjtd|t|j j_t |jj|t|tjtd | |j_t|d rt|j drt|j j d rt|j j jd rxt#|j j jj$|t|tjtd| |j j j_t|drt|j&drUt |j&j|t(t*|tjtd||j&_t|j&d rPt#|j&j$|t*|tjtd| |j&_|S#t$rYwxYw)aPatch the OpenAI client to make it traceable. Supports: - Chat and Responses API's - Sync and async OpenAI clients - create() and parse() methods - with and without streaming Args: client (Union[OpenAI, AsyncOpenAI]): The client to patch. tracing_extra (Optional[TracingExtra], optional): Extra tracing information. Defaults to None. chat_name (str, optional): The run name for the chat completions endpoint. Defaults to "ChatOpenAI". completions_name (str, optional): The run name for the completions endpoint. Defaults to "OpenAI". Returns: Union[OpenAI, AsyncOpenAI]: The patched client. Example: .. code-block:: python import openai from langsmith import wrappers # Use OpenAI client same as you normally would. client = wrappers.wrap_openai(openai.OpenAI()) # Chat API: messages = [ {"role": "system", "content": "You are a helpful assistant."}, { "role": "user", "content": "What physics breakthroughs do you predict will happen by 2300?", }, ] completion = client.chat.completions.create( model="gpt-4o-mini", messages=messages ) print(completion.choices[0].message.content) # Responses API: response = client.responses.create( model="gpt-4o-mini", messages=messages, ) print(response.output_text) .. versionchanged:: 0.3.16 Support for Responses API added. openair)AsyncAzureOpenAI AzureOpenAIazureAzureChatOpenAIrchat)rrrr)rrbeta completionsr responsesr)rrr)rrrr%rrrrrrrpartialrCrrrcrrrrrr)r rrrr7rrs r wrap_openairds:z"'RMK 8 fk *jAQ.R!K)I,  &2 &&#&.. $fk 1 &FKK"!-!!#&.. $e[ !F  FKK ( FKK$$m 4 FKK$$00' :-? KK   ( ( . .  $'!*!2!2(&+" .  $$*v{# 6##X .&2  ''' =+%.%6%6,fk& 'F   # 6##W -%7  &&-+%.%6%6,fk& &F   " ME    s&J,, J98J9c|rH |jdd}|jddx}rt||jd|d<|SiS#t$rd|icYSwxYw)NTjson) exclude_nonemoderurrvr)rwryrzr=r')rrrus r rrs} (((d(HF 7D11u1+A6::n5,'(M I (h' ' (sAA AA)returnzOptional[type])r*dictrr)r?r>r@r>rAr)riz list[Choice]rr)r{zlist[ChatCompletionChunk]rr)r{zlist[Completion]rrrR)r~rrz Optional[str]rr)rr)NNN)rrrMr>rrrOptional[TracingExtra]rOptional[Callable]rrrr)NN) rrrMr>rrrrrrrr)rzlist[ResponseStreamEvent]rr) r rrrrr>rr>rr)rrrr)7 __future__rrlogging collectionsrcollections.abcrtypingrrrr r r typing_extensionsr langsmithr ls_clientrlangsmith.schemasrrrrrr'openai.types.chat.chat_completion_chunkrrropenai.types.completionropenai.types.responsesrr getLoggerrr( lru_cacher!r/rCrrrrrzrrrrrrrrrDr rs"##()!RR* 3: CuX}c9:;   8 $  &;| *  ":>AA)6AAH #&-1/3*. &H&H &H&H* &H - &H ( &H&HZ-1/3 $E$E $E$E* $E - $E  $EN'9E'-1!$ I I*I I  I  IX rD