o h,@sUddlmZddlZddlZddlmZddlmZmZ 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 ryddl)m*Z*m+Z+ede dZ,eddddZ-e ee e,ge fZ.e ee ge fZ/eddGdddZ0e e0ge(e0fZ1de2d< eGdddeee-fZ3edddddd9d+d,Z4eddddd-d:d2d,Z4eddddd-d;d4d,Z4      d) 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])boundrT)frozenc@sDeZdZUded< ded< ded< dZded< dd d ZdS)HandoffInputDataz$str | tuple[TResponseInputItem, ...] input_historyztuple[RunItem, ...]pre_handoff_items new_itemsNzRunContextWrapper[Any] | None run_contextkwargsrreturncKst|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)selfr)r-P/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/handoffs.pyclone:szHandoffInputData.clone)r)rr*r$)__name__ __module__ __qualname____annotations__r(r/r-r-r-r.r$"s  r$r HandoffInputFilterc@seZdZUdZded< ded< ded< ded< ded < d Zd ed < d Zded< d Zded< dddZe dddZ e dddZ 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_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_enabledagentr!r*cCstd|jiS)N assistant)jsondumpsname)r,r@r-r-r.get_transfer_messageszHandoff.get_transfer_messagecCstd|jS)N transfer_to_)rtransform_string_function_stylerDclsr@r-r-r.default_tool_nameszHandoff.default_tool_namecCsd|jd|jp dS)NzHandoff to the z agent to handle the request. )rDhandoff_descriptionrHr-r-r.default_tool_descriptions z Handoff.default_tool_description)r@r!r*r6) r0r1r2__doc__r3r<r>r?rE classmethodrJrMr-r-r-r.r5Is0      r5)tool_name_overridetool_description_overrider<r?r@Agent[TContext]rP str | NonerQr<5Callable[[HandoffInputData], HandoffInputData] | Noner?Kbool | Callable[[RunContextWrapper[Any], Agent[Any]], MaybeAwaitable[bool]]r*"Handoff[TContext, Agent[TContext]]cCdSNr-)r@rPrQr<r?r-r-r.handoffsrY)rQrPr<r? on_handoff!OnHandoffWithInput[THandoffInput] input_typetype[THandoffInput]cCrWrXr-)r@rZr\rQrPr<r?r-r-r.rYs OnHandoffWithoutInputcCrWrXr-)r@rZrQrPr<r?r-r-r.rYs @OnHandoffWithInput[THandoffInput] | OnHandoffWithoutInput | Nonetype[THandoffInput] | NonePbool | Callable[[RunContextWrapper[Any], Agent[TContext]], MaybeAwaitable[bool]]c srs r r Jddur1tsJdt}t|jdkr(tdt}ndi}durIt}t|jdkrItd ddfdd } |p[t } |pbt } t |}dfdd } t| | || |j tr~| dSdS)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: contextctxRunContextWrapper[Any] input_jsonrSr*rRcsdurAdurA|durttdddidtdtj|dd}ttt}t |r:|||IdHS|||Sdur\tt }t |rX||IdHS||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 OnHandoffWithInputrinspectiscoroutinefunctionr^)rcrevalidated_input input_func no_input_func)r@r\rZrjr-r._invoke_handoffs8    z handoff.._invoke_handoff agent_baser!r=csRddlm}tsJdt||sJd||}t|r'|IdHS|S)Nr)rz is_enabled must be callable herezCan't handoff to a non-Agent)r@rcallable isinstancero isawaitable)rcrurresult)r?r-r. _is_enableds    zhandoff.._is_enabled)r7r8r9r:r<r;r?rX)rcrdrerSr*rR)rcrdrur!r*r=) rvro signaturelen parametersrr json_schemar5rJrMrrD) r@rPrQrZr\r<r?sigr9rtr7r8rzr-)r@r\r?rZrjr.rYsB      ) r@rRrPrSrQrSr<rTr?rUr*rV)r@rRrZr[r\r]rQrSrPrSr<rTr?rUr*rV)r@rRrZr^rQrSrPrSr<rTr?rUr*rV)NNNNNT)r@rRrPrSrQrSrZr_r\r`r<rTr?rar*rV)5 __future__rrorBcollections.abcr dataclassesrrr+typingrrrr r r pydanticr typing_extensionsr r exceptionsrritemsrrr(rr strict_schemar tracing.spansrutilrrr util._typesrr@rrrr rnr^r$r4r3r5rYr-r-r-r.sh       "F