o Zh|_@s ddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZddlmZmZddlmZdd lmZdd lmZmZmZdd lm Z erjdd l!Z!dd l"m#Z#ddl$m%Z&d$ddZ'd%ddZ(d&ddZ)d'ddZ*d(d d!Z+eGd"d#d#e Z,d S))) annotations) TYPE_CHECKINGAnyCallableDictOptionalSequenceTypeUnion)OpenAIAssistantRunnable OutputType)beta)CallbackManager)dumpd)RunnableConfig ensure_config)BaseTool)convert_to_openai_tool) BaseModelFieldmodel_validator)SelfN)NotGiven) ToolResourcesreturn openai.OpenAIc C^z ddl}|jddidWSty}ztd|d}~wty.}ztd|d}~ww)zGet the OpenAI client. Returns: openai.OpenAI: OpenAI client Raises: ImportError: If `openai` is not installed. AttributeError: If the installed `openai` version is not compatible. rN OpenAI-Beta assistants=v2default_headersBUnable to import openai, please install with `pip install openai`.wPlease make sure you are using a v1.23-compatible version of openai. You can install with `pip install "openai>=1.23"`.)openaiZOpenAI ImportErrorAttributeErrorr#er(g/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/agents/openai_assistant/base.py_get_openai_client$ r*openai.AsyncOpenAIc Cr)zGet the async OpenAI client. Returns: openai.AsyncOpenAI: Async OpenAI client Raises: ImportError: If `openai` is not installed. AttributeError: If the installed `openai` version is not compatible. rNrrrr!r")r# AsyncOpenAIr$r%r&r(r(r)_get_openai_async_client7r+r.file_idslistcCs.g}|D]}||ddiddigdq|S)aConvert file_ids into attachments File search and Code interpreter will be turned on by default. Args: file_ids (list): List of file_ids that need to be converted into attachments. Returns: list: List of attachments converted from file_ids. type file_searchcode_interpreter)Zfile_idtools)append)r/ attachmentsidr(r(r)"_convert_file_ids_into_attachmentsPs r8tool:Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]boolcCs"d}t|tod|vo|d|vS)aMDetermine if tool corresponds to OpenAI Assistants built-in. Args: tool (Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]): Tool that needs to be determined. Returns: A boolean response of true or false indicating if the tool corresponds to OpenAI Assistants built-in. )r3Z retrievalr2r1) isinstancedict)r9Zassistants_builtin_toolsr(r(r)_is_assistants_builtin_tooles   r>Dict[str, Any]cCst|r|St|S)aConvert a raw function/class to an OpenAI tool. Note that OpenAI assistants supports several built-in tools, such as "code_interpreter" and "retrieval." Args: tool (Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]): Tools or functions that need to be converted to OpenAI tools. Returns: Dict[str, Any]: A dictionary of tools that are converted into OpenAI tools. )r>r)r9r(r(r)_get_assistants_toolzsr@c@seZdZUdZeedZded< dZded< ded< d Z d ed < d Z d ed< e dddd?d+d,Ze ddd-d@d/d0Z d>d?d1d2ZdAd3d4ZdBd6d7ZdAd8d9ZdBd:d;ZdS)COpenAIAssistantV2Runnablea Run an OpenAI Assistant. Attributes: client (Any): OpenAI or AzureOpenAI client. async_client (Any): Async OpenAI or AzureOpenAI client. assistant_id (str): OpenAI assistant ID. check_every_ms (float): Frequency to check progress in milliseconds. as_agent (bool): Whether to use the assistant as a LangChain agent. Example using OpenAI tools: .. code-block:: python from langchain.agents.openai_assistant import OpenAIAssistantV2Runnable assistant = OpenAIAssistantV2Runnable.create_assistant( name="math 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 = assistant.invoke({"content": "What's 10 - 4 raised to the 2.7"}) Example using custom tools and AgentExecutor: .. code-block:: python from langchain.agents.openai_assistant import OpenAIAssistantV2Runnable from langchain.agents import AgentExecutor from langchain.tools import E2BDataAnalysisTool tools = [E2BDataAnalysisTool(api_key="...")] agent = OpenAIAssistantV2Runnable.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": "Analyze the data..."}) Example using custom tools and custom execution: .. code-block:: python from langchain.agents.openai_assistant import OpenAIAssistantV2Runnable from langchain.agents import AgentExecutor from langchain_core.agents import AgentFinish from langchain.tools import E2BDataAnalysisTool tools = [E2BDataAnalysisTool(api_key="...")] agent = OpenAIAssistantV2Runnable.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_clientstr assistant_idg@@floatcheck_every_msFr;as_agentafter)moderrcCs,|jdurddl}|jj}|j|d|_|S)z?Validate that the async client is set, otherwise initialize it.Nr)api_key)rDr#rCrLr-)selfr#rLr(r(r)validate_async_clients z/OpenAIAssistantV2Runnable.validate_async_client) model_kwargsrCtool_resources extra_bodyname instructionsr4Sequence[Union[BaseTool, dict]]modelrOdict[str, float]2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]rP7Optional[Union[AssistantToolResources, dict, NotGiven]]rQOptional[object]kwargsr c Ksd|pt}|durddlm} | }|jjjd||dd|D|||d|} |d| j|d| S) aCreate an OpenAI Assistant and instantiate the Runnable. Args: name (str): Assistant name. instructions (str): Assistant instructions. tools (Sequence[Union[BaseTool, dict]]): Assistant tools. Can be passed in OpenAI format or as BaseTools. tool_resources (Optional[Union[AssistantToolResources, dict, NotGiven]]): Assistant tool resources. Can be passed in OpenAI format. model (str): Assistant model to use. client (Optional[Union[openai.OpenAI, openai.AzureOpenAI]]): OpenAI or AzureOpenAI client. Will create default OpenAI client (Assistant v2) if not specified. model_kwargs: Additional model arguments. Only available for temperature and top_p parameters. extra_body: Additional body parameters to be passed to the assistant. Returns: OpenAIAssistantRunnable: The configured assistant runnable. Nr NOT_GIVENcSg|]}t|qSr(r@.0r9r(r(r) #z>OpenAIAssistantV2Runnable.create_assistant..)rRrSr4rPrUrQ)rFrCr()r* openai._typesr\r assistantscreater7) clsrRrSr4rUrOrCrPrQrZr\ assistantr(r(r)create_assistants "    z*OpenAIAssistantV2Runnable.create_assistantinputr=configOptional[RunnableConfig]r c Kst|}tj|d|d|dd}|jt|||dp#|d}t|dg}|dg|}zo|jrS|d rS| |d }|j j j j jdi|} nId |vrqd |d ||d dg|dd} ||| } n+d|vr|j j j jj|d |d d ||d d} ||} n |j j j j jdi|} || j| j} Wnty} z|| | d} ~ wwz|| } Wnty} z |j| | d| d} ~ ww|| | S)a(Invoke the assistant. Args: input (dict): 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. file_ids: (deprecated) File ids to include in new run. Use 'attachments' instead attachments: Assistant files to include in new run. (v2 API). 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. tool_resources: Override Assistant tool resources for this run (v2 API). run_metadata: Metadata to associate with new run. config (Optional[RunnableConfig]): Configuration for the run. Returns: OutputType: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. Raises: BaseException: If an error occurs during the invocation. callbackstagsmetadataZinheritable_callbacksZinheritable_tagsZinheritable_metadatarun_namerRr/r6intermediate_steps thread_idusercontentmessage_metadatarolerur6rnthread_metadatamessagesrnrun_idrurxr6rnNrnr()rr configuregeton_chain_startrget_namer8rI_parse_intermediate_stepsrCr threadsrunssubmit_tool_outputs_create_thread_and_runr{re _create_runZ _wait_for_runr7rs BaseExceptionon_chain_error _get_responser= on_chain_endrMrirjrZZcallback_managerZ run_managerfilesr6Z tool_outputsrunthread_r'responser(r(r)invoke+sh!     z OpenAIAssistantV2Runnable.invoke)rDrP.)rRrSr4rPrU)rFrDr()r.rcr\r rdrer7) rfrRrSr4rUrDrPrZr\Z openai_toolsrgr(r(r)acreate_assistants   z+OpenAIAssistantV2Runnable.acreate_assistantc s|pi}tj|d|d|dd}|jt|||dp$|d}t|dg}|dg|}z|jrW|d rW||d }|j j j j j di|Id H} nUd |vrxd |d ||ddg|dd} ||| Id H} n4d|vr|j j j jj|d |d d ||ddId H} ||Id H} n|j j j j j di|Id H} || j| jId H} Wnty} z|| | d } ~ wwz|| } Wnty} z |j| | d| d } ~ ww|| | S)a*Async invoke assistant. Args: input (dict): 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. file_ids: (deprecated) File ids to include in new run. Use 'attachments' instead attachments: Assistant files to include in new run. (v2 API). 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. tool_resources: Override Assistant tool resources for this run (v2 API). run_metadata: Metadata to associate with new run. config (Optional[RunnableConfig]): Configuration for the run. Returns: OutputType: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. Raises: BaseException: If an error occurs during the invocation. rlrmrnrorprqr/r6rrNrsrtrurvrwryrzr|r}r~r()rrrrrrr8rIrrDr rrr_acreate_thread_and_runr{re _acreate_runZ_await_for_runr7rsrrrr=rrr(r(r)ainvokesr!      z!OpenAIAssistantV2Runnable.ainvokecs>dfdd|D}|jjjjj|dfd|ji|S)zCreate a new run within an existing thread. Args: input (dict): The input data for the new run. Returns: Any: The created run object. )rSrUr4rP run_metadataZtruncation_strategyZmax_prompt_tokenscsi|] \}}|vr||qSr(r(r`kvZallowed_assistant_paramsr(r) +sz9OpenAIAssistantV2Runnable._create_run..rsrF)itemsrCr rrrerFrMriparamsr(rr)rs  z%OpenAIAssistantV2Runnable._create_runrcCsJdd|D}|d}r||d<|jjjjd|j|d|}|S)zCreate a new thread and run. Args: input (dict): The input data for the run. thread (dict): The thread data to create. Returns: Any: The created thread and run. cSi|] \}}|dvr||qS)rSrUr4rr(rr(r(r)r< zDOpenAIAssistantV2Runnable._create_thread_and_run..rPrFrNr()rrrCr rcreate_and_runrFrMrirrrPrr(r(r)r2s  z0OpenAIAssistantV2Runnable._create_thread_and_runcs>dd|D}|jjjjj|dfd|ji|IdHS)zAsynchronously create a new run within an existing thread. Args: input (dict): The input data for the new run. Returns: Any: The created run object. cSr))rSrUr4rPrr(rr(r(r)rSrz:OpenAIAssistantV2Runnable._acreate_run..rsrFN)rrDr rrrerFrr(r(r)rJs   z&OpenAIAssistantV2Runnable._acreate_runcsRdd|D}|d}r||d<|jjjjd|j|d|IdH}|S)zAsynchronously create a new thread and run simultaneously. Args: input (dict): The input data for the run. thread (dict): The thread data to create. Returns: Any: The created thread and run. cSrrr(rr(r(r)rhrzEOpenAIAssistantV2Runnable._acreate_thread_and_run..rPrNr()rrrDr rrrFrr(r(r)r^s   z1OpenAIAssistantV2Runnable._acreate_thread_and_run)rr)rRrErSrEr4rTrUrErOrVrCrWrPrXrQrYrZrrr )N)rir=rjrkrZrrr )rRrErSrEr4rTrUrErDrrPrXrZrrr )rir=rr)rir=rr=rr)__name__ __module__ __qualname____doc__rr*rC__annotations__rDrHrIrrN classmethodrhrrrrrrrr(r(r(r)rAs@ S     2 ^ . b  rA)rr)rr,)r/r0rr0)r9r:rr;)r9r:rr?)- __future__rtypingrrrrrrr r Z&langchain.agents.openai_assistant.baser r Zlangchain_core._apir Zlangchain_core.callbacksrZlangchain_core.loadrZlangchain_core.runnablesrrZlangchain_core.toolsrZ%langchain_core.utils.function_callingrZpydanticrrrZtyping_extensionsrr#rcrZopenai.types.beta.assistantrZAssistantToolResourcesr*r.r8r>r@rAr(r(r(r)s, (