Ng|_>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'd dZ(d!dZ)d"dZ*d#dZ+eGdde 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 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 l/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/agents/openai_assistant/base.py_get_openai_clientr+s  v}m_-MNNNN  P     =    A3 AAAopenai.AsyncOpenAIc 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. rNrrr r"r#)r$ AsyncOpenAIr&r'r(s r*_get_openai_async_clientr07s  !v!=/2RSSSS  P     =   r,file_idslistcPg}|D] }||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)r1 attachmentsids r*"_convert_file_ids_into_attachmentsr<Ps]K  !=1Fassistants_builtin_toolss r*_is_assistants_builtin_toolrFes; P4 7 t^ 7 &\5 5r=Dict[str, Any]cBt|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. )rFr)r>s r*_get_assistants_toolrIzs'#4((, %d+++r=ceZdZUdZeeZded< dZded< ded< d Z d ed < d Z d ed< e dd3dZ e iddddd4d#Z d5d6d)Ze ddd*d7d,Z d5d6d-Zd8d.Zd9d0Zd8d1Zd9d2ZdS):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_msFr@as_agentafter)moderrc`|j&ddl}|jj}|j||_|S)z?Validate that the async client is set, otherwise initialize it.Nr)api_key)rNr$rMrWr/)selfr$rWs r*validate_async_clientz/OpenAIAssistantV2Runnable.validate_async_clients=   $ MMMk)G 2 27 C C CD  r=) model_kwargsrMtool_resources extra_bodyname instructionsr8Sequence[Union[BaseTool, dict]]modelrZdict[str, float]2Optional[Union[openai.OpenAI, openai.AzureOpenAI]]r[7Optional[Union[AssistantToolResources, dict, NotGiven]]r\Optional[object]kwargsr c |p t}|ddlm} | }|jjjd||d|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_GIVENc,g|]}t|SrI.0r>s r* z>OpenAIAssistantV2Runnable.create_assistant..#s!@@@$'--@@@r=)r]r^r8r[r`r\)rPrMrj)r+ openai._typesrhr assistantscreater;) clsr]r^r8r`rZrMr[r\rerh assistants r*create_assistantz*OpenAIAssistantV2Runnable.create_assistantsD/-//  ! / / / / / /&N1FK*1 %@@%@@@)!     sF VFFvFFFr=inputrDconfigOptional[RunnableConfig]r c t|}tj|d|d|d}|t |||dp|}t|dg}|dg|z} |jrR|d r=| |d }|j j j j jdi|} nd |vrNd |d ||d dg|dd} ||| } nd|vrc|j j j j|d |d d ||d } ||} n!|j j j j jdi|} || j| j} n)#t,$r} || | d} ~ wwxYw || } || | S#t,$r0} || | | 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_namer]r1r:intermediate_steps thread_idusercontentmessage_metadatarolerr:r{thread_metadatamessagesr{run_idrrr:r{Nr{rj)rr configuregeton_chain_startrget_namer<rS_parse_intermediate_stepsrMrthreadsrunssubmit_tool_outputs_create_thread_and_runrrq _create_run _wait_for_runr;r BaseExceptionon_chain_error _get_response on_chain_endrDrXrurvrecallback_manager run_managerfilesr: tool_outputsrunthread_r)responses r*invokez OpenAIAssistantV2Runnable.invoke+s Bv&&*4"(**["9"9#ZZ//!'J!7!7    '55 $KKVZZ %;%;%Nt}}6  36::j"3M3MNNjj33e; & } Q+?!@!@ Q#==./   Hdk&.3GWW,WWE))%+',Y'7+6(- 2D(E(E !!& *; < <  11%@@&&K$,5<<+&!), +"YY'9:: =&&u--Hdk&.3GPP%PP$$SVS];;CC     & &q ) ) )G  ))#..H  $ $X . . .O      & &q388:: & > > >G s1)EH== I#II#'J K +KK )rNr[.s!EEEt,T22EEEr=)r]r^r8r[r`)rPrNrj)r0rorhrrprqr;) rrr]r^r8r`rNr[rerh openai_toolsrss r*acreate_assistantz+OpenAIAssistantV2Runnable.acreate_assistants<$A'?'A'A  ! / / / / / /&NEEuEEE &+6==%) >        sR <RR6RRRr=c K|pi}tj|d|d|d}|t |||dp|}t |dg}|dg|z} |jrX|d rC||d }|j j j j j di|d {V} nd |vrTd |d ||ddg|dd} ||| d {V} nd|vro|j j j j|d |d d ||dd {V} ||d {V} n'|j j j j j di|d {V} || j| jd {V} n)#t*$r} || | d } ~ wwxYw || } || | S#t*$r0} || | | d } ~ wwxYw)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. ryrzr{r|rrr1r:rNrrrrrrrrrrrj)rrrrrrr<rSrrNrrrr_acreate_thread_and_runrrq _acreate_run_await_for_runr;rrrrrrDrs r*ainvokez!OpenAIAssistantV2Runnable.ainvokesB2*4"(**["9"9#ZZ//!'J!7!7    '55 $KKVZZ %;%;%Nt}}6  36::j"3M3MNNjj33e; * }$ +?!@!@$ #==./   TD-2:?S"E))%+',Y'7+6(- 2D(E(E !!& *; < <  !88GGGGGGGG&&+08AHH+&!), +"YY'9:: I!--e44444444TD-2:?S++CFCMBBBBBBBBCC     & &q ) ) )G  ))#..H  $ $X . . .O      & &q388:: & > > >G s1 E7I I>"I99I>J.. K(8+K##K(cdfd|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. )r^r`r8r[ run_metadatatruncation_strategymax_prompt_tokensc$i|] \}}|v || Srjrj)rmkvallowed_assistant_paramss r* z9OpenAIAssistantV2Runnable._create_run..+s*RRR41aA9Q4Q4Q!Q4Q4Q4Qr=rrP)itemsrMrrrrqrP)rXruparamsrs @r*rz%OpenAIAssistantV2Runnable._create_runss$  SRRR5;;==RRR3t{',3 +   *    r=rcd|D}|dx}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. c"i|] \}}|dv || S)r^r`r8rrjrmrrs r*rzDOpenAIAssistantV2Runnable._create_thread_and_run..<4   1FFF qFFFr=r[rPrrj)rrrMrrcreate_and_runrPrXrurrr[rs r*rz0OpenAIAssistantV2Runnable._create_thread_and_run2s       #YY'788 8> 6'5F# $5dk&5 *      r=cKd|D}|jjjjj|dfd|ji|d{VS)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. c"i|] \}}|dv || S))r^r`r8r[rrjrs r*rz:OpenAIAssistantV2Runnable._acreate_run..Ss4   1XXX qXXXr=rrPN)rrNrrrrqrP)rXrurs r*rz&OpenAIAssistantV2Runnable._acreate_runJs       @T&+38? +   *          r=cKd|D}|dx}r||d<|jjjjd|j|d|d{V}|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. c"i|] \}}|dv || Srrjrs r*rzEOpenAIAssistantV2Runnable._acreate_thread_and_run..hrr=r[rNrj)rrrNrrrrPrs r*rz1OpenAIAssistantV2Runnable._acreate_thread_and_run^s       #YY'788 8> 6'5F# $AD%*2A *            r=)rr)r]rOr^rOr8r_r`rOrZrarMrbr[rcr\rdrerrr )N)rurDrvrwrerrr )r]rOr^rOr8r_r`rOrNrr[rcrerrr )rurDrr)rurDrrDrr)__name__ __module__ __qualname____doc__rr+rM__annotations__rNrRrSrrY classmethodrtrrrrrrrrjr=r*rKrKsQQf%(:;;;F;;;;'L-#N####EHF_'"""#"*,EIRV'+/G/G/G/G/G[/Gd?C\\\\\| RV+S+S+S+S+S[+S\?C`````D    20    (r=rK)rr)rr-)r1r2rr2)r>r?rr@)r>r?rrG)- __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$roropenai.types.beta.assistantrAssistantToolResourcesr+r0r<rFrIrKrjr=r*rs#""""""                    WVVVVVVV$$$$$$444444%%%%%%BBBBBBBB))))))HHHHHH6666666666""""""TMMM&&&&&&SSSSSS22**,,,,*ddddd 7dddddr=