o h@sddlmZmZmZddlmZddlmZmZddl m Z m Z ddl m Z mZddlmZedeed ZGd d d eeefZGd d d eeefZeeefZ eeefZdS))AnyGenericOptional)TypeVar)Agent AgentBase) ModelResponseTResponseInputItem)RunContextWrapperTContext)ToolTAgent)bounddefaultc @seZdZdZdeedeedeede e ddf dd Z deedeed e ddfd d Z deededdfd dZdeedededdfddZdeedededdfddZdeedededdfddZdeededededdf ddZdS) RunHooksBasezA class that receives callbacks on various lifecycle events in an agent run. Subclass and override the methods you need. contextagent system_prompt input_itemsreturnNcdS)z3Called just before invoking the LLM for this agent.NselfrrrrrrQ/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/lifecycle.py on_llm_startzRunHooksBase.on_llm_startresponsecr)z=Called immediately after the LLM call returns for this agent.Nrrrrrrrr on_llm_endzRunHooksBase.on_llm_endcr)zOCalled before the agent is invoked. Called each time the current agent changes.Nrrrrrrron_agent_start%szRunHooksBase.on_agent_startoutputcrz.Called when the agent produces a final output.Nrrrrr$rrr on_agent_end)r!zRunHooksBase.on_agent_end from_agentto_agentcr)zCalled when a handoff occurs.Nr)rrr(r)rrr on_handoff2r!zRunHooksBase.on_handofftoolcrz)Called concurrently with tool invocation.Nrrrrr+rrr on_tool_start;r!zRunHooksBase.on_tool_startresultcrzCalled after a tool is invoked.Nrrrrr+r/rrr on_tool_endDrzRunHooksBase.on_tool_end)__name__ __module__ __qualname____doc__r r rrstrlistr rr r rr#rr'r*r r.r2rrrrr sz          rc @seZdZdZdeededdfddZdeedededdfd d Z deeded eddfd d Z deedede ddfddZ deedede de ddf ddZdeedeedee deeddf ddZdeedeededdfddZdS)AgentHooksBasezA class that receives callbacks on various lifecycle events for a specific agent. You can set this on `agent.hooks` to receive events for that specific agent. Subclass and override the methods you need. rrrNcr)zhCalled before the agent is invoked. Called each time the running agent is changed to this agent.Nrr"rrron_startVszAgentHooksBase.on_startr$crr%rr&rrron_end[r!zAgentHooksBase.on_endsourcecr)zrCalled when the agent is being handed off to. The `source` is the agent that is handing off to this agent.Nr)rrrr<rrrr*drzAgentHooksBase.on_handoffr+crr,rr-rrrr.nr!zAgentHooksBase.on_tool_startr/crr0rr1rrrr2wrzAgentHooksBase.on_tool_endrrcr)z7Called immediately before the agent issues an LLM call.NrrrrrrrzAgentHooksBase.on_llm_startrcr)z=Called immediately after the agent receives the LLM response.Nrrrrrr r!zAgentHooksBase.on_llm_end)r3r4r5r6r r rr:rr;r*r r.r7r2rrr8r rr r rrrrr9Osz          r9N)typingrrrtyping_extensionsrrrritemsr r run_contextr r r+r rrr9RunHooks AgentHooksrrrrs  B F