.hK)UddlmZddlZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZmZmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-e rddl.m/Z/ddlm0Z0ddl1m2Z2eGddZ3e e$e%e4e'ge-e3fZ5de6d< GddeZ7GddeZ8eGdd e e%Z9dS)!) annotationsN) Awaitable) dataclassfield) TYPE_CHECKINGAnyCallableGenericLiteralcast) NotRequired TypeAlias TypedDict)AgentOutputSchemaBase)InputGuardrailOutputGuardrail)Handoff) ItemHelpers)logger)MCPUtil) ModelSettings)Model)RunContextWrapperTContext)FunctionToolResultTool function_tool) _transforms)MaybeAwaitable) AgentHooks) MCPServer) RunResultc*eZdZUded< dZded<dS)ToolsToFinalOutputResultboolis_final_outputNz Any | None final_output)__name__ __module__ __qualname____annotations__r(I/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/agent.pyr%r%s: $L####r.r%rToolsToFinalOutputFunctionceZdZUded<dS) StopAtToolsz list[str]stop_at_tool_namesN)r)r*r+r,r-r.r/r2r24s!!!!VVr.r2ceZdZUdZded<dS) MCPConfigzConfiguration for MCP servers.zNotRequired[bool]convert_schemas_to_strictN)r)r*r+__doc__r,r-r.r/r5r59s(((0000r.r5ceZdZUdZded< dZded< dZded< ee Z d ed < dZ d ed < ee Z ded< ee Z ded< ee Zded< ed Zded< ee Zded< ee Zded< dZded< dZded< dZd ed!< d"Zd#ed$< d5d)Z d6d7d/Zd8d2Zd9d3Zd9d4ZdS):AgentaAn agent is an AI model configured with instructions, tools, guardrails, handoffs and more. We strongly recommend passing `instructions`, which is the "system prompt" for the agent. In addition, you can pass `handoff_description`, which is a human-readable description of the agent, used when the agent is used inside tools/handoffs. Agents are generic on the context type. The context is a (mutable) object you create. It is passed to tool functions, handoffs, guardrails, etc. strnameNzZstr | Callable[[RunContextWrapper[TContext], Agent[TContext]], MaybeAwaitable[str]] | None instructions str | Nonehandoff_description)default_factoryz$list[Agent[Any] | Handoff[TContext]]handoffszstr | Model | Nonemodelrmodel_settings list[Tool]toolszlist[MCPServer] mcp_serversctSN)r5r-r.r/zAgent.s )++r.r5 mcp_configzlist[InputGuardrail[TContext]]input_guardrailszlist[OutputGuardrail[TContext]]output_guardrailsz(type[Any] | AgentOutputSchemaBase | None output_typezAgentHooks[TContext] | Nonehooks run_llm_againzYLiteral['run_llm_again', 'stop_on_first_tool'] | StopAtTools | ToolsToFinalOutputFunctiontool_use_behaviorTr&reset_tool_choicekwargsrreturnAgent[TContext]c &tj|fi|S)zMake a copy of the agent, with the given arguments changed. For example, you could do: ``` new_agent = agent.clone(instructions="New instructions") ``` ) dataclassesreplace)selfrQs r/clonez Agent.clones "4226222r. tool_nametool_descriptioncustom_output_extractor,Callable[[RunResult], Awaitable[str]] | Nonerc|t|ptjj|pdd fd }|S) aRTransform 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. ) name_overridedescription_overridecontextrinputr:rRcKddlm}|||jd{V}r|d{VStj|jS)Nr)Runner)starting_agentrbra)runrdrartext_message_outputs new_items)rarbrdoutputr[rWs r/ run_agentz Agent.as_tool..run_agents $ # # # # #!::#&F ' =44V<<<<<<<<<3F4DEE Er.)rarrbr:rRr:)rrtransform_string_function_styler;)rWrYrZr[rjs` ` r/as_toolz Agent.as_toolss, #]{'RSWS\']']!1!7R    F F F F F F    Fr. run_contextRunContextWrapper[TContext]cKt|jtr|jSt|jr|t j|jr:t tt|||d{VSt t|||S|jtj d|jdS)z$Get the system prompt for the agent.Nz1Instructions must be a string or a function, got ) isinstancer<r:callableinspectiscoroutinefunctionr rrerror)rWrms r/get_system_promptzAgent.get_system_prompts d' - - b$ $ d' ( ( b*4+<== G!)C.$2C2CKQU2V2VWWWWWWWWWC!2!2;!E!EFFF   * L`TM^`` a a atr.c|K|jdd}tj|j|d{VS)z1Fetches the available tools from the MCP servers.r6FN)rIgetrget_all_function_toolsrE)rWr6s r/ get_mcp_toolszAgent.get_mcp_toolssI$(O$7$78SUZ$[$[!3D4DF_`````````r.cNK|d{V}||jzS)z8All agent tools, including MCP tools and function tools.N)ryrD)rW mcp_toolss r/ get_all_toolszAgent.get_all_toolss6,,........ 4:%%r.)rQrrRrSrG)rYr=rZr=r[r\rRr)rmrnrRr=)rRrC)r)r*r+r7r,r<r>rlistr@rArrBrDrErIrJrKrLrMrOrPrXrlruryr|r-r.r/r9r9BsfIII      '+****6;U45P5P5PHPPPP !%E$$$$ %*E-$H$H$HNHHHHd333E33331#(5#>#>#>K>>>>"E2E2EFFFJFFFF(7rs"""""""%%%%%%((((((((GGGGGGGGGGGGGGGG??????????//////66666666))))))######444444449999999999''''''"%%%%%%!!!!!!          )1x $'9":;+,.) WWWWW)WWW   y&y&y&y&y&GH y&y& y&y&y&r.