.hS#NUddlmZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZdd lmZddl m!Z!m"Z"m#Z#erddl$m%Z%ede Z&e ee e&ge fZ'e ee ge fZ(edGddZ)e e)ge)fZ*de+d< eGdde eZ,e ddddd0d$Z-e dddd%d1d*Z-e dddd%d2d,Z- d3d4d/Z-dS)5) annotationsN) Awaitable) dataclass) TYPE_CHECKINGAnyCallableGenericcastoverload) TypeAdapter) TypeAliasTypeVar)ModelBehaviorError UserError)RunItemTResponseInputItem)RunContextWrapperTContext)ensure_strict_json_schema) SpanError)_error_tracing_json _transforms)Agent THandoffInput)defaultT)frozenc2eZdZUded< ded< ded<dS)HandoffInputDataz$str | tuple[TResponseInputItem, ...] input_historyztuple[RunItem, ...]pre_handoff_items new_itemsN)__name__ __module__ __qualname____annotations__L/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/handoffs.pyr r sH7777+***#"""r)r r HandoffInputFilterceZdZUdZded< ded< ded< ded< ded < d Zd ed < d Zded< ddZeddZ eddZ d S)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_schemazCCallable[[RunContextWrapper[Any], str], Awaitable[Agent[TContext]]]on_invoke_handoff agent_nameNzHandoffInputFilter | None input_filterTboolstrict_json_schemaagent Agent[Any]returncd|jd}|S)Nz{'assistant': 'z'})name)selfr7bases r*get_transfer_messagezHandoff.get_transfer_messagees1%*111 r)c:tjd|jS)N transfer_to_)rtransform_string_function_styler;clsr7s r*default_tool_namezHandoff.default_tool_nameis:;V%*;V;VWWWr)c*d|jd|jpdS)NzHandoff to the z agent to handle the request. )r;handoff_descriptionrBs r*default_tool_descriptionz Handoff.default_tool_descriptionms/ 1ej 1 1(.B 1 1 r))r7r8r9r.) r$r%r&__doc__r'r4r6r> classmethodrDrHr(r)r*r-r-4s NNN;B%%%%[ZZZOOO<.2L2222  $####XXX[X   [   r)r-)tool_name_overridetool_description_overrider4r7Agent[TContext]rK str | NonerLr45Callable[[HandoffInputData], HandoffInputData] | Noner9Handoff[TContext]cdSNr()r7rKrLr4s r*handoffrSus r))rLrKr4 on_handoff!OnHandoffWithInput[THandoffInput] input_typetype[THandoffInput]cdSrRr()r7rTrVrLrKr4s r*rSrSs r)OnHandoffWithoutInputcdSrRr()r7rTrLrKr4s r*rSrSs r)@OnHandoffWithInput[THandoffInput] | OnHandoffWithoutInput | Nonetype[THandoffInput] | Nonec| rsr r Jdxts Jdtj}t|jdkrt dt  }nAd i};tj}t|jdkrt d dd fd }|pt } |pt } t|}t| | |||j 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. z@You must provide either both on_input and input_type, or neitherNzon_handoff must be callablez5on_handoff must take two arguments: context and inputrz*on_handoff must take one argument: contextctxRunContextWrapper[Any] input_jsonrNr9r8cK|4tjtddditdt j|d}t tt}tj |r|||d{VnU|||nHFt t}tj |r||d{Vn ||S)Nz6Handoff function expected non-null input, but got Nonedetailszinput_json is None)messagedataF)json_str type_adapterpartial) rattach_error_to_current_spanrrr validate_jsonr OnHandoffWithInputrinspectiscoroutinefunctionrY) r_ravalidated_input input_func no_input_funcr7rVrTrgs r*_invoke_handoffz handoff.._invoke_handoffsG  !l&>!; X')=> ))abbb#1#)O 0?LLJ*:66 1 jo6666666666 30000  # !6 CCM*=99 ##mC(((((((((( c""" r))r/r0r1r2r4r3rR)r_r`rarNr9r8) callablerl signaturelen parametersrr json_schemar-rDrHrr;) r7rKrLrTrVr4sigr1rqr/r0rgs ` `` @r*rSrSs( :zjJ ##BB%BBBB ++ s~  ! # #STT T":.. (4466   !#J//C3>""a'' LMMM?C@#Fg&?&?&F&FI0[G4T4TUZ4[4[22CDD )+)!:    r)) r7rMrKrNrLrNr4rOr9rP)r7rMrTrUrVrWrLrNrKrNr4rOr9rP) r7rMrTrYrLrNrKrNr4rOr9rP)NNNNN)r7rMrKrNrLrNrTr[rVr\r4rOr9rP). __future__rrlcollections.abcr dataclassesrtypingrrrr r r pydanticr typing_extensionsr r exceptionsrritemsrr run_contextrr strict_schemar tracing.spansrutilrrrr7rrrkrYr r+r'r-rSr(r)r*rs """""""%%%%%%!!!!!!HHHHHHHHHHHHHHHH 0000000055555555........44444444444444$$$$$$4444444444555 05}EsJK "3C"8!93!>? $$!)*:);=M)M NNNNNF = = = = = gh= =  = @ &*,0JN   -1%)JN  -1%)JN  &*,0SW-1JN VVVVVVVr)