.hz ddlmZmZddlmZddlmZmZddlm Z GddeeZ Gdd eeZ d S) )AnyGeneric)Agent)RunContextWrapperTContext)Toolc eZdZdZdeedeeddfdZdeedeededdfdZ deed eed eeddfd Z deedeed e ddfd Z deedeed e de ddf dZdS)RunHookszA class that receives callbacks on various lifecycle events in an agent run. Subclass and override the methods you need. contextagentreturnNc KdS)zOCalled before the agent is invoked. Called each time the current agent changes.Nselfr r s M/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/lifecycle.pyon_agent_startzRunHooks.on_agent_start s  outputc KdSz.Called when the agent produces a final output.Nrrr r rs r on_agent_endzRunHooks.on_agent_end  r from_agentto_agentc KdS)zCalled when a handoff occurs.Nr)rr rrs r on_handoffzRunHooks.on_handoffrrtoolc KdSz Called before a tool is invoked.Nrrr r r s r on_tool_startzRunHooks.on_tool_start%rrresultc KdSzCalled after a tool is invoked.Nrrr r r r%s r on_tool_endzRunHooks.on_tool_end.  r)__name__ __module__ __qualname____doc__rrrrrrrr r$strr)rrrr r sm (2 ;@?      "8, X         "8, (O /        "8, X         "8, X            rr c eZdZdZdeedeeddfdZdeedeededdfdZ deedeed eeddfd Z deedeed e ddfd Z deedeed e d e ddf dZdS) AgentHookszA 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. r r rNc KdS)zhCalled before the agent is invoked. Called each time the running agent is changed to this agent.Nrrs ron_startzAgentHooks.on_start@s  rrc KdSrrrs ron_endzAgentHooks.on_endErrsourcec KdS)zrCalled when the agent is being handed off to. The `source` is the agent that is handing off to this agent.Nr)rr r r6s rrzAgentHooks.on_handoffNr*rr c KdSr"rr#s rr$zAgentHooks.on_tool_startXrrr%c KdSr'rr(s rr)zAgentHooks.on_tool_endar*r)r+r,r-r.rrrr3rr5rr r$r/r)rrrr1r19sm  &7&A %PX/ ^b     "8, X         "8, X h        "8, X         "8, X            rr1N) typingrrr r run_contextrrr r r r1rrrr<s44444444. . . . . wx . . . b0 0 0 0 0 "0 0 0 0 0 r