.Ui,UddlmZddlZddlZddlmZddlmZmZ ddl m Z m Z m Z mZmZmZddlmZddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lm Z dd l!m"Z"ddl#m$Z$m%Z%m&Z&ddl'm(Z(e rddl)m*Z*m+Z+ede Z,edddZ-e ee e,ge fZ.e ee ge fZ/edGddZ0e e0ge(e0fZ1de2d< eGddeee-fZ3eddddd d%d Z4eddddd! d&d"Z4eddddd! d'd#Z4 d( d)d$Z4y)*) annotationsN) Awaitable) dataclassreplace) TYPE_CHECKINGAnyCallableGenericcastoverload) TypeAdapter) TypeAliasTypeVar)ModelBehaviorError UserError)RunItemTResponseInputItem)RunContextWrapperTContext)ensure_strict_json_schema) SpanError)_error_tracing_json _transforms)MaybeAwaitable)Agent AgentBase THandoffInput)defaultTAgentAgentBase[Any]z Agent[Any])boundr T)frozencJeZdZUded< ded< ded< dZded< d d Zy) HandoffInputDataz$str | tuple[TResponseInputItem, ...] input_historyztuple[RunItem, ...]pre_handoff_items new_itemsNzRunContextWrapper[Any] | None run_contextc t|fi|S)z Make a copy of the handoff input data, with the given arguments changed. For example, you could do: ``` new_handoff_input_data = handoff_input_data.clone(new_items=()) ``` )dataclasses_replace)selfkwargss k/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/handoffs.pyclonezHandoffInputData.clone:s#42622)r.rreturnr&)__name__ __module__ __qualname____annotations__r*r0r1r/r&r&"s>77+*#" 26K.5 3r1r&rHandoffInputFilterceZdZUdZded< ded< ded< ded< ded < d Zd ed < d Zded< d Zded< ddZe ddZ e ddZ y )Handoffa1A handoff is when an agent delegates a task to another agent. For example, in a customer support scenario you might have a "triage agent" that determines which agent should handle the user's request, and sub-agents that specialize in different areas like billing, account management, etc. str tool_nametool_descriptionzdict[str, Any]input_json_schemaz:Callable[[RunContextWrapper[Any], str], Awaitable[TAgent]]on_invoke_handoff agent_nameNzHandoffInputFilter | None input_filterTboolstrict_json_schemazObool | Callable[[RunContextWrapper[Any], AgentBase[Any]], MaybeAwaitable[bool]] is_enabledcDtjd|jiS)N assistant)jsondumpsname)r-agents r/get_transfer_messagezHandoff.get_transfer_messageszz; 344r1cFtjd|jS)N transfer_to_)rtransform_string_function_stylerIclsrJs r/default_tool_namezHandoff.default_tool_names::\%**;VWWr1cBd|jd|jxsdS)NzHandoff to the z agent to handle the request. )rIhandoff_descriptionrOs r/default_tool_descriptionz Handoff.default_tool_descriptions-ejj\)G((.B/ 1 r1)rJr"r2r;) r3r4r5__doc__r6rArCrDrK classmethodrQrUr7r1r/r:r:Is N;B%%RQO<.2L+2  $# _.5XX  r1r:)tool_name_overridetool_description_overriderArDcyNr7)rJrXrYrArDs r/handoffr\s*-r1)rYrXrArDcyr[r7)rJ on_handoff input_typerYrXrArDs r/r\r\s*-r1cyr[r7)rJr^rYrXrArDs r/r\r\s*-r1c x rs r rJdftsJdtj}t|jdk7r t dt  j}n>d i}8tj}t|jdk7r t d d d  fd } |xstj} |xstj} t|}d fd } t| | || |jtr| S S)aCreate a handoff from an agent. Args: agent: The agent to handoff to, or a function that returns an agent. tool_name_override: Optional override for the name of the tool that represents the handoff. tool_description_override: Optional override for the description of the tool that represents the handoff. on_handoff: A function that runs when the handoff is invoked. input_type: the type of the input to the handoff. If provided, the input will be validated against this type. Only relevant if you pass a function that takes an input. input_filter: a function that filters the inputs that are passed to the next agent. is_enabled: Whether the handoff is enabled. Can be a bool or a callable that takes the run context and agent and returns whether the handoff is enabled. Disabled handoffs are hidden from the LLM at runtime. zBYou must provide either both on_handoff and input_type, or neitherNzon_handoff must be callablez5on_handoff must take two arguments: context and inputrz*on_handoff must take one argument: contextcK|-tjtddditdt j |d}t tt}tj|r|||d{S|||S?t t}tj|r||d{S||S7T7w)Nz6Handoff function expected non-null input, but got Nonedetailszinput_json is None)messagedataF)json_str type_adapterpartial) rattach_error_to_current_spanrrr validate_jsonr OnHandoffWithInputrinspectiscoroutinefunctionOnHandoffWithoutInput) ctx input_jsonvalidated_input input_func no_input_funcrJr_r^rhs r/_invoke_handoffz handoff.._invoke_handoffs  !l&>!;; X')=> ))abb#11#)O 0?LJ**:6 o666 30  # !6 CM**=9#C((( c" 7 )s%BC CAC  C C C cKddlm}tsJdt||sJd||}t j |r |d{S|S7w)Nr)rz is_enabled must be callable herezCan't handoff to a non-Agent)rJrcallable isinstancerm isawaitable)rp agent_baserresultrDs r/ _is_enabledzhandoff.._is_enableds`  #G%GG#*e,L.LL,C,   v &<   sAAAA)r<r=r>r?rAr@rDr[)rpRunContextWrapper[Any]rq str | Noner2Agent[TContext])rpr}rzr"r2rB) rwrm signaturelen parametersrr json_schemar:rQrUrrI)rJrXrYr^r_rArDsigr>rur<r=r|rhs` `` ` @r/r\r\sg2 :zjL I #B%BB# + s~~ ! #ST T":. (446   !##J/C3>>"a' LMM?C #1; @#Fg&?&?&FI0[G4T4TUZ4[22CD  )+)!::"*:"6; =G r1) rJrrXr~rYr~rA5Callable[[HandoffInputData], HandoffInputData] | NonerDKbool | Callable[[RunContextWrapper[Any], Agent[Any]], MaybeAwaitable[bool]]r2"Handoff[TContext, Agent[TContext]])rJrr^z!OnHandoffWithInput[THandoffInput]r_ztype[THandoffInput]rYr~rXr~rArrDrr2r)rJrr^rorYr~rXr~rArrDrr2r)NNNNNT)rJrrXr~rYr~r^z@OnHandoffWithInput[THandoffInput] | OnHandoffWithoutInput | Noner_ztype[THandoffInput] | NonerArrDzPbool | Callable[[RunContextWrapper[Any], Agent[TContext]], MaybeAwaitable[bool]]r2r)5 __future__rrmrGcollections.abcr dataclassesrrr,typingrrr r r r pydanticr typing_extensionsrr exceptionsrritemsrrr*rr strict_schemar tracing.spansrutilrrr util._typesrrJrrrr!rlror&r8r6r:r\r7r1r/rs" %AHH 05.44$44''5  !1< H05}EsJK "3C"8!93!>? $333D!)*:);^L\=])] ^I^F C gh&'C  C L &*,0JN^b - -#- * - H - \ -(- - -1%)JN^b -  -2 -$ - * - # -H -\ -( -  - -1%)JN^b- -&- * - # - H -\-(- -&*,0SW-1JNRVh h"h *hQ h + h H hPh(hr1