ihuddlmZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZddlmZdd lmZmZmZdd lmZdd lm Z dd l!m"Z"m#Z#m$Z$dd l%m&Z&erddl'Z'ddl(m)Z)ddl*m+Z+GddeZ,GddeZ-ddZ.ddZ/ ddZ0 ddZ1ee e-e,e de dfZ2Gddee e2fZ3y) ) annotationsN)JSONDecodeError)sleep) TYPE_CHECKINGAnyCallableDictListOptionalSequenceTupleTypeUnion) AgentAction AgentFinish)CallbackManager)dumpd)RunnableConfigRunnableSerializable ensure_config)BaseTool)convert_to_openai_tool) BaseModelFieldmodel_validator)Self) ThreadMessage)RequiredActionFunctionToolCallc8eZdZUdZded<ded<eddZy)OpenAIAssistantFinishzuAgentFinish with run and thread metadata. Parameters: run_id: Run id. thread_id: Thread id. strrun_id thread_idcyz]Check if the class is serializable by LangChain. Returns: False Fclss d/var/www/html/dev/engine/venv/lib/python3.12/site-packages/langchain/agents/openai_assistant/base.pyis_lc_serializablez(OpenAIAssistantFinish.is_lc_serializable0Nreturnbool__name__ __module__ __qualname____doc____annotations__ classmethodr*r&r,r)r r %s% KNr,r cBeZdZUdZded<ded<ded<eddZy) OpenAIAssistantActionzAgentAction with info needed to submit custom tool output to existing run. Parameters: tool_call_id: Tool call id. run_id: Run id. thread_id: Thread id r! tool_call_idr"r#cyr%r&r's r)r*z(OpenAIAssistantAction.is_lc_serializableGr+r,Nr-r0r&r,r)r8r8:s+ KNr,r8c ddl}|jS#t$r}td|d}~wt$r}td|d}~wwxYwNrzBUnable to import openai, please install with `pip install openai`.zuPlease make sure you are using a v1.1-compatible version of openai. You can install with `pip install "openai>=1.1"`.)openaiOpenAI ImportErrorAttributeErrorr=es r)_get_openai_clientrCQs] v}}  P    <   A  , A  AA c ddl}|jS#t$r}td|d}~wt$r}td|d}~wwxYwr<)r= AsyncOpenAIr?r@rAs r)_get_openai_async_clientrGas` !v!!##  P    <  rDcDd}t|txr d|vxr|d|vS)zd |d |jd d g|jd d}|j||}nd|vrd|jjjjj!|d|d d |jd } |j#|}n:|jjjjjdi|}|j%|j&|j(} |j/|} |j1| | S#t*$r} |j-| | d} ~ wwxYw#t*$r(} |j-| |j3| d} ~ wwxYw)a/Invoke assistant. Args: input: Runnable input dict that can have: content: User message when starting a new run. thread_id: Existing thread to use. run_id: Existing run to use. Should only be supplied when providing the tool output for a required action after an initial invocation. message_metadata: Metadata to associate with new message. thread_metadata: Metadata to associate with new thread. Only relevant when new thread being created. instructions: Additional run instructions. model: Override Assistant model for this run. tools: Override Assistant tools for this run. parallel_tool_calls: Allow Assistant to set parallel_tool_calls for this run. top_p: Override Assistant top_p for this run. temperature: Override Assistant temperature for this run. max_completion_tokens: Allow setting max_completion_tokens for this run. max_prompt_tokens: Allow setting max_prompt_tokens for this run. run_metadata: Metadata to associate with new run. config: Runnable config. Defaults to None. Return: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. callbackstagsmetadatainheritable_callbacksinheritable_tagsinheritable_metadatarun_namerdintermediate_stepsr#usercontentmessage_metadataroler{rrthread_metadatamessagesrrr"r{r~rrNrrr&)rr configuregeton_chain_startrget_namer[_parse_intermediate_stepsrVrhthreadsrunssubmit_tool_outputs_create_thread_and_runrrj _create_run _wait_for_runrkr# BaseExceptionon_chain_error _get_response on_chain_endrM r`inputconfigrlcallback_manager run_manager tool_outputsrunthread_rBresponses r)invokezOpenAIAssistantRunnable.invokesZBv&*44"(**["9#ZZ/!'J!7  '55 $KVZZ %;%Nt}}6 $ }}+?!@#==./  Hdkk&&..33GGW,WE)%+',Y'7(- 2D(E!!& *; < 11%@&KK$$,,55<<+&!),"YY'9: = &&u-Hdkk&&..33GGP%P$$SVVS]];C ))#.H  $ $X .O   & &q )G     & &q388: & >G s1E6H+I+ I 4II  J#I;;J)rWcK|xs t}|Dcgc] }t|}}|jjj ||||d{} |d| j |d|Scc}w7w)a Async create an AsyncOpenAI Assistant and instantiate the Runnable. Args: name: Assistant name. instructions: Assistant instructions. tools: Assistant tools. Can be passed in OpenAI format or as BaseTools. model: Assistant model to use. async_client: AsyncOpenAI client. Will create default async_client if not specified. Returns: AsyncOpenAIAssistantRunnable configured to run using the created assistant. rcN)rXrWr&)rGrRrhrirjrk) r(rdrerfrgrWrlrN openai_toolsrms r)acreate_assistantz)OpenAIAssistantRunnable.acreate_assistantos4$A'?'A ?DEt,T2E E&++66==% >  R <R6RRF sA6A/.A6A4A6cK|xsi}tj|jd|jd|jd}|jt |||jdxs|j } |j rq|jdr`|j|dd{}|jjjjjdi|d{}nd |vrFd |d |jd d g|jdd}|j||d{}nd|vrt|jjjjj|d |d d |jd d{} |j!|d{}nB|jjjjjdi|d{}|j#|j$|j&d{} |j-|} |j/| | S77a7777d78#t($r} |j+| | d} ~ wwxYw#t($r(} |j+| |j1| d} ~ wwxYww)aAsync invoke assistant. Args: input: Runnable input dict that can have: content: User message when starting a new run. thread_id: Existing thread to use. run_id: Existing run to use. Should only be supplied when providing the tool output for a required action after an initial invocation. message_metadata: Metadata to associate with a new message. thread_metadata: Metadata to associate with new thread. Only relevant when a new thread is created. instructions: Overrides the instructions of the assistant. additional_instructions: Appends additional instructions. model: Override Assistant model for this run. tools: Override Assistant tools for this run. parallel_tool_calls: Allow Assistant to set parallel_tool_calls for this run. top_p: Override Assistant top_p for this run. temperature: Override Assistant temperature for this run. max_completion_tokens: Allow setting max_completion_tokens for this run. max_prompt_tokens: Allow setting max_prompt_tokens for this run. run_metadata: Metadata to associate with new run. config: Runnable config. Defaults to None. kwargs: Additional arguments. Return: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. rprqrrrsrwrxryNr#rzr{r|r}rrr"rrr&)rrrrrrr[_aparse_intermediate_stepsrWrhrrr_acreate_thread_and_runrrj _acreate_run_await_for_runrkr#rrrrrMrs r)ainvokezOpenAIAssistantRunnable.ainvokesF2*44"(**["9#ZZ/!'J!7  '55 $KVZZ %;%Nt}}6 ( }}+?!@%)%D%D./&  TD--22::??SS"E)%+',Y'7(- 2D(E!!& *; < !88GG&++0088AAHH+&!),"YY'9: I !--e44TD--22::??SS++CFFCMMBBC ))#.H  $ $X .O[ H 5C   & &q )G     & &q388: & >G sB K  4I2I!AI2I$A I2I'AI2-I*.I2I,AI2 I. -I27I08I2=JK !I2$I2'I2*I2,I2.I20I22 J;JJK  K#KKK c|d\}}|j|j|j}t}|jr7|jj j Dchc]}|j}}|Dcgc]+\}}|j|vrt||jd-} }}| |j|jd} | Scc}wcc}}wN)outputr9)rr"r# rr"r#setrequired_actionr tool_callsrkr9r! r`ry last_action last_outputrrequired_tool_call_idstcactionrrrs r)rz1OpenAIAssistantRunnable._parse_intermediate_stepss$6b#9 [  !3!3[5J5JK!$    # 3 3 G G R R&& "& #5 ""&<<6{F4G4G H  )!(($..  #"& s 'C0Cc|jDcic] \}}|dvr||}}}|jjjjj |dfd|j i|Scc}}wN) rergrfadditional_instructionsparallel_tool_callstop_p temperaturemax_completion_tokensmax_prompt_tokens run_metadatar#rX)itemsrVrhrrrjrXr`rkvparamss r)rz#OpenAIAssistantRunnable._create_runs  1   qD  "4t{{'',,33 +  **   # sA2c|jDcic] \}}|dvr||}}}|jjjjd|j |d|}|Scc}}wN) rergrfrrrrrr)rXrr&)rrVrhrcreate_and_runrXr`rrrrrrs r)rz.OpenAIAssistantRunnable._create_thread_and_runs  1   qD  6dkk&&55 **    + sA'c  |jdk(rddltjjj dd}tjjj dd}|dkDxs |dk(xr|dk\|j jjjj|jd}|Dcgc]}|j|jk(s| }}|js|S|Dcgc]}|jD]}|}}}t!fd|Drd j#d |D}t%||j|jd d |j|j S|jdk(r|js |j&j(j*Sg} |j&j(j*D]} | j,} t/j0| j2d} t;| dk(r d| vr| d} | j=t?| j8| | jd |j|j| St/j@|jCd}t7d|jd|dcc}wcc}}w#t4$r+} t7d| j2d| j8| d} ~ wwxYw)N completedr.ascorderc3K|]n}r4t|jjjjn3t|jjjj pywNrLtypesrhrTextContentBlockMessageContentText.0r{r=version_gte_1_14s r) z8OpenAIAssistantRunnable._get_response..Lf  (w (9(9(A(A(R(RS# ))11DD A4A7 c3HK|]}|jjywrtextvaluerr{s r)rz8OpenAIAssistantRunnable._get_response..W"L'7<<#5#5"L "rr#r" return_valueslogr"r#requires_actionFstrict*Received invalid JSON function arguments:  for function __arg1rN tool_inputr9rr"r#indentUnexpected run status: . Full run info: ))"statusr=intversionVERSIONsplitrVrhrrlistr#r"rkr[r{alljoinr rrrfunctionjsonloads argumentsr ValueErrorrdlenappendr8dumpsrMr`r major_version minor_versionrmsg new_messages msg_contentansweractions tool_callrargsrBrun_infor=rs @@r)rz%OpenAIAssistantRunnable._get_response7s :: $  6 6 < H,4LCszzSVV7KCLLL==##'3 #s{{8C F  &  "LV"LL($!$!ff vv--  ZZ, ,==**>>IIIG 00DDOO  $--::h&8&8GD t9>h$&6>D)%]]#'%.\\"vv"%--   *Nzz#((*Q7H)#**5I(STU qMB'$D#--.nX]]OMs*K4K K ?!K L&K>>Lcd}|rf|jjjjj ||}|j dv}|rt |jdz |rfSNT)r#) in_progressqueuedi)rVrhrrretrieverrrZr`r"r#rrs r)rz%OpenAIAssistantRunnable._wait_for_runsf ++""**//8898UC**(AAKd))D01   r,cK|d\}}|j|j|j}t}|jr7|jj j Dchc]}|j}}|Dcgc]+\}}|j|vrt||jd-} }}| |j|jd} | Scc}wcc}}wwrrrs r)rz2OpenAIAssistantRunnable._aparse_intermediate_stepss$6b#9 [  !3!3[5J5JK!$    # 3 3 G G R R&& "& #5 ""&<<6{F4G4G H  )!(($..  #"& sA'C)C<C0C3*Cc K|jDcic] \}}|dvr||}}}|jjjjj |dfd|j i|d{Scc}}w7 wr)rrWrhrrrjrXrs r)rz$OpenAIAssistantRunnable._acreate_runs  1   qD  "@T&&++3388?? +  **    # " sBA<AB7B8 BcK|jDcic] \}}|dvr||}}}|jjjjd|j |d|d{}|Scc}}w7 wr)rrWrhrrrXrs r)rz/OpenAIAssistantRunnable._acreate_thread_and_runs  1   qD  BD%%**22AA **     +  sA9A1AA9*A7+ A9c *K|jdk(rddltjjj dd}tjjj dd}|dkDxs |dk(xr|dk\|j jjjj|jdd{}|Dcgc]}|j|jk(s| }}|js|S|Dcgc]}|jD]}|}}}t!fd|Drd j#d |D}t%||j|jd d |j|j S|jdk(r|js |j&j(j*Sg} |j&j(j*D]} | j,} t/j0| j2d} t;| dk(r d| vr| d} | j=t?| j8| | jd |j|j| St/j@|jCd}t7d|jd|d7cc}wcc}}w#t4$r+} t7d| j2d| j8| d} ~ wwxYww)Nrrrrrrrc3K|]n}r4t|jjjjn3t|jjjj pywrrrs r)rz9OpenAIAssistantRunnable._aget_response..rrrc3HK|]}|jjywrrrs r)rz9OpenAIAssistantRunnable._aget_response..rrrrrrFrrrrrrrrrr)"rr=rrrrrWrhrrrr#r"rkr[r{rrr rrrrrrrrrrdrrr8rrMrs @@r)_aget_responsez&OpenAIAssistantRunnable._aget_responses :: $  6 6 < >IIIG 00DDOO  $--::h&8&8GD t9>h$&6>D)%]]#'%.\\"vv"%--   *Nzz#((*Q7H)#**5I(STU wMB'$D#--.nX]]OMs\CLKLK>KLK1CL !K*B%L L L%&L  LLcKd}|r|jjjjj ||d{}|j dv}|r*t j|jdz d{|rS7C7 wr) rWrhrrrrasynciorrZrs r)rz&OpenAIAssistantRunnable._await_for_runs ))..66;;DD)EC**(AAKmmD$7$7$$>???   @s*AB B 9B ?B B B  B )r.r)rdr!rer!rfSequence[Union[BaseTool, dict]]rgr!rVz2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]rlrr.rTr)rrMrzOptional[RunnableConfig]rlrr. OutputType)rdr!rer!rfr rgr!rWzr0s"    ;4%XX)H66"7 K*K.    D    , D , ,  )*, R 243CDR r,