a bg|_@s:ddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZddlmZmZddlmZdd lmZdd lmZmZmZdd lm Z erdd l!Z!dd l"m#Z#ddl$m%Z&ddddZ'ddddZ(dddddZ)dddddZ*dddd 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) ToolResourcesz openai.OpenAIreturnc Cszzddl}|jddidWStyH}ztd|WYd}~n6d}~0tyt}ztd|WYd}~n d}~00dS)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'~/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/agents/openai_assistant/base.py_get_openai_clients r)zopenai.AsyncOpenAIc Cszzddl}|jddidWStyH}ztd|WYd}~n6d}~0tyt}ztd|WYd}~n d}~00dS)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_client7s r+list)file_idsrcCs.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 r5z:Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]bool)toolrcCs"d}t|to d|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. )r0Z retrievalr/r.) isinstancedict)r7Zassistants_builtin_toolsr'r'r(_is_assistants_builtin_tooles   r:zDict[str, Any]cCst|r |St|SdS)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. N)r:r)r7r'r'r(_get_assistants_toolzsr;c @s0eZdZUdZeedZded<dZded<ded<d Z d ed <d Z d ed<e ddddddZ e iddddddddddddddd ddZd6dd dd!d"d#d$Ze ddd%ddddd&dddd'd(d)Zd7dd dd!d"d*d+Zddd,d-d.Zdddd/d0d1Zddd,d2d3Zdddd/d4d5ZdS)8OpenAIAssistantV2Runnablea 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_msFr6as_agentafter)moderrcCs,|jdur(ddl}|jj}|j|d|_|S)z?Validate that the async client is set, otherwise initialize it.Nr)api_key)r?r"r>rGr*)selfr"rGr'r'r(validate_async_clients  z/OpenAIAssistantV2Runnable.validate_async_client) model_kwargsr>tool_resources extra_bodyzSequence[Union[BaseTool, dict]]zdict[str, float]z2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]z7Optional[Union[AssistantToolResources, dict, NotGiven]]zOptional[object]r ) name instructionsr1modelrJr>rKrLkwargsrc Ksd|pt}|dur"ddlm} | }|jjjf||dd|D|||d|} |f| 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_GIVENcSsg|] }t|qSr'r;.0r7r'r'r( #z>OpenAIAssistantV2Runnable.create_assistant..)rMrNr1rKrOrL)rAr>)r) openai._typesrRr assistantscreater4) clsrMrNr1rOrJr>rKrLrPrR assistantr'r'r(create_assistants"     z*OpenAIAssistantV2Runnable.create_assistantr9zOptional[RunnableConfig]r )inputconfigrPrc Kst|}tj|d|d|dd}|jt|||dpF|d}t|dg}|dg|}z|jr|d r| |d }|j j j j jfi|} nd |vrd |d ||d dg|dd} ||| } nXd|vr"|j j j jj|d |d d ||d d} ||} n|j j j j jfi|} || j| j} Wn4ty} z|| | WYd} ~ n d} ~ 00z|| } Wn<ty} z"|j| | d| WYd} ~ nd} ~ 00|| | SdS)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_namerMr-r3intermediate_steps thread_idusercontentmessage_metadatarolerir3rbthread_metadatamessagesrbrun_idrirlr3rbNrb)rr configuregeton_chain_startrget_namer5rD_parse_intermediate_stepsr>r threadsrunssubmit_tool_outputs_create_thread_and_runrorZ _create_runZ _wait_for_runr4rg BaseExceptionon_chain_error _get_responser9 on_chain_endrHr^r_rPZcallback_managerZ run_managerfilesr3Z tool_outputsrunthread_r&responser'r'r(invoke+s`!      z OpenAIAssistantV2Runnable.invoke)r?rKz.)rMrNr1rKrO)rAr?)r+rXrRr rYrZr4) r[rMrNr1rOr?rKrPrRZ openai_toolsr\r'r'r(acreate_assistants   z+OpenAIAssistantV2Runnable.acreate_assistantc s|pi}tj|d|d|dd}|jt|||dpF|d}t|dg}|dg|}z|jr|d r||d }|j j j j j fi|Id H} nd |vrd |d ||ddg|dd} ||| Id H} njd|vr<|j j j jj|d |d d ||ddId H} ||Id H} n|j j j j j fi|Id H} || j| jId H} Wn4ty} z|| | WYd } ~ n d } ~ 00z|| } Wn<ty} z"|j| | d| WYd } ~ nd } ~ 00|| | Sd 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. r`rarbrcrdrer-r3rfNrgrhrirjrkrmrnrprqrr)rrsrtrurrvr5rDrwr?r rxryrz_acreate_thread_and_runrorZ _acreate_runZ_await_for_runr4rgr}r~rr9rrr'r'r(ainvokesh!       z!OpenAIAssistantV2Runnable.ainvoke)r^rcs>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. )rNrOr1rK run_metadataZtruncation_strategyZmax_prompt_tokenscsi|]\}}|vr||qSr'r'rUkvZallowed_assistant_paramsr'r( +rWz9OpenAIAssistantV2Runnable._create_run..rgrA)itemsr>r rxryrZrArHr^paramsr'rr(r|s   z%OpenAIAssistantV2Runnable._create_run)r^rrcCsJdd|D}|d}r(||d<|jjjjf|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. cSsi|]\}}|dvr||qS)rNrOr1rr'rr'r'r(r<szDOpenAIAssistantV2Runnable._create_thread_and_run..rKrAr)rrtr>r rxcreate_and_runrArHr^rrrKrr'r'r(r{2s  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. cSsi|]\}}|dvr||qS))rNrOr1rKrr'rr'r'r(rSsz:OpenAIAssistantV2Runnable._acreate_run..rgrAN)rr?r rxryrZrArr'r'r(rJs  z&OpenAIAssistantV2Runnable._acreate_runcsPdd|D}|d}r(||d<|jjjjf|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. cSsi|]\}}|dvr||qSrr'rr'r'r(rhszEOpenAIAssistantV2Runnable._acreate_thread_and_run..rKrN)rrtr?r rxrrArr'r'r(r^s   z1OpenAIAssistantV2Runnable._acreate_thread_and_run)N)N)__name__ __module__ __qualname____doc__rr)r>__annotations__r?rCrDrrI classmethodr]rrrr|r{rrr'r'r'r(r<s4 S    &2^ ".br<)- __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"rXrZopenai.types.beta.assistantrZAssistantToolResourcesr)r+r5r:r;r<r'r'r'r(s* (