Ng5mddlmZddlZddlZddlmZddlmZddlmZm 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-d#dZ.d$dZ/d%dZ0d&dZ1ee e-e,e de d fZ2Gd!d"ee e2fZ3dS)') annotationsN)JSONDecodeError)sleep) TYPE_CHECKINGAnyCallableDictListOptionalSequenceTupleTypeUnion) AgentAction AgentFinish)CallbackManager)dumpd)RunnableConfigRunnableSerializable ensure_config)BaseTool)convert_to_openai_tool) BaseModelFieldmodel_validator)Self) ThreadMessage)RequiredActionFunctionToolCallc@eZdZUdZded<ded<ed dZdS) OpenAIAssistantFinishzuAgentFinish with run and thread metadata. Parameters: run_id: Run id. thread_id: Thread id. strrun_id thread_idreturnboolcdSz]Check if the class is serializable by LangChain. Returns: False Fclss b/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/agents/openai_assistant/base.pyis_lc_serializablez(OpenAIAssistantFinish.is_lc_serializable0 uNr$r%__name__ __module__ __qualname____doc____annotations__ classmethodr,r(r.r+r r %sRKKKNNN[r.r cJeZdZUdZded<ded<ded<ed dZd S) 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#r$r%cdSr'r(r)s r+r,z(OpenAIAssistantAction.is_lc_serializableGr-r.Nr/r0r(r.r+r8r8:s^KKKNNN[r.r8r$ openai.OpenAIc 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_clientrDQs  v}  P     <    A/ AA  Aopenai.AsyncOpenAIc ddl}|jS#t$r}td|d}~wt$r}td|d}~wwxYwr=)r> AsyncOpenAIr@rArBs r+_get_openai_async_clientrIas  !v!###  P     <   rEtool:Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]r%cLd}t|to d|vo |d|vS)zd,Zd?d.Zd@d1Zdd4Zd?d5Zd@d6ZdS)AOpenAIAssistantRunnablea Run an OpenAI Assistant. Example using OpenAI tools: .. code-block:: python from langchain_experimental.openai_assistant import OpenAIAssistantRunnable interpreter_assistant = OpenAIAssistantRunnable.create_assistant( name="langchain assistant", instructions="You are a personal math tutor. Write and run code to answer math questions.", tools=[{"type": "code_interpreter"}], model="gpt-4-1106-preview" ) output = interpreter_assistant.invoke({"content": "What's 10 - 4 raised to the 2.7"}) Example using custom tools and AgentExecutor: .. code-block:: python from langchain_experimental.openai_assistant import OpenAIAssistantRunnable from langchain.agents import AgentExecutor from langchain.tools import E2BDataAnalysisTool tools = [E2BDataAnalysisTool(api_key="...")] agent = OpenAIAssistantRunnable.create_assistant( name="langchain assistant e2b tool", instructions="You are a personal math tutor. Write and run code to answer math questions.", tools=tools, model="gpt-4-1106-preview", as_agent=True ) agent_executor = AgentExecutor(agent=agent, tools=tools) agent_executor.invoke({"content": "What's 10 - 4 raised to the 2.7"}) Example using custom tools and custom execution: .. code-block:: python from langchain_experimental.openai_assistant import OpenAIAssistantRunnable from langchain.agents import AgentExecutor from langchain_core.agents import AgentFinish from langchain.tools import E2BDataAnalysisTool tools = [E2BDataAnalysisTool(api_key="...")] agent = OpenAIAssistantRunnable.create_assistant( name="langchain assistant e2b tool", instructions="You are a personal math tutor. Write and run code to answer math questions.", tools=tools, model="gpt-4-1106-preview", as_agent=True ) def execute_agent(agent, tools, input): tool_map = {tool.name: tool for tool in tools} response = agent.invoke(input) while not isinstance(response, AgentFinish): tool_outputs = [] for action in response: tool_output = tool_map[action.tool].invoke(action.tool_input) tool_outputs.append({"output": tool_output, "tool_call_id": action.tool_call_id}) response = agent.invoke( { "tool_outputs": tool_outputs, "run_id": action.run_id, "thread_id": action.thread_id } ) return response response = execute_agent(agent, tools, {"content": "What's 10 - 4 raised to the 2.7"}) next_response = execute_agent(agent, tools, {"content": "now add 17.241", "thread_id": response.thread_id}) )default_factoryrclientN async_clientr! assistant_idg@@floatcheck_every_msFr%as_agentafter)moder$rc`|j&ddl}|jj}|j||_|S)Nr)api_key)r[r>rZrcrH)selfr>rcs r+validate_async_clientz-OpenAIAssistantRunnable.validate_async_clients;   $ MMMk)G 2 27 C C CD  r.)rZname instructionstoolsSequence[Union[BaseTool, dict]]model2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]kwargsc |p t}|jj||d|D|}|d|j|d|S)a*Create an OpenAI 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. client: OpenAI or AzureOpenAI client. Will create a default OpenAI client if not specified. kwargs: Additional arguments. Returns: OpenAIAssistantRunnable configured to run using the created assistant. c,g|]}t|Sr(rV.0rJs r+ z.s!@@@$'--@@@r.rfrgrhrj)r\rZr()rDbeta assistantscreateid)r*rfrgrhrjrZrl assistants r+create_assistantz(OpenAIAssistantRunnable.create_assistantso2/-//K*11%@@%@@@ 2  sF VFFvFFFr.inputrQconfigOptional[RunnableConfig] OutputTypec $t|}tj|d|d|d}|t |||dp|} |jrR|dr=||d}|j j j j j di|}nd|vrMd |d |d d g|d d}|||}nd|vrb|j j j j|d|d d |d } ||}n!|j j j j j di|}||j|j}n)#t*$r} || | d} ~ wwxYw ||} || | S#t*$r0} || || d} ~ wwxYw)aInvoke 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. 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_namerfintermediate_stepsr#usercontentmessage_metadatarolerrthread_metadatamessagesrr"rrrNrr()rr configuregeton_chain_startrget_namer__parse_intermediate_stepsrZrtthreadsrunssubmit_tool_outputs_create_thread_and_runrrv _create_run _wait_for_runrwr# BaseExceptionon_chain_error _get_response on_chain_endrQ rdrzr{rlcallback_manager run_manager tool_outputsrunthread_rCresponses r+invokezOpenAIAssistantRunnable.invokes6v&&*4"(**["9"9#ZZ//!'J!7!7    '55 $KKVZZ %;%;%Nt}}6  $ } Q+?!@!@ Q#==./   Hdk&.3GWW,WWE))%+',Y'7(- 2D(E(E!!& *; < <  11%@@&&K$,5<<+&!),"YY'9:: = &&u--Hdk&.3GPP%PP$$SVS];;CC     & &q ) ) )G  ))#..H  $ $X . . .O      & &q388:: & > > >G s1-EG?? H% H  H%)I J+J  J)r[.s!EEEt,T22EEEr.rsN)r\r[r()rIrtrurvrw) r*rfrgrhrjr[rl openai_toolsrxs r+acreate_assistantz)OpenAIAssistantRunnable.acreate_assistantis4$A'?'A'A EEuEEE &+6==% >        sR <RR6RRRr.cfK|pi}tj|d|d|d}|t |||dp|} |jr^|drI||dd{V}|jj j j j di|d{V}nd |vrSd |d |d d g|dd}| ||d{V}nd|vrn|jj j j|d |d d |d d{V} ||d{V}n'|jj j j j di|d{V}||j|jd{V}n)#t($r} || | d} ~ wwxYw ||} || | S#t($r0} || || d} ~ wwxYw)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: Additional run instructions. model: Override Assistant model for this run. tools: Override Assistant tools 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]]. rrrrrrrNr#rrrrrrr"rrr()rrrrrrr__aparse_intermediate_stepsr[rtrrr_acreate_thread_and_runrrv _acreate_run_await_for_runrwr#rrrrrQrs r+ainvokezOpenAIAssistantRunnable.ainvokes82*4"(**["9"9#ZZ//!'J!7!7    '55 $KKVZZ %;%;%Nt}}6  ( }" +?!@!@" %)%D%D./&&       TD-2:?S"E))%+',Y'7(- 2D(E(E!!& *; < <  !88GGGGGGGG&&+08AHH+&!),"YY'9:: I !--e44444444TD-2:?S++CFCMBBBBBBBBCC     & &q ) ) )G  ))#..H  $ $X . . .O      & &q388:: & > > >G s1$E;H I*II I66 J0+J++J0r'List[Tuple[OpenAIAssistantAction, str]]c|d\}}||j|j}t|jrd|jjjDfd|D}||j|jd}|S)Nch|] }|j Sr(rwrqtcs r+ zDOpenAIAssistantRunnable._parse_intermediate_steps..'&&&&&&r.cVg|]%\}}|jvt||jd&S)outputr9r9r!rqactionrrequired_tool_call_idss r+rrzEOpenAIAssistantRunnable._parse_intermediate_steps..G   "&<<<6{{F4G H H<<z7OpenAIAssistantRunnable._create_run..4   1FFF qFFFr.r#r\)itemsrZrtrrrvr\rdrzparamss r+rz#OpenAIAssistantRunnable._create_runsn       4t{',3 +   *    r.rcd|D}|jjjjd|j|d|}|S)Nc"i|] \}}|dv || Srr(rs r+rzBOpenAIAssistantRunnable._create_thread_and_run..rr.r\rr()rrZrtrcreate_and_runr\rdrzrrrs r+rz.OpenAIAssistantRunnable._create_thread_and_runsj       6dk&5 *      r.rc  jdkr3ddl t jjdd}t jjdd}|dkp |dko|dk|jjjj j d}fd|D}|j s|Sd |D}t fd |Drd d |D}t|j jd djj Sjdkr|j sjjjSg}jjjD]}|j} t+j| jd} n2#t0$r%} t3d| jd| j| d} ~ wwxYwt7| dkr d| vr| d} |t;| j| |jdjj |St+jd} t3djd| d)N completedr.ascorderc4g|]}|jjk|Sr(r"rwrqmsgrs r+rrz9OpenAIAssistantRunnable._get_response.."'LLLCszSV7K7KC7K7K7Kr.c&g|]}|jD]}|Sr(rrqr msg_contents r+rrz9OpenAIAssistantRunnable._get_response..%; #s{8C r.c3K|]N}r$t|jjjjn#t|jjjjVOdSNrPtypesrtrTextContentBlockMessageContentTextrqrr>version_gte_1_14s r+ z8OpenAIAssistantRunnable._get_response..(u   (Jw (9(A(RSSS#!2!:!M      r. c3.K|]}|jjVdSrtextvaluerqrs r+rz8OpenAIAssistantRunnable._get_response..2'"L"L'7<#5"L"L"L"L"L"Lr.rr#r" return_valueslogr"r#requires_actionFstrict*Received invalid JSON function arguments:  for function __arg1rJ tool_inputr9rr"r#indentUnexpected run status: . Full run info: )) statusr>intversionVERSIONsplitrZrtrrlistr#r_alljoinr rwrrrfunctionjsonloads argumentsr ValueErrorrflenappendr8dumpsrQrdr major_version minor_versionr new_messagesansweractions tool_callrargsrCrun_infor>rs ` @@r+rz%OpenAIAssistantRunnable._get_responses( : $ $ MMM 6 < HMLLL8LLLL= $##'3F      &      M"L"LV"L"L"LLL($!$!f v-    Z, , ,= J*>IIG 0DO   $-:h&8GGGDD&$M#-MM=E]MM t99>>h$&6&6>D)%]#'%.\"v"%-     Nz#((**Q777HU#*UU(UUU sF G) G  Gr"r#cd}|rO|jjjj||}|jdv}|rt |jdz |O|SNT)r#) in_progressqueuedi)rZrtrrretrieverrr^rdr"r#r1rs r+rz%OpenAIAssistantRunnable._wait_for_run]sn  2+"*/8898UUC*(AAK 2d)D0111  2  r.c K|d\}}||j|jd{V}t|jrd|jjjDfd|D}||j|jd}|S)Nrch|] }|j Sr(rrs r+rzEOpenAIAssistantRunnable._aparse_intermediate_steps..mrr.cVg|]%\}}|jvt||jd&Srrrs r+rrzFOpenAIAssistantRunnable._aparse_intermediate_steps..prr.rrrs @r+rz2OpenAIAssistantRunnable._aparse_intermediate_stepsfs$6b#9 [&&{'9;;PQQQQQQQQ!$   && # 3 G R&&& "    "4   )!($.   #"r.cKd|D}|jjjjj|dfd|ji|d{VS)Nc"i|] \}}|dv || Srr(rs r+rz8OpenAIAssistantRunnable._acreate_run..}rr.r#r\)rr[rtrrrvr\rs r+rz$OpenAIAssistantRunnable._acreate_run|s       @T&+38? +   *          r.cKd|D}|jjjjd|j|d|d{V}|S)Nc"i|] \}}|dv || Srr(rs r+rzCOpenAIAssistantRunnable._acreate_thread_and_run..rr.rr()rr[rtrrr\rs r+rz/OpenAIAssistantRunnable._acreate_thread_and_runs       BD%*2A *            r.c  Kjdkr9ddl t jjdd}t jjdd}|dkp |dko|dk|jjjj j dd{V}fd|D}|j s|Sd |D}t fd |Drd d |D}t|j jd djj Sjdkr|j sjjjSg}jjjD]}|j} t+j| jd} n2#t0$r%} t3d| jd| j| d} ~ wwxYwt7| dkr d| vr| d} |t;| j| |jdjj |St+jd} t3djd| d)Nrrrrrrrc4g|]}|jjk|Sr(rrs r+rrz:OpenAIAssistantRunnable._aget_response..rr.c&g|]}|jD]}|Sr(rrs r+rrz:OpenAIAssistantRunnable._aget_response..rr.c3K|]N}r$t|jjjjn#t|jjjjVOdSrrrs r+rz9OpenAIAssistantRunnable._aget_response..rr.rc3.K|]}|jjVdSrrrs r+rz9OpenAIAssistantRunnable._aget_response..rr.rrrrFr r r r rrrrrr) rr>rrrrr[rtrrrr#r_rrr rwrrrrrr r!rr"rfr#r$r8r%rQr&s ` @@r+_aget_responsez&OpenAIAssistantRunnable._aget_responsesL : $ $ MMM 6 < IIG 0DO   $-:h&8GGGDD&$M#-MM=E]MM t99>>h$&6&6>D)%]#'%.\"v"%-     Nz#((**Q777HU#*UU(UUU s F'' G1 GGcKd}|r`|jjjj||d{V}|jdv}|r"t j|jdz d{V|`|Sr0) r[rtrrr3rasynciorr^r4s r+rz&OpenAIAssistantRunnable._await_for_runs  @).6;DD)EC*(AAK @mD$7$$>?????????  @ r.)r$r)rfr!rgr!rhrirjr!rZrkrlrr$rXr)rzrQr{r|rlrr$r})rfr!rgr!rhrirjr!r[rrlrr$rX)rrr$rQ)rzrQr$r)rzrQrrQr$r)rrr$r)r"r!r#r!r$r)r1r2r3r4rrDrZr5r[r^r_rrer6ryrrrrrrrrrrrrArr(r.r+rXrXsHKKZ%(:;;;F;;;;'L-#N####;HF_'"""#"FJGGGGG[GD?CPPPPPd !S!S!S!S!S[!SH?CUUUUUn####,         HHHHT####,         HHHHT      r.rX)r$r;)r$rF)rJrKr$r%)rJrKr$rT)4 __future__rrCrrtimertypingrrrr r r r r rrlangchain_core.agentsrrlangchain_core.callbacksrlangchain_core.loadrlangchain_core.runnablesrrrlangchain_core.toolsr%langchain_core.utils.function_callingrpydanticrrrtyping_extensionsrr>openai.types.beta.threadsrrQs""""""                         ;:::::::444444%%%%%%XXXXXXXXXX))))))HHHHHH6666666666""""""MMM777777 K*K.             , , , ,  )*, U U U U U 243CDU U U U U r.