o h@sddlmZddlZddlZddlZddlmZddlmZmZddl m Z m Z m Z ddl mZmZmZmZmZddlmZddlmZmZmZmZmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z+d dl,m-Z-m.Z.d dl/m0Z0d dl1m2Z2m3Z3d dl4m5Z5m6Z6m7Z7m8Z8m9Z9d dl:m;Z;mZ>d dl?m@Z@mAZAd dlBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOd dlPmQZQd dlRmRZRd dlSmTZTd dlUmVZVd dlWmXZXmYZYd dlZm[Z[m\Z\d dl]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfd dlgmhZhd dlimjZjmkZkmlZlmmZmd dlnmoZompZpmqZqmrZrmsZsmtZtmuZud dlvmwZwmxZxe r?d d lymzZzGd!d"d"Z{e{Z|e.d#dd$Z}eGd%d&d&Z~eGd'd(d(ZeGd)d*d*ZeGd+d,d,ZeGd-d.d.ZeGd/d0d0ZeGd1d2d2ZeGd3d4d4ZeGd5d6d6ZeGd7d8d8ZeGd9d:d:ZdOd@dAZGdBdCdCZGdDdEdEZGdFdGdGZGdHdIdIZdPdMdNZdS)Q) annotationsN) Awaitable) dataclassfield) TYPE_CHECKINGAnycast)ResponseComputerToolCallResponseFileSearchToolCallResponseFunctionToolCallResponseFunctionWebSearchResponseOutputMessage)ResponseCodeInterpreterToolCall) ActionClickActionDoubleClick ActionDragActionKeypress ActionMoveActionScreenshot ActionScroll ActionType ActionWait))ComputerCallOutputAcknowledgedSafetyCheck)ComputerCallOutputMcpApprovalResponse)ImageGenerationCallLocalShellCallMcpApprovalRequestMcpCall McpListTools)ResponseReasoningItem)AgentToolsToFinalOutputResult)AgentOutputSchemaBase) AsyncComputerComputer)AgentsExceptionModelBehaviorError#ToolInputGuardrailTripwireTriggered$ToolOutputGuardrailTripwireTriggered UserError)InputGuardrailInputGuardrailResultOutputGuardrailOutputGuardrailResult)HandoffHandoffInputData) HandoffCallItemHandoffOutputItem ItemHelpersMCPApprovalRequestItemMCPApprovalResponseItemMCPListToolsItemMessageOutputItem ModelResponse ReasoningItemRunItem ToolCallItemToolCallOutputItemTResponseInputItem)RunHooks)logger) ModelSettings) ModelTracing)RunContextWrapperTContext)RunItemStreamEvent StreamEvent) ComputerToolComputerToolSafetyCheckData FunctionToolFunctionToolResult HostedMCPToolLocalShellCommandRequestLocalShellToolMCPToolApprovalRequestTool) ToolContext)ToolInputGuardrailDataToolInputGuardrailResultToolOutputGuardrailDataToolOutputGuardrailResult) SpanErrorTrace function_spanget_current_traceguardrail_span handoff_spantrace)_coro_error_tracing) RunConfigc@ eZdZdS)QueueCompleteSentinelN__name__ __module__ __qualname__rereQ/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/_run_impl.pyr`nsr`Fis_final_output final_outputc@s6eZdZUeedZded< dd d Zdd dZdS)AgentToolUseTracker)default_factoryzlist[tuple[Agent, list[str]]]agent_to_toolsagent Agent[Any] tool_names list[str]returnNonecsDtfdd|jDd}|r|d|dS|j|fdS)Nc3 |] }|dkr|VqdSrNre.0itemrmrerf }z3AgentToolUseTracker.add_tool_use..r!)nextrlextendappend)selfrmro existing_datarerxrf add_tool_use|sz AgentToolUseTracker.add_tool_useboolcs2tfdd|jDd}|duot|ddkS)Nc3rsrtrerurxrerfryrzz5AgentToolUseTracker.has_used_tools..r!r)r{rllen)r~rmrrerxrfhas_used_toolssz"AgentToolUseTracker.has_used_toolsN)rmrnrorprqrr)rmrnrqr) rbrcrdrlistrl__annotations__rrrerererfrjws  rjc@eZdZUded<ded<dS)ToolRunHandoffr0handoffr tool_callNrbrcrdrrerererfr  rc@r)ToolRunFunctionr rrI function_toolNrrerererfrrrc@r)ToolRunComputerActionr rrG computer_toolNrrerererfrrrc@r)ToolRunMCPApprovalRequestr request_itemrKmcp_toolNrrerererfrrrc@r)ToolRunLocalShellCallrrrMlocal_shell_toolNrrerererfrrrc@sPeZdZUded<ded<ded<ded<d ed <d ed <d ed<dddZdS)ProcessedResponse list[RunItem] new_itemslist[ToolRunHandoff]handoffslist[ToolRunFunction] functionslist[ToolRunComputerAction]computer_actionslist[ToolRunLocalShellCall]local_shell_callsrp tools_usedlist[ToolRunMCPApprovalRequest]mcp_approval_requestsrqrcCst|j|j|j|j|jgSN)anyrrrrrr~rererfhas_tools_or_approvals_to_runsz/ProcessedResponse.has_tools_or_approvals_to_runN)rqr)rbrcrdrrrerererfrs rc@eZdZUded<dS)NextStepHandoffrn new_agentNrrerererfr rc@r)NextStepFinalOutputroutputNrrerererfrrrc@r_)NextStepRunAgainNrarerererfrsrc@sbeZdZUded< ded< ded< ded< ded < d ed < d ed < edddZdS)SingleStepResultstr | list[TResponseInputItem]original_inputr9model_responserpre_step_itemsnew_step_itemsz8NextStepHandoff | NextStepFinalOutput | NextStepRunAgain next_steplist[ToolInputGuardrailResult]tool_input_guardrail_resultslist[ToolOutputGuardrailResult]tool_output_guardrail_resultsrqcCs |j|jS)z`Items generated during the agent run (i.e. everything generated after `original_input`).)rrrrererfgenerated_itemss z SingleStepResult.generated_itemsN)rqr)rbrcrdrpropertyrrerererfrs" rtracing_disabledrtrace_include_sensitive_datarqrBcCs|rtjS|r tjStjSr)rBDISABLEDENABLEDENABLED_WITHOUT_DATA)rrrererfget_model_tracing_impls rc@seZdZedmddZednddZedod#d$Zedpd,d-Zedqd2d3Zedrd6d7Z edsd"sz:RunImpl.execute_tools_and_side_effects..)rmapproval_requestsr) rmrrrr run_handoffsrrr)rm tool_resultsrrzhModel returned a final output of None. Not raising an error because we assumeyou know what you're doing.) rmrrrrrirrrrcSsg|] }t|tr|qSre) isinstancer8rurererfr]srrrrrrr)rr|rasynciogatherexecute_function_tool_callsrexecute_computer_actionsrrexecute_mcp_approval_requestsrexecute_handoffs"_check_for_final_output_from_toolsrh output_typestrrir@errorexecute_final_outputr4extract_last_textraw_itemr is_plain_text validate_jsonrr)clsrmrrrrrrrrrfunction_resultsrrcomputer_resultsapproval_resultsrcheck_tool_use message_itemspotential_final_output_textrirererfexecute_tools_and_side_effectss             z&RunImpl.execute_tools_and_side_effectsrntool_use_trackerrjmodel_settingsrAcCs&|jdur||rtj|ddS|S)znResets tool choice to None if the agent has used tools and the agent's reset_tool_choice flag is True.TN) tool_choice)reset_tool_choicer dataclassesreplace)rrmrrrererfmaybe_reset_tool_choiceszRunImpl.maybe_reset_tool_choice all_tools list[Tool]responser list[Handoff]c Cs"g}g}g}g} g} g} g} dd|D} dd|D}tdd|Dd}tdd|Dd}dd|D}|jD]}t|trM|t||dn8t|trb|t||d| d n#t|trw|t||d| d nt|t r|t ||dnt|t r|t||d| d |st td id td| t||dnt|tr|t||d|j|vrt tdd|jid td|jd||j}|jr| t||dntd|jdnt|tr|t||dn~t|tr|t||d| dnit|tr1|t||d| dnTt|trF|t||d| dn?t|trt|t||d| d|sjt tdid td| t||dnt|tstdt |q:t|tsq:| |j!|j!| vr|t"||dt#|| |j!d}||q:|j!|vr|dur|j!d kr|t||d|t$|t%|d!q:t td"d#|j!id d$|j!d%|j!}t||t||d|t$|||j!d!q:t&|||| | | | d&S)'NcSsi|]}|j|qSre) tool_namervrrererf z2RunImpl.process_model_response..cSsi|] }t|tr|j|qSre)rrInamervtoolrererfrscs|] }t|tr|VqdSr)rrGrrererfryz1RunImpl.process_model_response..csrr)rrMrrererfryrcSs"i|] }t|tr|jd|qS) server_label)rrK tool_configrrererfrs  rrm file_search web_search computer_usezComputer tool not foundmessagedataz7Model produced computer action without a computer tool.)rrzMCP server label not foundrzMCP server label z not found)rrz MCP server z has no on_approval_request hookmcpimage_generationcode_interpreter local_shellzLocal shell tool not foundz;Model produced local shell call without a local shell tool.)rrz"Unexpected output type, ignoring: )rrjson_tool_call)rrzTool not foundrzTool z not found in agent )rrrrrrr)'r{rrr r}r8r r<r r r:r r]attach_error_to_current_spanrUr(rrr5ron_approval_requestrr@warningrr7rrrrrr typerr2rr_build_litellm_json_tool_callr)rrmrrrritemsrrrrrr handoff_map function_maprrhosted_mcp_server_maprserverrrrererfprocess_model_responses                              zRunImpl.process_model_response func_toolrI tool_contextToolContext[TContext]rr str | Nonecs|jsdS|jD]:}|t||dIdH}|t||d|jddkr-t||d|jddkr;|jdS|jddkrCq q dS) aExecute input guardrails for a tool. Args: func_tool: The function tool being executed. tool_context: The tool execution context. agent: The agent executing the tool. tool_input_guardrail_results: List to append guardrail results to. Returns: None if tool execution should proceed, or a message string if execution should be skipped. Raises: ToolInputGuardrailTripwireTriggered: If a guardrail triggers an exception. N)contextrm guardrailrrraise_exceptionreject_contentr allow)tool_input_guardrailsrunrQr}rRbehaviorr))rrrrmrr#gr_outrererf_execute_input_guardrailsCs0   z!RunImpl._execute_input_guardrails real_resultrrrc s|js|S|}|jD]<}|t|||dIdH}|t||d|jddkr0t||d|jddkr?|jd}|S|jddkrGq q |S) a7Execute output guardrails for a tool. Args: func_tool: The function tool being executed. tool_context: The tool execution context. agent: The agent executing the tool. real_result: The actual result from the tool execution. tool_output_guardrail_results: List to append guardrail results to. Returns: The final result after guardrail processing (may be modified). Raises: ToolOutputGuardrailTripwireTriggered: If a guardrail triggers an exception. )r!rmrNr"rr$r%r r&)tool_output_guardrailsr(rSr}rTr)r*) rrrrmr,r final_resultoutput_guardrailr*rererf_execute_output_guardrailszs<    z"RunImpl._execute_output_guardrailsrr csLt|||||jr|j|||ntIdH|||jIdHS)aExecute the core tool function with before/after hooks. Args: func_tool: The function tool being executed. tool_context: The tool execution context. agent: The agent executing the tool. hooks: The run hooks to execute. tool_call: The tool call details. Returns: The result from the tool execution. N)rr on_tool_startrr\noop_coroutineon_invoke_tool arguments)rrrrmrrrererf_execute_tool_with_hookss   z RunImpl._execute_tool_with_hooksrrr`tuple[list[FunctionToolResult], list[ToolInputGuardrailResult], list[ToolOutputGuardrailResult]]c s|ggd fdd }g}|D]}|j} ||| |jqtj|IdH} fd d t|| D} | fS) NrrIrr rqrc slt|j}tj|j|d}jr|j|j_zNj ||dIdH}|dur,|}n7j |||dIdH}j |||dIdH}t |||jr[j|||ntIdH|}Wn.ty}z"ttd|jt|ddt|tr|td|jd ||d}~wwjr||j_Wd|SWd|S1swY|S) N)r)rrrmr)rrrmrr)rrrmr,rzError running tool)rrr zError running tool z: )rWrrPfrom_agent_contextcall_idrr4 span_datainputr+r5r0rr on_tool_endrr\r2 Exceptionr]rrUrrr'r+r) rrspan_fnrrejected_messager.r,rermrrrrrrrerfrun_single_tools|            @ @@z.run_single_toolc s8g|]\}}t|j|t|t|jt|ddqS)rrrm)rrr)rJrr=r4tool_call_output_itemrr)rvtool_runrrxrerfr4s z7RunImpl.execute_function_tool_calls..)rrIrr rqr)rr}rrrzip) rrmrrrrrAtasksrDrresultsfunction_tool_resultsrer@rfrs E   z#RunImpl.execute_function_tool_callscallsrc s4g}|D]}|tj|||||dIdHq|S)N)rmcallrrr)r}LocalShellActionexecute)rrmrIrrrrGrJrererfexecute_local_shell_callsCs   z!RunImpl.execute_local_shell_callsrrc sg}|D]T}d}|jjrH|jjrHg}|jjD]0} t|||j| d} |j| } t| r2| IdHn| } | rD|t| j | j | j dqt d|t j||||||dIdHq|S)N) ctx_wrapperrmr safety_check)idcoder z/Computer tool safety check was not acknowledged)rmactionrrracknowledged_safety_checks)rpending_safety_checksron_safety_checkrHinspect isawaitabler}rrPrQr r+ComputerActionrL) rrmrrrrrGrR acknowledgedcheckr maybeackrererfr[sF     z RunImpl.execute_computer_actionsrrrc  s,t|dk} | rd|fdd|ddD|d} tjd} | j} | || jjIdH}|j| j_ | rNdd|D}| t dd |id | t t| j| ||d t|j||d jrvjj||d ntIdH| jp| r| jnd}|rtdtt|trt|n|t|t||d}t|st !| t dddid t"d|||}t#$|r|IdH}t|tst !| t dddid t"d|t|j%t&r|j%nt|j%}t|j'}t|j(}Wdn 1swYt)||||t*|ggdS)Nr!z.Multiple handoffs detected, ignoring this one.cs$g|]}tt|jdqS)rB)r=r4rCrrrmoutput_messagererfrsz,RunImpl.execute_handoffs..r) from_agentcSsg|]}|jjqSre)r agent_namerrererfrrzMultiple handoffs requestedrequested_agentsr )rmr source_agent target_agent)r!r_to_agent)rmsourcezFiltering inputs for handoff) input_historypre_handoff_itemsr run_contextzInvalid input filterdetailsznot callable()zInvalid input filter: zInvalid input filter resultznot a HandoffInputDatazInvalid input filter result: r)+rr|rZrron_invoke_handoffrr4r9rd set_errorrUr}r3r4rCget_transfer_messagerr on_handoffrr\r2 input_filterhandoff_input_filterr@debugr1rrtuplecallabler]attach_error_to_spanr+rVrWrfrrgrrr)rrmrrrrrrrrmultiple_handoffsactual_handoff span_handoffrrrarnhandoff_input_datafilteredrer]rfrs                  \zRunImpl.execute_handoffsrrcs4d fdd fdd|D}tj|IdHS) Napproval_requestrrqr;cs|jj}|dus Jd|t|j}t|r |IdH}n|}|dd}|jj|ddd}|ds<|r<||d<t|dS)Nz.Callback is required for MCP approval requestsreasonapprovemcp_approval_response)approval_request_idr{rr) rrrNrrVrWgetrPr6)rycallbackmaybe_awaitable_resultrrzr)rmrrerfrun_single_approvals(     zBRunImpl.execute_mcp_approval_requests..run_single_approvalcsg|]}|qSrere)rvry)rrerfr1rz9RunImpl.execute_mcp_approval_requests..)ryrrqr;)rr)rrmrrrFre)rmrrrfrsz%RunImpl.execute_mcp_approval_requestsric s2|||||IdHt||||t|| | dS)Nr)run_final_output_hooksrr) rrmrrrrrirrrrrererfr4szRunImpl.execute_final_outputcs<t|||||jr|j|||ntIdHdSr)rr on_agent_endron_endr\r2)rrmrrrirererfrPs zRunImpl.run_final_output_hooksr#InputGuardrail[TContext]r:r!r-csVt|}||||IdH}|jj|j_|WdS1s$wYdSrrYget_namer(rtripwire_triggeredr9 triggered)rrmr#r:r!span_guardrailrrererfrun_single_input_guardrail_s  $z"RunImpl.run_single_input_guardrailOutputGuardrail[TContext] agent_outputr/csXt|}|j|||dIdH}|jj|j_|WdS1s%wYdS)N)rmrr!r)rr#rmrr!rrrererfrun_single_output_guardraills  $z#RunImpl.run_single_output_guardrailqueue2asyncio.Queue[StreamEvent | QueueCompleteSentinel]cCs|D]u}t|trt|dd}n`t|trt|dd}nTt|tr(t|dd}nHt|tr4t|dd}n   zComputerAction.executerr&rr rcs4|j}t|tr||j|j|j|St|tr(||j|j|St|t r<| dd|j D|St|t rK| |j|St|tr\||j|j|St|tri||St|tr~||j|j|j|j|St|tr||j|St|tr||S)NcSg|]}|j|jfqSrexyrvprererfr3z7ComputerAction._get_screenshot_sync..rRrrclickrrbuttonr double_clickrdragpathrkeypresskeysrmover screenshotrscrollscroll_xscroll_yrrtextrwaitrrrrRrererfr's:         z#ComputerAction._get_screenshot_syncr%cs@|j}t|tr||j|j|jIdHnt|tr(||j|jIdHnqt|t r<| dd|j DIdHn]t|t rK| |jIdHnNt|tr\||j|jIdHn=t|tri|IdHn0t|tr~||j|j|j|jIdHnt|tr||jIdHn t|tr|IdH|IdHS)NcSrrerrrererfrOrz8ComputerAction._get_screenshot_async..rrrererfrCs*          z$ComputerAction._get_screenshot_async)rmrrRrrrrrrr^rSrrqr;)rr&rr rqr)rr%rr rqr)rbrcrdrrLrrrerererfrXs 3 rXc@seZdZedd dZdS)rKrmrrJrrrrrrr^rqr;c st||||j|jr|j|||jntIdHt||jd}|j |}t |r6|IdH}n|}t| |||j||jrO|j |||j|ntIdHt ||d|jj|ddS)N)rNr local_shell_call_output)rrPrr)rrr1rrr\r2rLrexecutorrVrWr;r=r8) rrmrJrrrrequestrrrererfrLas:        zLocalShellAction.executeN) rmrrJrrrrrrr^rqr;)rbrcrdrrLrerererfrK`srKrr rIcCs"d dd}t|j|ji|ddd S) N_ctxToolContext[Any]valuerrqcs"t|trddl}||S|S)Nr)rrjsonloads)rrrrererfr3s   z5_build_litellm_json_tool_call..on_invoke_toolT)r descriptionparams_json_schemar3strict_json_schema is_enabled)rrrrrqr)rIr)rr3rererfrs r)rrrrrqrB)rr rqrI) __future__rrrrVcollections.abcrrrtypingrrropenai.types.responsesr r r r r :openai.types.responses.response_code_interpreter_tool_callr2openai.types.responses.response_computer_tool_callrrrrrrrrr0openai.types.responses.response_input_item_paramr+openai.types.responses.response_input_paramrr+openai.types.responses.response_output_itemrrrrr.openai.types.responses.response_reasoning_itemr rmr"r#rr$rr%r& exceptionsr'r(r)r*r+r#r,r-r.r/rr0r1rr2r3r4r5r6r7r8r9r:r;r<r=r> lifecycler?r@rrAmodels.interfacerBrhrCrD stream_eventsrErFrrGrHrIrJrKrLrMrNrOrrPtool_guardrailsrQrRrSrTtracingrUrVrWrXrYrZr[utilr\r]r(r^r`QUEUE_COMPLETE_SENTINELrrjrrrrrrrrrrrrrrXrKrrerererfs   ,   <    , $     ]%n3