.UiddlmZmZmZddlmZddlmZmZddl m Z m Z ddl m Z mZddlmZedee ZGd d eeefZGd d eeefZeeefZ eeefZy))AnyGenericOptional)TypeVar)Agent AgentBase) ModelResponseTResponseInputItem)RunContextWrapperTContext)ToolTAgent)bounddefaultc eZdZdZdeedeedeede e ddf dZ deedeed e ddfd Z deededdfd Zdeeded eddfd ZdeedededdfdZdeedededdfdZdeededededdf dZy) RunHooksBasezA class that receives callbacks on various lifecycle events in an agent run. Subclass and override the methods you need. contextagent system_prompt input_itemsreturnNc Kyw)z3Called just before invoking the LLM for this agent.Nselfrrrrs l/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/lifecycle.py on_llm_startzRunHooksBase.on_llm_start  responsec Kyw)z=Called immediately after the LLM call returns for this agent.Nrrrrr!s r on_llm_endzRunHooksBase.on_llm_end  r c Kyw)zOCalled before the agent is invoked. Called each time the current agent changes.Nrrrrs ron_agent_startzRunHooksBase.on_agent_start%s  r outputc Kywz.Called when the agent produces a final output.Nrrrrr)s r on_agent_endzRunHooksBase.on_agent_end)r%r from_agentto_agentc Kyw)zCalled when a handoff occurs.Nr)rrr.r/s r on_handoffzRunHooksBase.on_handoff2r%r toolc Kywz)Called concurrently with tool invocation.Nrrrrr2s r on_tool_startzRunHooksBase.on_tool_start;r%r resultc KywzCalled after a tool is invoked.Nrrrrr2r7s r on_tool_endzRunHooksBase.on_tool_endDrr )__name__ __module__ __qualname____doc__r r rrstrlistr rr r$rr(rr-r1rr6r;rrrr so "8, X  }  ,-     "8, X      ,=h,G PV [_  "8,       "8,       "8,       "8,        rBrc eZdZdZdeededdfdZdeedededdfdZ deeded eddfd Z deeded e ddfd Z deeded e d e ddf dZdeedeedee deeddf dZdeedeededdfdZy)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. rrrNc Kyw)zhCalled before the agent is invoked. Called each time the running agent is changed to this agent.Nrr's ron_startzAgentHooksBase.on_startVs  r r)c Kywr+rr,s ron_endzAgentHooksBase.on_end[r%r sourcec Kyw)zrCalled when the agent is being handed off to. The `source` is the agent that is handing off to this agent.Nr)rrrrIs rr1zAgentHooksBase.on_handoffdrr r2c Kywr4rr5s rr6zAgentHooksBase.on_tool_startnr%r r7c Kywr9rr:s rr;zAgentHooksBase.on_tool_endwrr rrc Kyw)z7Called immediately before the agent issues an LLM call.Nrrs rrzAgentHooksBase.on_llm_startrr r!c Kyw)z=Called immediately after the agent receives the LLM response.Nrr#s rr$zAgentHooksBase.on_llm_endr%r )r<r=r>r?r r rrFrrHr1rr6r@r;rrrAr rr r$rrBrrDrDOsn  &7&A & UY  "8,       "8,       "8,       "8,         "8, X  }  ,-     "8, X     rBrDN)typingrrrtyping_extensionsrrrr itemsr r run_contextr r r2rrrrDRunHooks AgentHooksrrBrrUs|))%#44 I >? 78V+,? DC WXv-.C L % (# HeO , rB