U hm@sddlmZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZddlmZdd lmZmZmZdd lmZmZm Z dd l!m"Z"dd l#m$Z$erddl%Z%dd l&m'Z'ddl(m)Z)GdddeZ*GdddeZ+ddddZ,ddddZ-dddddZ.dddd d!Z/ee e+e*e d"e d#fZ0Gd$d%d%ee e0fZ1dS)&) annotationsN)JSONDecodeError)sleep) TYPE_CHECKINGAnyCallableDictListOptionalSequenceTupleTypeUnion) AgentAction AgentFinish)CallbackManager)dumpd) BaseModelFieldroot_validator)RunnableConfigRunnableSerializable ensure_config)BaseTool)convert_to_openai_tool) ThreadMessage)RequiredActionFunctionToolCallc@s4eZdZUdZded<ded<eddddZd S) OpenAIAssistantFinishzuAgentFinish with run and thread metadata. Parameters: run_id: Run id. thread_id: Thread id. strrun_id thread_idboolreturncCsdSz]Check if the class is serializable by LangChain. Returns: False Fclsr%r%J/tmp/pip-unpacked-wheel-bo69hh5q/langchain/agents/openai_assistant/base.pyis_lc_serializable/sz(OpenAIAssistantFinish.is_lc_serializableN__name__ __module__ __qualname____doc____annotations__ classmethodr)r%r%r%r(r$s rc@s<eZdZUdZded<ded<ded<edddd 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_idrr r!r"cCsdSr$r%r&r%r%r(r)Fsz(OpenAIAssistantAction.is_lc_serializableNr*r%r%r%r(r19s r1z openai.OpenAIr"c Csnzddl}|WStk r>}ztd|W5d}~XYn,tk rh}ztd|W5d}~XYnXdSNrzBUnable 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 ImportErrorAttributeErrorr4er%r%r(_get_openai_clientPs r9zopenai.AsyncOpenAIc Csnzddl}|WStk r>}ztd|W5d}~XYn,tk rh}ztd|W5d}~XYnXdSr3)r4 AsyncOpenAIr5r6r7r%r%r(_get_openai_async_client`s r;z:Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]r!)toolr#cCs"d}t|to d|ko |d|kS)zd?Zdddd5d@dAZdS)DOpenAIAssistantRunnablea 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_agentT)preZskip_on_failurer?)valuesr#cCs2|ddkr.ddl}|dj}|j|d|d<|S)NrErrD)api_key)r4rLr:)r'rKr4rLr%r%r(validate_async_clients   z-OpenAIAssistantRunnable.validate_async_client)rDzSequence[Union[BaseTool, dict]]z2Optional[Union[openai.OpenAI, openai.AzureOpenAI]])name instructionstoolsmodelrDkwargsr#cKs@|pt}|jjj||dd|D|d}|f|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. cSsg|] }t|qSr%rA.0r<r%r%r( sz.rNrOrPrQ)rFrD)r9beta assistantscreateid)r'rNrOrPrQrDrR assistantr%r%r(create_assistants  z(OpenAIAssistantRunnable.create_assistantzOptional[RunnableConfig] OutputType)inputconfigr#c Cst|}tj|d|d|dd}|jt|||dd}z|jrz|drz||d}|jj j j j f|}nd|krd |d |d d g|d d}| ||}nPd|kr|jj j jj|d|d d |d d}||}n|jj j j j f|}||j|j}Wn2tk rH} z|| | W5d} ~ XYnXz||} Wn:tk r} z|j| |d| W5d} ~ XYnX|| | SdS)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]]. callbackstagsmetadataZinheritable_callbacksZinheritable_tagsZinheritable_metadatarun_namerNintermediate_stepsr usercontentmessage_metadatarolerircthread_metadatamessagesrcrrirlrcNrc)rr configuregeton_chain_startrrI_parse_intermediate_stepsrDrXthreadsrunssubmit_tool_outputs_create_thread_and_runrorZ _create_run _wait_for_runr[r BaseExceptionon_chain_error _get_responser? on_chain_end) selfr_r`callback_manager run_manager tool_outputsrunthread_r8responser%r%r(invokes\     zOpenAIAssistantRunnable.invoke)rEz.rWN)rFrE)r;rXrYrZr[) r'rNrOrPrQrErRZ openai_toolsr\r%r%r(acreate_assistanths  z)OpenAIAssistantRunnable.acreate_assistant)r_r`rRr#c s|pi}tj|d|d|dd}|jt|||dd}z|jr|dr||dIdH}|jjj j j f|IdH}nd |krd |d |d d g|dd}| ||IdH}ndd|kr|jjj j j|d |d d |d dIdH} ||IdH}n|jjj j j f|IdH}||j|jIdH}Wn2tk rt} z|| | W5d} ~ XYnXz||} Wn:tk r} z|j| |d| W5d} ~ XYnX|| | SdS)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]]. rarbrcrdrerfrgNr rhrirjrkrmrnrrprq)rrrrsrtrrI_aparse_intermediate_stepsrErXrvrwrx_acreate_thread_and_runrorZ _acreate_run_await_for_runr[r r|r}r~r?r) rr_r`rRrrrrrrr8rr%r%r(ainvokesd         zOpenAIAssistantRunnable.ainvokez'List[Tuple[OpenAIAssistantAction, str]])rgr#csb|d\}}||j|j}t|jrszDOpenAIAssistantRunnable._parse_intermediate_steps..cs*g|]"\}}|jkrt||jdqS)outputr2r2rrUactionrZrequired_tool_call_idsr%r(rVs zEOpenAIAssistantRunnable._parse_intermediate_steps..rrr r{rr setrequired_actionrx tool_callsrrgZ last_actionZ last_outputrrrxr%rr(rus  z1OpenAIAssistantRunnable._parse_intermediate_steps)r_r#cCs6dd|D}|jjjjj|dfd|ji|S)NcSsi|]\}}|dkr||qS)rOrQrPZ run_metadatar%rUkvr%r%r( sz7OpenAIAssistantRunnable._create_run..r rF)itemsrDrXrvrwrZrFrr_paramsr%r%r(rzs z#OpenAIAssistantRunnable._create_run)r_rr#cCs4dd|D}|jjjjf|j|d|}|S)NcSsi|]\}}|dkr||qSrr%rr%r%r(rszBOpenAIAssistantRunnable._create_thread_and_run..rFr)rrDrXrvcreate_and_runrFrr_rrrr%r%r(rys z.OpenAIAssistantRunnable._create_thread_and_run)rr#c sjdkrddltjjdd}tjjdd}|dkpT|dkoT|dk|jjjj j j dd}fdd |D}|j s|Sd d |D}t fd d |Drd dd |D}t|j jddjj dSjdkr|j sjjjSg}jjjD]}|j} ztj| jdd} Wn>tk rl} ztd| jd| j| W5d} ~ XYnXt| dkrd| kr| d} |t| j| |jdjj dq|Stjdd} tdjd| ddS)N completedr.ascordercsg|]}|jjkr|qSr%rr[rUmsgrr%r(rV!s z9OpenAIAssistantRunnable._get_response..cSsg|]}|jD]}|qqSr%rirUrZ msg_contentr%r%r(rV$sc3s6|].}rt|jjjjnt|jjjjVqdSNr>typesrXrvZTextContentBlockZMessageContentTextrUrir4version_gte_1_14r%r( 's z8OpenAIAssistantRunnable._get_response.. css|]}|jjVqdSrtextvaluerr%r%r(r1srr rZ return_valueslogrr requires_actionFstrict*Received invalid JSON function arguments:  for function __arg1r<Z tool_inputr2rrr indentUnexpected run status: . Full run info: )) statusr4intversionVERSIONsplitrDrXrvrolistr rIalljoinrr[rrxrfunctionjsonloads argumentsr ValueErrorrNlenappendr1dumpsr? rr major_version minor_versionroZ new_messagesZansweractionsZ tool_callrargsr8Zrun_infor%r4rrr(r~sz      z%OpenAIAssistantRunnable._get_response)rr r#cCs@d}|r<|jjjjj||d}|jdk}|rt|jdq|SNT)r ) in_progressZqueuedi)rDrXrvrwretrieverrrHrrr rrr%r%r(r{\s z%OpenAIAssistantRunnable._wait_for_runcsh|d\}}||j|jIdH}t|jrBdd|jjjDfdd|D}||j|jd}|S)NrcSsh|] }|jqSr%rrr%r%r(rlszEOpenAIAssistantRunnable._aparse_intermediate_steps..cs*g|]"\}}|jkrt||jdqSrrrrr%r(rVos zFOpenAIAssistantRunnable._aparse_intermediate_steps..rrrr%rr(res  z2OpenAIAssistantRunnable._aparse_intermediate_stepscs<dd|D}|jjjjj|dfd|ji|IdHS)NcSsi|]\}}|dkr||qSrr%rr%r%r(r|sz8OpenAIAssistantRunnable._acreate_run..r rF)rrErXrvrwrZrFrr%r%r(r{s z$OpenAIAssistantRunnable._acreate_runcs:dd|D}|jjjjf|j|d|IdH}|S)NcSsi|]\}}|dkr||qSrr%rr%r%r(rszCOpenAIAssistantRunnable._acreate_thread_and_run..r)rrErXrvrrFrr%r%r(rs  z/OpenAIAssistantRunnable._acreate_thread_and_runc sjdkrddltjjdd}tjjdd}|dkpT|dkoT|dk|jjjj j j ddIdH}fdd |D}|j s|Sd d |D}t fd d |Drd dd |D}t|j jddjj dSjdkr|j sjjjSg}jjjD]}|j} ztj| jdd} Wn>tk rr} ztd| jd| j| W5d} ~ XYnXt| dkrd| kr| d} |t| j| |jdjj dq|Stjdd} tdjd| ddS)Nrrrrrrrcsg|]}|jjkr|qSr%rrrr%r(rVs z:OpenAIAssistantRunnable._aget_response..cSsg|]}|jD]}|qqSr%rrr%r%r(rVsc3s6|].}rt|jjjjnt|jjjjVqdSrrrrr%r(rs z9OpenAIAssistantRunnable._aget_response..rcss|]}|jjVqdSrrrr%r%r(rsrrrrFrrrrrrrrrr) rr4rrrrrErXrvrorr rIrrrr[rrxrrrrrrrrNrrr1rr?rr%rr(_aget_responsesz       z&OpenAIAssistantRunnable._aget_responsecsNd}|rJ|jjjjj||dIdH}|jdk}|rt|jdIdHq|Sr) rErXrvrwrrasynciorrHrr%r%r(rs   z&OpenAIAssistantRunnable._await_for_run)N)N)r+r,r-r.rr9rDr/rErHrIrrMr0r]rrrrurzryr~r{rrrrrr%r%r%r(rBs8 M     "R  $W  J   JrB)2 __future__rrrrtimertypingrrrrr r r r r rZlangchain_core.agentsrrZlangchain_core.callbacksrZlangchain_core.loadrZlangchain_core.pydantic_v1rrrZlangchain_core.runnablesrrrZlangchain_core.toolsrZ%langchain_core.utils.function_callingrr4Zopenai.types.beta.threadsrZs<   0