o Zhv@s|ddlmZddlZddlZddlmZddlmZddlmZm 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&er~ddl'Z'ddl(m)Z)ddl*m+Z+GdddeZ,GdddeZ-d'ddZ.d(ddZ/d)ddZ0d*d!d"Z1ee e-e,e d#e d$fZ2Gd%d&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@s0eZdZUdZded<ded<ed ddZd S) OpenAIAssistantFinishzuAgentFinish with run and thread metadata. Parameters: run_id: Run id. thread_id: Thread id. strrun_id thread_idreturnboolcCdSz]Check if the class is serializable by LangChain. Returns: False Fclsr&r&]/var/www/html/lang_env/lib/python3.10/site-packages/langchain/agents/openai_assistant/base.pyis_lc_serializable0z(OpenAIAssistantFinish.is_lc_serializableNr"r#__name__ __module__ __qualname____doc____annotations__ classmethodr*r&r&r&r)r%s rc@s8eZdZUdZded<ded<ded<ed dd Zd 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#cCr$r%r&r'r&r&r)r*Gr+z(OpenAIAssistantAction.is_lc_serializableNr,r-r&r&r&r)r4:s r4r" openai.OpenAIc CVz ddl}|WSty}ztd|d}~wty*}ztd|d}~wwNrzBUnable 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"`.)openaiZOpenAI ImportErrorAttributeErrorr9er&r&r)_get_openai_clientQ$ r>openai.AsyncOpenAIc Cr7r8)r9 AsyncOpenAIr:r;r<r&r&r)_get_openai_async_clientar?rBtool:Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]r#cCs"d}t|tod|vo|d|vS)zd?ZdMd@dAZdNdBdCZdOdDdEZdS)POpenAIAssistantRunnablea 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"rcCs,|jdurddl}|jj}|j|d|_|S)Nr)api_key)rNr9rMrUrA)selfr9rUr&r&r)validate_async_clients z-OpenAIAssistantRunnable.validate_async_client)rMname instructionstoolsSequence[Union[BaseTool, dict]]model2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]kwargscKs@|pt}|jjj||dd|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. cSg|]}t|qSr&rJ.0rCr&r&r) z.rXrYrZr\)rOrMNr&)r>beta assistantscreateid)r(rXrYrZr\rMr^ assistantr&r&r)create_assistants  z(OpenAIAssistantRunnable.create_assistantinputrGconfigOptional[RunnableConfig] OutputTypec Kst|}tj|d|d|dd}|jt|||dp#|d}zq|jrC|drC||d}|j j j j j di|}nKd|vrdd |d |d |d d g|dd}|||}n*d|vr|j j j jj|d|d d |d d} ||}n |j j j j j di|}||j|j}Wnty} z|| | d} ~ wwz||} Wnty} z |j| |d| d} ~ ww|| | S)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. 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. attachments: A list of files attached to the message, and the tools they should be added to. 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]]. callbackstagsmetadataZinheritable_callbacksZinheritable_tagsZinheritable_metadatarun_namerXintermediate_stepsr!usercontentmessage_metadata attachments)rolerxrrrzthread_metadatamessagesrrr rxr{rrNrrr&)rr configuregeton_chain_startrget_namerR_parse_intermediate_stepsrMrfthreadsrunssubmit_tool_outputs_create_thread_and_runr~rh _create_run _wait_for_runrir! BaseExceptionon_chain_error _get_responserG on_chain_end rVrlrmr^Zcallback_managerZ run_manager tool_outputsrunthread_r=responser&r&r)invokesb#     zOpenAIAssistantRunnable.invoke)rN.reN)rOrNr&)rBrfrgrhri) r(rXrYrZr\rNr^Z openai_toolsrjr&r&r)acreate_assistantrs  z)OpenAIAssistantRunnable.acreate_assistantc s|pi}tj|d|d|dd}|jt|||dp$|d}z|jrJ|drJ||dIdH}|jj j j j di|IdH}nSd |vrjd |d |d d g|dd}| ||IdH}n3d|vr|jj j jj|d |d d |d dIdH} ||IdH}n|jj j j j di|IdH}||j|jIdH}Wnty} z|| | d} ~ wwz||} Wnty} z |j| |d| d} ~ ww|| | S)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]]. rprqrrrsrtrurvNr!rwrxry)r{rxrrr|r}r rrr&)rrrrrrrR_aparse_intermediate_stepsrNrfrrr_acreate_thread_and_runr~rh _acreate_run_await_for_runrir!rrrrGrrr&r&r)ainvokesj#       zOpenAIAssistantRunnable.ainvokerv'List[Tuple[OpenAIAssistantAction, str]]csb|d\}}||j|j}t|jrdd|jjjDfdd|D}||j|jd}|S)NcSh|]}|jqSr&rirbZtcr&r&r) zDOpenAIAssistantRunnable._parse_intermediate_steps..c*g|]\}}|jvrt||jdqS)outputr5r5rrbactionrZrequired_tool_call_idsr&r)rc  zEOpenAIAssistantRunnable._parse_intermediate_steps..rr r!rr r!setrequired_actionr tool_callsrVrvZ last_actionZ last_outputrrrr&rr)rs  z1OpenAIAssistantRunnable._parse_intermediate_stepscCs6dd|D}|jjjjj|dfd|ji|S)NcSi|] \}}|dvr||qS) rYr\rZZadditional_instructionsparallel_tool_callstop_p temperaturemax_completion_tokensmax_prompt_tokens run_metadatar&rbkvr&r&r) z7OpenAIAssistantRunnable._create_run..r!rO)itemsrMrfrrrhrOrVrlparamsr&r&r)r s z#OpenAIAssistantRunnable._create_runrcCs4dd|D}|jjjjd|j|d|}|S)NcSr) rYr\rZrrrrrrr&rr&r&r)r#rzBOpenAIAssistantRunnable._create_thread_and_run..rOrr&)rrMrfrcreate_and_runrOrVrlrrrr&r&r)r"s z.OpenAIAssistantRunnable._create_thread_and_runrc sjdkrrddltjjdd}tjjdd}|dkp*|dko*|dk|jjjj j j dd}fdd |D}|j sE|Sd d |D}t fd d |Drbd dd |D}t|j jddjj dSjdkr|j sjjjSg}jjjD]H}|j} z tj| jdd} Wnty} z td| jd| j| d} ~ wwt| dkrd| vr| d} |t| j| |jdjj dq|Stjdd} tdjd| d)N completedr.ascordercg|] }|jjkr|qSr&r rirbmsgrr&r)rcIz9OpenAIAssistantRunnable._get_response..cSg|] }|jD]}|qqSr&rxrbrZ msg_contentr&r&r)rcL c38|]}rt|jjjjnt|jjjjVqdSNrFtypesrfrZTextContentBlockZMessageContentTextrbrxr9version_gte_1_14r&r) O   z8OpenAIAssistantRunnable._get_response.. cs|]}|jjVqdSrtextvaluerr&r&r)rZrr!r Z return_valueslogr r!requires_actionFstrict*Received invalid JSON function arguments:  for function __arg1rCZ tool_inputr5rr r!indentUnexpected run status: . Full run info: )) statusr9intversionVERSIONsplitrMrfrr~listr!rRalljoinrrirrrfunctionjsonloads argumentsr ValueErrorrXlenappendr4dumpsrG rVr major_version minor_versionr~Z new_messagesZansweractionsZ tool_callrargsr=Zrun_infor&r9rrr)r:s       z%OpenAIAssistantRunnable._get_responser r!cCsBd}|r|jjjjj||d}|jdv}|rt|jd|s|SNT)r!) in_progressZqueuedi)rMrfrrretrieverrrQrVr r!rrr&r&r)rs z%OpenAIAssistantRunnable._wait_for_runcsd|d\}}||j|j}t|jrdd|jjjDfdd|D}||j|jd}|S)NrcSrr&rrr&r&r)rrzEOpenAIAssistantRunnable._aparse_intermediate_steps..crrrrrr&r)rcrzFOpenAIAssistantRunnable._aparse_intermediate_steps..rrrr&rr)rs   z2OpenAIAssistantRunnable._aparse_intermediate_stepscs>dd|D}|jjjjj|dfd|ji|IdHS)NcSrrr&rr&r&r)rrz8OpenAIAssistantRunnable._acreate_run..r!rO)rrNrfrrrhrOrr&r&r)rs  z$OpenAIAssistantRunnable._acreate_runcs<dd|D}|jjjjd|j|d|IdH}|S)NcSrrr&rr&r&r)rrzCOpenAIAssistantRunnable._acreate_thread_and_run..rr&)rrNrfrrrOrr&r&r)rs  z/OpenAIAssistantRunnable._acreate_thread_and_runc sjdkrvddltjjdd}tjjdd}|dkp+|dko+|dk|jjjj j j ddIdH}fdd |D}|j sI|Sd d |D}t fd d |Drfd dd |D}t|j jddjj dSjdkr|j sjjjSg}jjjD]H}|j} z tj| jdd} Wnty} z td| jd| j| d} ~ wwt| dkrd| vr| d} |t| j| |jdjj dq|Stjdd} tdjd| d)Nrrrrrrrcrr&rrrr&r)rcrz:OpenAIAssistantRunnable._aget_response..cSrr&rrr&r&r)rcrc3rrrrrr&r)rrz9OpenAIAssistantRunnable._aget_response..rcsrrrrr&r&r)rrrrrrFrrrrrrrrrr) rr9rrrrrNrfrr~rr!rRrrrrirrrrrrrrrrXrrr4rrGrr&r r)_aget_responses        z&OpenAIAssistantRunnable._aget_responsecsRd}|r'|jjjjj||dIdH}|jdv}|r%t|jdIdH|s|Sr) rNrfrrrrasynciorrQrr&r&r)rs  z&OpenAIAssistantRunnable._await_for_run)r"r)rXrrYrrZr[r\rrMr]r^rr"rKr)rlrGrmrnr^rr"ro)rXrrYrrZr[r\rrNrr^rr"rK)rvrr"rG)rlrGr"r)rlrGrrGr"r)rrr"r)r rr!rr"r)r.r/r0r1rr>rMr2rNrQrRrrWr3rkrrrrrrrrrrrrrr&r&r&r)rKsD M    " [ $ ^    K   KrK)r"r6)r"r@)rCrDr"r#)rCrDr"rI)4 __future__rrrrtimertypingrrrrr r r r r rZlangchain_core.agentsrrZlangchain_core.callbacksrZlangchain_core.loadrZlangchain_core.runnablesrrrZlangchain_core.toolsrZ%langchain_core.utils.function_callingrZpydanticrrrZtyping_extensionsrr9Zopenai.types.beta.threadsrZrBrHrJrorKr&r&r&r)s@   0