h6_:ddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZddlmZmZddlmZdd lmZdd lmZmZmZdd lm Z erdd l!Z!dd l"m#Z#ddl$m%Z&ddZ'ddZ(ddZ) ddZ* ddZ+eGdde Z,y )) annotations) TYPE_CHECKINGAnyCallableDictOptionalSequenceTypeUnion)OpenAIAssistantRunnable OutputType)beta)CallbackManager)dumpd)RunnableConfig ensure_config)BaseTool)convert_to_openai_tool) BaseModelFieldmodel_validator)SelfN)NotGiven) ToolResourcesc ddl}|jddiS#t$r}td|d}~wt$r}td|d}~wwxYw)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"`.)openaiOpenAI ImportErrorAttributeErrorr"es n/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langchain_community/agents/openai_assistant/base.py_get_openai_clientr)se v}}m_-MNN  P    =   A  0 A  AA c ddl}|jddiS#t$r}td|d}~wt$r}td|d}~wwxYw)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&s r(_get_openai_async_clientr-7sg !v!!=/2RSS  P    =  r*cLg}|D]}|j|ddiddigd|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)file_idtools)append)file_ids attachmentsids r("_convert_file_ids_into_attachmentsr8PsIK !=1Fs r(_get_assistants_toolrBzs#4( %d++r9c\eZdZUdZeeZded< dZded< ded< d Z d ed < d Z d ed< e dddZ e idddd ddZ d ddZe ddd d dZ d ddZd!dZd"dZd!dZd"dZy)#OpenAIAssistantV2Runnablea 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_msFboolas_agentafter)modec|j2ddl}|jj}|j||_|S)z?Validate that the async client is set, otherwise initialize it.Nr)api_key)rGr"rFrQr,)selfr"rQs r(validate_async_clientz/OpenAIAssistantV2Runnable.validate_async_clients<    $ kk))G 2 2 27 CD  r9) model_kwargsrFtool_resources extra_bodyc |xs t}|ddlm} | }|jjj d|||D cgc] } t | c} |||d|} |d| j|d| Scc} w)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. r NOT_GIVEN)name instructionsr3rUmodelrV)rIrF)r) openai._typesrYr assistantscreaterBr7) clsrZr[r3r\rTrFrUrVkwargsrYr> assistants r(create_assistantz*OpenAIAssistantV2Runnable.create_assistantsD/-/  ! /&N1FKK**11 %:?@$'-@)!   F VFvFF AsA2c jt|}tj|jd|jd|jd}|j t |||jdxs|j }t|jdg}|jdg|z} |jr`|jd rO|j|d }|jjjjjdi|} nd |vr?d |d ||jd dg|jdd} |j|| } nd|vre|jjjj j#|d |d d ||jd } |j%|} n:|jjjjjdi|} |j'| j(| j*} |j1| } |j3| | S#t,$r} |j/| | d} ~ wwxYw#t,$r(} |j/| | j5| d} ~ wwxYw)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. callbackstagsmetadatainheritable_callbacksinheritable_tagsinheritable_metadatarun_namerZr5r6intermediate_steps thread_idusercontentmessage_metadatarolerrr6rhthread_metadatamessagesrhrun_idrrrur6rhNrhr])rr configuregeton_chain_startrget_namer8rM_parse_intermediate_stepsrFrthreadsrunssubmit_tool_outputs_create_thread_and_runrxr` _create_run _wait_for_runr7rp BaseExceptionon_chain_error _get_response on_chain_endr=rRinputconfigrbcallback_manager run_managerfilesr6 tool_outputsrunthread_r'responses r(invokez OpenAIAssistantV2Runnable.invoke+sBv&*44"(**["9#ZZ/!'J!7  '55 $KVZZ %;%Nt}}6 36::j"3MNjj3e; & }}+?!@#==./  Hdkk&&..33GGW,WE)%+',Y'7+6(- 2D(E !!& *; < 11%@&KK$$,,55<<+&!), +"YY'9: =&&u-Hdkk&&..33GGP%P$$SVVS]];C ))#.H  $ $X .O   & &q )G     & &q388: & >G s1E8I9J I>&I99I> J2 #J--J2)rGrUcK|xs t}|ddlm}|}|D cgc] } t| } } |jj j ||| ||d{} |d| j|d|Scc} w7w)aCreate an AsyncOpenAI 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. async_client (Optional[Union[openai.OpenAI, openai.AzureOpenAI]]): OpenAI or AzureOpenAI async client. Will create default async_client if not specified. Returns: AsyncOpenAIAssistantRunnable: The configured assistant runnable. NrrX)rZr[r3rUr\)rIrGr])r-r^rYrBrr_r`r7) rarZr[r3r\rGrUrbrYr> openai_toolsrcs r(acreate_assistantz+OpenAIAssistantV2Runnable.acreate_assistants<$A'?'A  ! /&N?DEt,T2E E&++66==%) >  R <R6RRF sBA:/B A?!Bc K|xsi}tj|jd|jd|jd}|jt |||jdxs|j }t |jdg}|jdg|z} |jri|jd rX|j|d }|jjjjjdi|d {} nd |vrGd |d ||jddg|jdd} |j|| d {} nd|vru|jjjjj!|d |d d ||jdd {} |j#|d {} nB|jjjjjdi|d {} |j%| j&| j(d {} |j/| } |j1| | S7`7777a75#t*$r} |j-| | d } ~ wwxYw#t*$r(} |j-| | j3| d } ~ wwxYww)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. rfrgrhrirmrnr5r6roNrprqrrrsrtrvrwryrzr{r])rr|r}r~rrr8rMrrGrrrr_acreate_thread_and_runrxr` _acreate_run_await_for_runr7rprrrrr=rs r(ainvokez!OpenAIAssistantV2Runnable.ainvokesB2*44"(**["9#ZZ/!'J!7  '55 $KVZZ %;%Nt}}6 36::j"3MNjj3e; * }}+?!@#==./  TD--22::??SS"E)%+',Y'7+6(- 2D(E !!& *; < !88GG&++0088AAHH+&!), +"YY'9: I!--e44TD--22::??SS++CFFCMMBBC ))#.H  $ $X .OY H5C   & &q )G     & &q388: & >G sB:K1=A.J+J ,A J7J8AJJJ0J1AJ3J4-J!J"J'J=8K1 JJJJJJ J:"J55J::K1= K.#K))K..K1cd}|jDcic] \}}||vs ||}}}|jjjjj |dfd|j i|Scc}}w)zCreate a new run within an existing thread. Args: input (dict): The input data for the new run. Returns: Any: The created run object. )r[r\r3rU run_metadatatruncation_strategymax_prompt_tokensrprI)itemsrFrrrr`rI)rRrallowed_assistant_paramskvparamss r(rz%OpenAIAssistantV2Runnable._create_runs$  $);;=R41aA9Q4Q!Q$RR3t{{'',,33 +  **   Ss A5A5c |jDcic] \}}|dvr||}}}|jdx}r||d<|jjjj d|j |d|}|Scc}}w)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. r[r\r3rrUrIrr])rr}rFrrcreate_and_runrIrRrrrrrrUrs r(rz0OpenAIAssistantV2Runnable._create_thread_and_run2s  1FF qD  #YY'78 8> 8'5F# $5dkk&&55 **     sA?c K|jDcic] \}}|dvr||}}}|jjjjj |dfd|j i|d{Scc}}w7 w)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. )r[r\r3rUrrprIN)rrGrrrr`rI)rRrrrrs r(rz&OpenAIAssistantV2Runnable._acreate_runJs  1XX qD  @T&&++3388?? +  **      sBA<AB7B8 Bc&K|jDcic] \}}|dvr||}}}|jdx}r||d<|jjjj d|j |d|d{}|Scc}}w7 w)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. rrUrNr])rr}rGrrrrIrs r(rz1OpenAIAssistantV2Runnable._acreate_thread_and_run^s  1FF qD  #YY'78 8> 8'5F# $AD%%**22AA **       sBB ABB B)returnr)rZrHr[rHr3Sequence[Union[BaseTool, dict]]r\rHrTzdict[str, float]rFz2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]rU7Optional[Union[AssistantToolResources, dict, NotGiven]]rVzOptional[object]rbrrr )N)rr=rzOptional[RunnableConfig]rbrrr )rZrHr[rHr3rr\rHrGz:Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]rrL)r>rrzDict[str, Any])- __future__rtypingrrrrrr r r &langchain.agents.openai_assistant.baser r langchain_core._apirlangchain_core.callbacksrlangchain_core.loadrlangchain_core.runnablesrrlangchain_core.toolsr%langchain_core.utils.function_callingrpydanticrrrtyping_extensionsrr"r^ropenai.types.beta.assistantrAssistantToolResourcesr)r-r8r@rBrDr]r9r(rs"   W$4%B)H66"&S22* D *, D,,*d 7ddr9