o hxM@s UddlmZddlZddlZddlZddlmZddlmZmZddl m Z m Z m Z m Z mZmZddlmZddlmZmZmZdd lmZdd lmZmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5ddl6m7Z7ddl8m9Z9e rddl:m;Z;mm?Z?ddl@mAZAddlBmCZCeGdddZDe e/e0eEe3ge9eDfZFdeGd< Gd d!d!eZHGd"d#d#eZIeGd$d%d%e e0ZJeGd&d'd'eJe e0ZKdS)() annotationsN) Awaitable) dataclassfield) TYPE_CHECKINGAnyCallableGenericLiteralcast)ResponsePromptParam) NotRequired TypeAlias TypedDictAgentOutputSchemaBase)InputGuardrailOutputGuardrail)Handoff) ItemHelpers)logger)MCPUtil) ModelSettings)get_default_model_settings!gpt_5_reasoning_settings_requiredis_gpt_5_defaultModel)DynamicPromptFunctionPrompt PromptUtil)RunContextWrapperTContext) FunctionToolFunctionToolResultTool function_tool) _transforms)MaybeAwaitable) AgentHooksRunHooks) MCPServer)Session) RunResult) RunConfigc@s$eZdZUded< dZded<dS)ToolsToFinalOutputResultboolis_final_outputNz Any | None final_output)__name__ __module__ __qualname____annotations__r3r8r8M/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/agent.pyr0(s  r0rToolsToFinalOutputFunctionc@seZdZUded<dS) StopAtToolsz list[str]stop_at_tool_namesN)r4r5r6r7r8r8r8r9r;>s r;c@seZdZUdZded<dS) MCPConfigzConfiguration for MCP servers.zNotRequired[bool]convert_schemas_to_strictN)r4r5r6__doc__r7r8r8r8r9r=Cs r=c@s~eZdZUdZded< dZded< eedZded < eedZ d ed < ed d dZ ded< dddZ dddZ dS) AgentBasez+Base class for `Agent` and `RealtimeAgent`.strnameN str | Nonehandoff_descriptiondefault_factory list[Tool]toolszlist[MCPServer] mcp_serverscCstSN)r=r8r8r8r9eszAgentBase.r= mcp_config run_contextRunContextWrapper[TContext]returncs(|jdd}t|j|||IdHS)z1Fetches the available tools from the MCP servers.r>FN)rLgetrget_all_function_toolsrI)selfrMr>r8r8r9 get_mcp_toolshs   zAgentBase.get_mcp_toolscsdIdH}d fdd tjfdd jDIdH}d d tj|D}g||S) z8All agent tools, including MCP tools and function tools.Ntoolr&rOr1csNt|tsdS|j}t|tr|S|}t|r#t|IdHSt|S)NT) isinstancer$ is_enabledr1inspect isawaitable)rTattrres)rMrRr8r9_check_tool_enabledss    z4AgentBase.get_all_tools.._check_tool_enabledc3s|]}|VqdSrJr8).0t)r[r8r9 sz*AgentBase.get_all_tools..cSsg|]\}}|r|qSr8r8)r\r]okr8r8r9 sz+AgentBase.get_all_tools..)rTr&rOr1)rSasynciogatherrHzip)rRrM mcp_toolsresultsenabledr8)r[rMrRr9 get_all_toolsos   zAgentBase.get_all_tools)rMrNrOrG) r4r5r6r?r7rDrlistrHrIrLrSrgr8r8r8r9r@Ls    r@c@seZdZUdZdZded< dZded< eedZ ded < dZ d ed < ee dZ d ed < eedZ ded< eedZded< dZded< dZded< dZded< dZded< ddZd>d"d#Z        d?d@d5d6ZdAd9d:ZdBdtdt|j jt|j t sOtdt|j jt|j t s`tdt|j j|jdur{t|jts{t|js{tdt|jj|jdurt|jst|jd std t|jjt|jt std t|jj|jdurt|jtsd d lm}t|j|stdt|jjt|jtstdt|jj|jdurtdurt|jtdust|jdur|jtkrt|_t|jt stdt|jjt|jt s!tdt|jj|jdurId dlm}t|jt|fsI||jdusItdt|jj|j durgd dl!m"}t|j |sgtdt|j jt|j#trt|j#dvst|j#t st|j#stdt|j#jt|j$t%stdt|j$jdS)Nr) get_originz!Agent name must be a string, got z8Agent handoff_description must be a string or None, got z Agent tools must be a list, got z&Agent mcp_servers must be a list, got z%Agent mcp_config must be a dict, got zt|ptj|p d|dd fdd } | S) aDTransform this agent into a tool, callable by other agents. This is different from handoffs in two ways: 1. In handoffs, the new agent receives the conversation history. In this tool, the new agent receives generated input. 2. In handoffs, the new agent takes over the conversation. In this tool, the new agent is called as a tool, and the conversation is continued by the original agent. Args: tool_name: The name of the tool. If not provided, the agent's name will be used. tool_description: The description of the tool, which should indicate what it does and when to use it. custom_output_extractor: A function that extracts the output from the agent. If not provided, the last message from the agent will be used. is_enabled: Whether the tool is enabled. Can be a bool or a callable that takes the run context and agent and returns whether the tool is enabled. Disabled tools are hidden from the LLM at runtime. ) name_overridedescription_overriderVcontextr"inputrArOc sdddlm}m}durn|}|j||j|d IdH}r,|IdHSt|jS)Nr)DEFAULT_MAX_TURNSRunner) starting_agentrrrrrrrrr)runrrrrtext_message_outputs new_items)rrrrresolved_max_turnsoutputrrrrrrrrRrr8r9 run_agents"   z Agent.as_tool..run_agentN)rr"rrArOrA)r'r(transform_string_function_stylerB) rRrrrrVrrrrrrrrr8rr9as_tool~s!z Agent.as_toolrMrNcst|jtr |jSt|jrPt|j}t|j}t |dkr3t dt |ddd|Dt |jrGt t t|||IdHSt t|||S|jduratdt|jjdS)NzR'instructions' callable must accept exactly 2 arguments (context, agent), but got z: cSsg|]}|jqSr8)rB)r\pr8r8r9r`sz+Agent.get_system_prompt..z:Instructions must be a string or a callable function, got )rUrjrAr}rW signaturerh parametersvalueslenrziscoroutinefunctionr rrerrorr{r4)rRrMsigparamsr8r8r9get_system_prompts0        zAgent.get_system_promptResponsePromptParam | Nonecst|j||IdHS)zGet the prompt for the agent.N)r!to_model_inputrk)rRrMr8r8r9 get_promptszAgent.get_prompt)rrrOr)NTNNNNNN)rrCrrCrrrVrrrrrrrrrrCrrCrrrOr&)rMrNrOrC)rMrNrOr)r4r5r6r?rjr7rkrrhrlrmrrnrorprqrrrtrurrrrrr8r8r8r9risP          =ri)L __future__rrarrWcollections.abcrrrryrrrr r r ,openai.types.responses.response_prompt_paramr typing_extensionsr rrrr guardrailrrrlritemsrrmcprrnrmodels.default_modelsrrrrrpromptsrr r!rMr"r#rTr$r%r&r'utilr( util._typesr)rr*r+r,memory.sessionr-resultr.rr/r0rhr:r7r;r=r@rir8r8r8r9sV                   7