蘺h;ddlmZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlmZddlmZmZddlmZdd lmZdd lmZmZmZdd lmZdd lmZerddl Z dd l!m"Z"ddl#m$Z$GddeZ%GddeZ&ddZ'ee e&e%e de dfZ(Gddee e(fZ)dS)) annotationsN)JSONDecodeError)sleep) TYPE_CHECKINGAnyDictListOptionalSequenceTupleUnion)format_tool_to_openai_tool) AgentAction AgentFinish)dumpd)Field)RunnableConfigRunnableSerializable ensure_config)BaseTool)CallbackManager) ThreadMessage)RequiredActionFunctionToolCallc@eZdZUdZded<ded<ed dZdS) OpenAIAssistantFinishz)AgentFinish with run and thread metadata.strrun_id thread_idreturnboolcdSNFclss n/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain/agents/openai_assistant/base.pyis_lc_serializablez(OpenAIAssistantFinish.is_lc_serializableuNrr __name__ __module__ __qualname____doc____annotations__ classmethodr'r#r)r&rrsL33KKKNNN[r)rcJeZdZUdZded<ded<ded<ed dZd S) OpenAIAssistantActionzJAgentAction with info needed to submit custom tool output to existing run.r tool_call_idrrrr cdSr"r#r$s r&r'z(OpenAIAssistantAction.is_lc_serializable+r(r)Nr*r+r#r)r&r3r3$sXTTKKKNNN[r)r3r openai.OpenAIc ddl}|jS#t$r}td|d}~wt$r}td|d}~wwxYw)NrzBUnable 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 ImportErrorAttributeError)r8es r&_get_openai_clientr=0s  v}  P     <   s A/ AA  ArrceZdZUdZeeZded< ded< dZded < d Z d ed < e d dd*dZ d+d,dZ d-d!Z d.d"Zd/d$Zd0d&Zd1d)Zd S)2OpenAIAssistantRunnablea 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_factoryr6clientr assistant_idg@@floatcheck_every_msFr as_agentN)rAname instructionstoolsSequence[Union[BaseTool, dict]]modelOptional[openai.OpenAI]kwargsrrc  |p t}g}|D]=}t|tr|nt|} || >|jj||||} |dd| ji|S)aCreate 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 client. Will create default client if not specified. Returns: OpenAIAssistantRunnable configured to run using the created assistant. )rFrGrHrJrBr#) r= isinstancedictrappendbeta assistantscreateid) r%rFrGrHrJrArL openai_toolstooloai_tool assistants r&create_assistantz(OpenAIAssistantRunnable.create_assistants./-//  * *D"4..T4Nt4T4T     ) ) ) )K*11% 2  s77 7777r)inputrOconfigOptional[RunnableConfig] OutputTypec Rt|}tj|d|d|d}|t |||d} |jrS|dr>||d}|jj j j j di|}nd|vrbd |d |d g|d d g|dd}| ||}nd|vrw|jj j j|d|d d |d g|d }||}n!|jj 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. file_ids: File ids to include in new run. Used for retrieval. 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: Return: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. callbackstagsmetadata)inheritable_callbacksinheritable_tagsinheritable_metadatarun_name)rFintermediate_stepsrusercontentfile_idsmessage_metadata)rolerhrirathread_metadata)messagesrar)rhrkriraN)rar#)rr configuregeton_chain_startrrE_parse_intermediate_stepsrArQthreadsrunssubmit_tool_outputs_create_thread_and_runrmrS _create_run _wait_for_runrTr BaseExceptionon_chain_error _get_response on_chain_endrO) selfrZr[callback_manager run_manager tool_outputsrunthread_r<responses r&invokezOpenAIAssistantRunnable.invokes8v&&*4"(**["9"9#ZZ//!'J!7!7    '55 $KKVZZ %;%;6  & } Q+?!@!@ Q#==./   Hdk&.3GWW,WWE))%+',Y'7(- *b(A(A(- 2D(E(E !!& *; < <  11%@@&&K$,5<<+&!),"YYz266"YY'9:: =&&u--Hdk&.3GPP%PP$$SVS];;CC     & &q ) ) )G  ))#..H  $ $X . . .O      & &q388:: & > > >G s1EzDOpenAIAssistantRunnable._parse_intermediate_steps..s'" " " BE" " " r)cVg|]%\}}|jvt||jd&S))outputr4)r4r)ractionrrequired_tool_call_idss r& zEOpenAIAssistantRunnable._parse_intermediate_steps.."sG   "&<<<6{{F4G H H<<z7OpenAIAssistantRunnable._create_run../4   1FFF qFFFr)rrB)itemsrArQrrrsrSrB)r|rZparamss r&rvz#OpenAIAssistantRunnable._create_run.sn       4t{',3 +   *    r)rcd|D}|jjjjd|j|d|}|S)Nc"i|] \}}|dv || Srr#rs r&rzBOpenAIAssistantRunnable._create_thread_and_run..;rr))rBrr#)rrArQrrcreate_and_runrB)r|rZrrrs r&ruz.OpenAIAssistantRunnable._create_thread_and_run:sj       6dk&5 *      r)rc  jdkrddl |jjjjjd}fd|D}|js|Sd|D}t fd|Drd d |D}t|jj d d j j Sjd kr|jsj jjSg}j jjD]}|j} t#j|jd}n2#t($r%} t+d|jd|j| d} ~ wwxYwt/|dkr d|vr|d}|t3|j||j d j j|St#jd} t+djd| d)N completedrasc)orderc4g|]}|jjk|Sr#)rrT)rmsgrs r&rz9OpenAIAssistantRunnable._get_response..Ps'LLLCszSV7K7KC7K7K7Kr)c&g|]}|jD]}|Sr#)rh)rr msg_contents r&rz9OpenAIAssistantRunnable._get_response..Ss; #s{8C r)c3`K|](}t|jjjjV)dSN)rNtypesrQrrMessageContentText)rrhr8s r& z8OpenAIAssistantRunnable._get_response..VsL7FL$5$=$PQQr) c3.K|]}|jjVdSr)textvalue)rrhs r&rz8OpenAIAssistantRunnable._get_response..Zs'"L"L'7<#5"L"L"L"L"L"Lr))rrr) return_valueslogrrrequires_actionF)strictz*Received invalid JSON function arguments: z for function __arg1)rV tool_inputr4rrr)indentzUnexpected run status: z. Full run info: ))statusr8rArQrrrmlistrrEalljoinrrTrrtrfunctionjsonloads argumentsr ValueErrorrFlenrPr3dumpsrO) r|rrm new_messagesansweractions tool_callrargsr<run_infor8s ` @r&rzz%OpenAIAssistantRunnable._get_responseGs : $ $ MMM{'/8== U>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 s D&& E0 EErrcd}|rO|jjjj||}|jdv}|rt |jdz |O|S)NT)r) in_progressqueuedi)rArQrrrsretrieverrrD)r|rrrrs r&rwz%OpenAIAssistantRunnable._wait_for_runsn  2+"*/8898UUC*(AAK 2d)D0111  2  r))rFrrGrrHrIrJrrArKrLrrr?r)rZrOr[r\rr])rfrrrO)rZrOrr)rZrOrrOrr)rrrr)rrrrrr)r,r-r.r/rr=rAr0rDrEr1rYrrqrvrurzrwr#r)r&r?r?Hs6KKZ"E2DEEEFEEEE#N####;HF+/#8#8#8#8#8[#8L?CSSSSSj####(         <<<<|r)r?)rr6)* __future__rrrtimertypingrrrr r r r r +langchain_community.tools.convert_to_openairlangchain_core.agentsrrlangchain_core.loadrlangchain_core.pydantic_v1rlangchain_core.runnablesrrrlangchain_core.toolsrlangchain.callbacks.managerrr8openai.types.beta.threadsrrs0"""""" SSSSSSSSSSSSSSSSSSSSRRRRRR::::::::%%%%%%,,,,,,XXXXXXXXXX))))))777777MMM777777 K     K          )*, DDDDD243CDDDDDDr)