tj(yddlmZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZdd lmZdd lmZmZmZdd lmZdd lmZdd lmZmZmZddl m!Z!m"Z"e rddl#Z#ddl$m%Z%ddl&m'Z'GddeZ(GddeZ)d$dZ*d%dZ+d&dZ,d'dZ-e.e)e(ze.d ze.d!zZ/Gd"d#ee0e/fZ1dS)() annotationsN)CallableSequence)JSONDecodeError)sleep) TYPE_CHECKINGAny) AgentAction AgentFinish)CallbackManager)dumpd)RunnableConfigRunnableSerializable ensure_config)BaseTool)convert_to_openai_tool) BaseModelFieldmodel_validator)Selfoverride) ThreadMessage)RequiredActionFunctionToolCallc@eZdZUdZded<ded<ed dZdS) OpenAIAssistantFinishzoAgentFinish with run and thread metadata. Args: run_id: Run id. thread_id: Thread id. strrun_id thread_idreturnboolcdSz]Check if the class is serializable by LangChain. Returns: False Fclss l/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_classic/agents/openai_assistant/base.pyis_lc_serializablez(OpenAIAssistantFinish.is_lc_serializable+ uNrr __name__ __module__ __qualname____doc____annotations__ classmethodr'r#r)r&rr sRKKKNNN[r)rcJeZdZUdZded<ded<ded<ed dZd S) OpenAIAssistantActionzAgentAction with info needed to submit custom tool output to existing run. Args: tool_call_id: Tool call id. run_id: Run id. thread_id: Thread id r tool_call_idrrrr cdSr"r#r$s r&r'z(OpenAIAssistantAction.is_lc_serializableBr(r)Nr*r+r#r)r&r3r35s^KKKNNN[r)r3r openai.OpenAIc ddl}|jS#t$r}d}t||d}~wt$r}d}t||d}~wwxYwNrzBUnable 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 ImportErrorAttributeErrorr9emsgs r&_get_openai_clientr@Ls ) v} &&&R#A% ))) < S!!q( ) A1 AAAopenai.AsyncOpenAIc ddl}|jS#t$r}d}t||d}~wt$r}d}t||d}~wwxYwr8)r9 AsyncOpenAIr;r<r=s r&_get_openai_async_clientrE\s ) !v!### &&&R#A% ))) < S!!q( )rAtool6dict[str, Any] | type[BaseModel] | Callable | BaseToolr cLd}t|to d|vo |d|vS)zd+Zd?d-Zd@d/ZdAd2Zd=d3Zd>d4Zd?d5Zd@d6ZdAd7ZdS)BOpenAIAssistantRunnablea Run an OpenAI Assistant. Example using OpenAI tools: ```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: ```python from langchain_experimental.openai_assistant import OpenAIAssistantRunnable from langchain_classic.agents import AgentExecutor from langchain_classic.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: ```python from langchain_experimental.openai_assistant import OpenAIAssistantRunnable from langchain_classic.agents import AgentExecutor from langchain_core.agents import AgentFinish from langchain_classic.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_factoryr clientN async_clientr assistant_idg@@floatcheck_every_msFr as_agentafter)moderrc`|j&ddl}|jj}|j||_|S)Nr)api_key)rWr9rVr_rD)selfr9r_s r&_validate_async_clientz.OpenAIAssistantRunnable._validate_async_clients;   $ MMMk)G 2 27 C C CD  r))rVname instructionstoolsSequence[BaseTool | dict]model)openai.OpenAI | openai.AzureOpenAI | Nonekwargsc |p t}|jj||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. c,g|]}t|Sr#rR.0rFs r& z.s!@@@$'--@@@r)rbrcrdrf)rXrVr#)r@beta assistantscreateid)r%rbrcrdrfrVrh assistants r&create_assistantz(OpenAIAssistantRunnable.create_assistantso2/-//K*11%@@%@@@ 2  sF VFFvFFFr)inputrMconfigRunnableConfig | None OutputTypec Ht|}tj|d|d|d}|t |||dp|} |jrR|dr=||d}|j j j j j di|}nd|vrad |d |d |d d g|dd}|||}nd|vrb|j j j j|d|d d |d } ||}n!|j j j j j di|}||j|j}n(#t*$r} || d} ~ wwxYw ||} || | S#t*$r/} || |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. 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. **kwargs: Additional arguments. Returns: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. callbackstagsmetadatainheritable_callbacksinheritable_tagsinheritable_metadatarun_namerbintermediate_stepsrusercontentmessage_metadata attachments)rolerr}rthread_metadatamessagesr}rrrr}Nr}r#)rr configuregeton_chain_startr get_namer[_parse_intermediate_stepsrVrpthreadsrunssubmit_tool_outputs_create_thread_and_runrrr _create_run _wait_for_runrsr BaseExceptionon_chain_error _get_response on_chain_endrM r`rvrwrhcallback_manager run_manager tool_outputsrunthread_r>responses r&invokezOpenAIAssistantRunnable.invoke!sNv&&*4"(**["9"9#ZZ//!'J!7!7    '55 $KK J'':4==??6  % } Q+?!@!@ Q#==./   Hdk&.3GWW,WWE))%+',Y'7(- 2D(E(E+099]+C+C !!& *; < <  11%@@&&K$,5<<+&!),"YY'9:: = &&u--Hdk&.3GPP%PP$$SVS];;CC     & &q ) ) )   ))#..H  $ $X . . .O      & &q388:: & > > >  s1-E%H H8H33H8<I(( J!2*JJ!)rW3openai.AsyncOpenAI | openai.AsyncAzureOpenAI | NonecK|p t}d|D}|jj||||d{V}|d|j|d|S)a5Async create an AsyncOpenAI 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. async_client: AsyncOpenAI client. Will create default async_client if not specified. **kwargs: Additional arguments. Returns: AsyncOpenAIAssistantRunnable configured to run using the created assistant. c,g|]}t|Sr#rkrls r&rnz=OpenAIAssistantRunnable.acreate_assistant..s!EEEt,T22EEEr)roN)rXrWr#)rErprqrrrs) r%rbrcrdrfrWrh openai_toolsrts r&acreate_assistantz)OpenAIAssistantRunnable.acreate_assistants2$A'?'A'A EEuEEE &+6==% >        sR <RR6RRRr)cnK|pi}tj|d|d|d}|t |||dp|} |jr^|drI||dd{V}|jj j j j di|d{V}nd |vrSd |d |d d g|dd}| ||d{V}nd|vrn|jj j j|d |d d |d d{V} ||d{V}n'|jj j j j di|d{V}||j|jd{V}n(#t($r} || d} ~ wwxYw ||d{V} || | S#t($r/} || |d} ~ wwxYw)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. kwargs: Additional arguments. Returns: If self.as_agent, will return Union[List[OpenAIAssistantAction], OpenAIAssistantFinish]. Otherwise, will return OpenAI types Union[List[ThreadMessage], List[RequiredActionFunctionToolCall]]. r{r|r}r~rrrNrrrr)rrr}rrrrrr#)r rrrr rr[_aparse_intermediate_stepsrWrprrr_acreate_thread_and_runrrr _acreate_run_await_for_runrsrrr_aget_responserrMrs r&ainvokezOpenAIAssistantRunnable.ainvokesL2*4"(**["9"9#ZZ//!'J!7!7    '55 $KK J'':4==??6  ( }" +?!@!@" %)%D%D./&&       TD-2:?S"E))%+',Y'7(- 2D(E(E!!& *; < <  !88GGGGGGGG&&+08AHH+&!),"YY'9:: I !--e44444444TD-2:?S++CFCMBBBBBBBBCC     & &q ) ) )   !0055555555H  $ $X . . .O      & &q388:: & > > >  s1$E;H I*II I;; J4*J//J4r'list[tuple[OpenAIAssistantAction, str]]c|d\}}||j|j}t|jrd|jjjDfd|D}||j|jdS)Nch|] }|j Sr#rsrmtcs r& zDOpenAIAssistantRunnable._parse_intermediate_steps..'&&&&&&r)cVg|]%\}}|jvt||jd&S)outputr4r4rrmactionrrequired_tool_call_idss r&rnzEOpenAIAssistantRunnable._parse_intermediate_steps..G   "&<<<6{{F4G H H<<z7OpenAIAssistantRunnable._create_run..!@   1    q   r)rrX)itemsrVrprrrrrXr`rparamss r&rz#OpenAIAssistantRunnable._create_run sr  "((**   "4t{',3 { #  *    r)rcd|D}|jjjjd|j|d|S)Nc"i|] \}}|dv || S) rcrfrdrrrrrrr#rs r&rzBOpenAIAssistantRunnable._create_thread_and_run..9@   1    q   r)rXrr#)rrVrprcreate_and_runrXr`rrrs r&rz.OpenAIAssistantRunnable._create_thread_and_run8sj  "((**    7t{'6 *     r)rc jdkr@ddltjjdd}tjjdd}|dkp |dko|dk|jjjj j d}fd|D}|j s|Sd |D}d |D}tfd |Drd d |D}t|j j|ddjj Sjdkr|j sjjjSg}jjjD]} | j} t+j| jd} n4#t0$r'} d| jd| j} t5| | d} ~ wwxYwt7| dkr d| vr| d} |t;| j| | jdjj |St+jd}djd|} t5| )N completedr.ascorderc4g|]}|jjk|Sr#rrsrmr?rs r&rnz9OpenAIAssistantRunnable._get_response.._'LLLCszSV7K7KC7K7K7Kr)c&g|]}|jD]}|Sr#rrmr? msg_contents r&rnz9OpenAIAssistantRunnable._get_response..b; #s{8C r)c&g|]}|jD]}|Sr#)r)rmr? attachments r&rnz9OpenAIAssistantRunnable._get_response..es;"S_7A r)c3K|]N}r$t|jjjjn#t|jjjjVOdSNrLtypesrprTextContentBlockMessageContentTextrmrr9version_gte_1_14s r& z8OpenAIAssistantRunnable._get_response..hv   (Jw (9(A(RSSS# )1D      r) c3.K|]}|jjVdSrtextvaluermrs r&rz8OpenAIAssistantRunnable._get_response..s'"L"L'7<#5"L"L"L"L"L"Lr))rrrr return_valueslogrrrequires_actionFstrict*Received invalid JSON function arguments:  for function __arg1rF tool_inputr4r rrindentUnexpected run status: . Full run info: ) statusr9intversionVERSIONsplitrVrprrlistrr[alljoinrrsrrrfunctionjsonloads argumentsrrb ValueErrorlenappendr3dumpsrM)r`r major_version minor_versionr new_messagesanswerractions tool_callr"argsr>r?run_infor9rs ` @@r&rz%OpenAIAssistantRunnable._get_responseOsA : $ $ MMM 6 < HMLLL8LLLL= $##'3F&2K      &      M"L"LV"L"L"LLL($!$!f#.  v-     :* * *= J*>IIG 0DO   $-1:h&8GGGDD&111M#-MM=E]MM%S//q0 1 t99>>h$&6&6>D)%]#'%.\"v"%-     N:chhjj333R RRRRoosF,, G6"GGrrcd}|rO|jjjj||}|jdv}|rt |jdz |O|SNT)r) in_progressqueuedi)rVrprrretrieverrrZr`rrr4rs r&rz%OpenAIAssistantRunnable._wait_for_runsn  2+"*/8898UUC*(AAK 2d)D0111  2  r)cK|d\}}||j|j}t|jrd|jjjDfd|D}||j|jdS)Nrch|] }|j Sr#rrs r&rzEOpenAIAssistantRunnable._aparse_intermediate_steps..rr)cVg|]%\}}|jvt||jd&Srrrs r&rnzFOpenAIAssistantRunnable._aparse_intermediate_steps..rr)rrrs @r&rz2OpenAIAssistantRunnable._aparse_intermediate_stepss,B/ Q  !3[5JKK!$   && # 3 G R&&& "    "4   )!($.   r)cKd|D}|jjjjj|dfd|ji|d{VS)Nc"i|] \}}|dv || Srr#rs r&rz8OpenAIAssistantRunnable._acreate_run..rr)rrX)rrWrprrrrrXrs r&rz$OpenAIAssistantRunnable._acreate_runs  "((**   "@T&+38? { #  *          r)cKd|D}|jjjjd|j|d|d{VS)Nc"i|] \}}|dv || Srr#rs r&rzCOpenAIAssistantRunnable._acreate_thread_and_run..rr)rr#)rrWrprrrXrs r&rz/OpenAIAssistantRunnable._acreate_thread_and_runs  "((**    CT&+3B *           r)c Kjdkr9ddltjjdd}tjjdd}|dkp |dko|dk|jjjj j dd{V}fd|D}|j s|Sd |D}tfd |Drd d |D}t|j jd djj Sjdkr|j sjjjSg}jjjD]}|j} t+j| jd} n4#t0$r'} d| jd| j} t5| | d} ~ wwxYwt7| dkr d| vr| d} |t;| j| |jdjj |St+jd} djd| } t5| )Nrrrrrrrc4g|]}|jjk|Sr#rrs r&rnz:OpenAIAssistantRunnable._aget_response..rr)c&g|]}|jD]}|Sr#rrs r&rnz:OpenAIAssistantRunnable._aget_response..rr)c3K|]N}r$t|jjjjn#t|jjjjVOdSrrrs r&rz9OpenAIAssistantRunnable._aget_response..rr)rc3.K|]}|jjVdSrrrs r&rz9OpenAIAssistantRunnable._aget_response.. rr))rrrr r r Frrrrrrrrr) rr9rrrrrWrprrrrr[r r!rrsrrrr"r#r$r%rrbr&r'r(r3r)rM)r`rr*r+rr,r-r.r/r"r0r>r?r1r9rs ` @@r&rz&OpenAIAssistantRunnable._aget_responsesD : $ $ MMM 6 < IIG 0DO   $-1:h&8GGGDD&111M#-MM=E]MM%S//q0 1 t99>>h$&6&6>D)%]#'%.\"v"%-     N:chhjj333R RRRRoos F'' G1"GGcKd}|r`|jjjj||d{V}|jdv}|r"t j|jdz d{V|`|Sr3) rWrprrr6rasynciorrZr7s r&rz&OpenAIAssistantRunnable._await_for_run7s  @).6;DD#EC*(AAK @mD$7$$>????????? @ r))rr)rbrrcrrdrerfrrVrgrhr rrTr)rvrMrwrxrhr rry)rbrrcrrdrerfrrWrrhr rrT)rrrrM)rrMrr )rrMrrMrr )rr rr )rrrrrr )r,r-r.r/rr@rVr0rWrZr[rrar1rurrrrrrrrrrrrrrr#r)r&rTrTs_[[z%(:;;;F;;;;+L1#N####;HH_'"""#"=AGGGGG[GB)-____X_BMQ S S S S S[ SD)-aaaaXaF    ,    0    .MMMM^    ,    0    .IIIIV      r)rT)rr6)rrB)rFrGrr )rFrGrrP)2 __future__rrEr#collections.abcrrrtimertypingrr langchain_core.agentsr r langchain_core.callbacksr langchain_core.loadr langchain_core.runnablesrrrlangchain_core.toolsr%langchain_core.utils.function_callingrpydanticrrrtyping_extensionsrrr9openai.types.beta.threadsrrTs"""""" ........  ;:::::::444444%%%%%%XXXXXXXXXX))))))HHHHHH6666666666,,,,,,,,MMM K*K. ) ) ) ) ) ) ) )     ( ( ( (  ? +,- t t t t t 243CDt t t t t r)