o Zh=@sddlmZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZmZmZe rfdd lmZmZdd lmZmZmZdd l m!Z!dd l"m#Z#e dedde fdZ$e%e&Z'ej(dRddZ)dSddZ*dTddZ+dUd!d"Z,dVd%d&Z-dWd(d)Z.dXd,d-Z/dYd0d1Z0   dZd[d;d<Z1  d\d]d>d?Z2d^dBdCZ3GdDdEdEedFdGZ4ddHddId_dMdNZ5d`dPdQZ6dS)a) annotationsN) defaultdict)Mapping) TYPE_CHECKINGAnyCallableOptionalTypeVarUnion) TypedDict)client) run_helpers)InputTokenDetailsOutputTokenDetails UsageMetadata) AsyncOpenAIOpenAI)ChatCompletionChunkChoiceChoiceDeltaToolCall) Completion)ResponseStreamEventCrr)boundreturnOptional[type]cCs(z ddlm}|WStyYdSw)NrNotGiven)Z openai._typesr ImportErrorrrQ/var/www/html/lang_env/lib/python3.10/site-packages/langsmith/wrappers/_openai.py_get_not_given%s   r!ddictc sfztdur |WSfdd|DWSty2}ztd||WYd}~Sd}~ww)Ncs2i|]\}}t|s|dr|dus||qS)Zextra_N) isinstance startswith.0kvZ not_givenrr 4sz$_strip_not_given..zError stripping NotGiven: )r!items Exceptionloggererror)r"err*r _strip_not_given/s r1 model_typestrkwargscCs^t|}|d}|rt|tr|g}d||d|d|dp+|dp+|d|dS) NstopopenaimodelZ temperatureZ max_tokensZmax_completion_tokensZmax_output_tokens)Z ls_providerZ ls_model_typeZ ls_model_nameZls_temperatureZ ls_max_tokensZls_stop)r1getr$r3)r2r4strippedr5rrr _infer_invocation_params>s  r:choices list[Choice]c CsRtt|}ddd}|D]}t|dr#t|jddr#|jj|d<nq tt}|D]o}t|drt|jddrB|d|jj7<t|jddr}|dsUddd|d<t|jj d d}|ri|dd |7<t|jj d d}|r}|dd |7<t|jd dr|jj }|dur|D] }||j  |qq*|rd d t t|dD|d <|D]]\} } | tdd| Ddtdd| Dddddd|d | <| D]8} t| ddrt| jd d}|r|d | dd |7<t| jd d}|r|d | dd |7<qq|rt|dddndtdd|Dd|dS)N assistantrolecontentdeltar@rA function_call)name argumentsrDrE tool_callscSsg|]}dqSNr)r'_rrr osz#_reduce_choices..cs|] }|jr|jVqdSrG)idr'crrr sz"_reduce_choices..csrKrG)typerMrrr rOtrP)indexrLrQfunctionrSrrRcss"|] }t|ddr|jVqdS) finish_reasonN)getattrrTrMrrr rOs  )rRrTmessage)listreversedhasattrrUrBr@rrAr8rCrFrRappendrangemaxkeysr,nextrS) r;Zreversed_choicesrVrNrFZname_Z arguments_Ztool_calls_listZ tool_callrRZtool_call_chunkschunkrrr _reduce_choicesQsx    " r` all_chunkslist[ChatCompletionChunk]cCstt}|D]}|jD] }||j|q q|r+|d}dd|D|d<n dddddigi}|d d}|rAt|nd|d <|S) NcSsg|]}t|qSr)r`)r'r;rrr rIsz _reduce_chat..r;rVr=r>r?usageusage_metadata) rrWr;rRrZ model_dumpvaluespop_create_usage_metadata)raZchoices_by_indexr_choicer"oai_token_usagerrr _reduce_chats    rllist[Completion]cCsjg}|D]}|jdj}|dur||qd|}|r,|d}d|ig|d<|Sdd|igi}|S)Nrr>rctextr;)r;rnrZjoinrf)raZ all_contentr_rAr"rrr _reduce_completionss    rprkrc Cs|dp |dp d}|dp|dpd}|dp ||}|dp,|dp,id |dp:|dp:id d }|d pK|d pKid |d pY|d pYidd}t|||tdidd|Dtdidd|DdS)NZ prompt_tokens input_tokensrZcompletion_tokens output_tokens total_tokensZprompt_tokens_detailsZinput_tokens_detailsZ audio_tokensZ cached_tokens)audioZ cache_readZcompletion_tokens_detailsZoutput_tokens_detailsZreasoning_tokens)rtZ reasoningcSi|] \}}|dur||qSrGrr&rrr r+z*_create_usage_metadata..cSrurGrr&rrr r+rv)rqrrrsinput_token_detailsoutput_token_detailsr)r8rrr,r)rkrqrrrsrwrxrrr risR      rioutputsrc Cslz|}|dd}|rt|nd|d<|WSty5}ztd|d|iWYd}~Sd}~ww)Nrdrez"Error processing chat completion: output)rfrhri BaseExceptionr.debug)ryZrdictrkr0rrr _process_chat_completions r}original_createrrD reduce_fn tracing_extraOptional[TracingExtra]invocation_params_fnOptional[Callable]process_outputscsZ|pitfdd}tfdd}tr+|S|S)Nc sBtjdd|ddurndtd}||i|SNllmstreamTrDZrun_typerZprocess_inputsZ_invocation_params_fnrr)r traceabler8r1argsr4 decoratorrrDr~rrtextrarr creates z_get_wrapper..createc sRt|}tjdd|ddurndtd}||i|IdHSr)r1r rr8rrrr acreates z_get_wrapper..acreate functoolswrapsr is_async)r~rDrrrrrrrrr _get_wrappers   roriginal_parsecsV|pitfdd}tfdd}tr)|S|S)Nc s0tjdddtd}||i|SNrrr)r rr1rrrDrrrrr parse"s z!_get_parse_wrapper..parsec s@t|}tjdddtd}||i|IdHSr)r1r rrrrr aparse/s z"_get_parse_wrapper..aparser)rrDrrrrrrrr _get_parse_wrappers   reventslist[ResponseStreamEvent]cCs&|D]}|jdkrt|jSqiS)Nzresponse.completed)rQ_process_responses_api_outputresponse)reventrrr _reduce_response_events@s  rc@s&eZdZUded<ded<ded<dS) TracingExtrazOptional[Mapping[str, Any]]metadatazOptional[list[str]]tagszOptional[ls_client.Client]r N)__name__ __module__ __qualname____annotations__rrrr rGs  rF)totalZ ChatOpenAI)r chat_namecompletions_namer rrc Cs|pi}t|jjj|t|ttdtd|jj_t|jj|t |ttdd|j_t |drXt |j drXt |j jdrXt |j jjdrXt |j jjj |t|ttdd|j jj_ t |drt |jd rut|jj|tt|ttdd |j_t |jdrt |jj |t|ttdd|j_ |S) 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. chat)rrrr)rrbeta completionsr responsesr)rrr)rrrrrlrpartialr:r}rprYrrrrrr)r rrrrrr wrap_openaiMsl=              rrcCsT|r(z|jddd}|dd}rt||d<|WSty'd|iYSwiS)NTjson)Z exclude_nonemoderdrerz)rfrhrir-)rrzrdrrr rs   r)rr)r"r#rr#)r2r3r4r#)r;r<rr#)rarbrr#)rarmrr#)rkr#rr)ryr)NNN)r~rrDr3rrrrrrrrrr)NN) rrrDr3rrrrrrrr)rrrr#) r rrrrr3rr3rr)rrrr#)7 __future__rrlogging collectionsrcollections.abcrtypingrrrrr r typing_extensionsr Z langsmithr Z ls_clientr Zlangsmith.schemasrrrr6rrZ'openai.types.chat.chat_completion_chunkrrrZopenai.types.completionrZopenai.types.responsesrr getLoggerrr. lru_cacher!r1r:r`rlrprir}rrrrrrrrrr sP             >   / - ' {