.hy-$ddlmZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZmZddlmZddlmZdd lmZmZd d lmZd d lmZmZd d lmZd dlm Z mZd dl!m"Z"d dl#m#Z#d dl$m%Z%d dl&m'Z'd dl(m)Z)d dl*m+Z+edZ,e e,e fZ-e ee%e e,fe fZ.e e-e,e.e,fZ/eGddZ0eGddZ1eGddZ2eGddZ3eGddZ4e e1e2e3e4fZ5 d;d&Z6e e%e e7ge+e8fZ9e dddd'dd'd(ddddd'e6d'd(d?d:Z:dS)@) annotationsN) Awaitable) dataclass)AnyCallableLiteralUnionoverload)FiltersRankingOptions) UserLocation)ValidationError) Concatenate ParamSpec)_debug) AsyncComputerComputer)ModelBehaviorError)DocstringStylefunction_schema)RunItem)logger)RunContextWrapper) SpanError)_error_tracing)MaybeAwaitable ToolParamsc2eZdZUded< ded< ded<dS)FunctionToolResult FunctionTooltoolroutputrrun_itemN)__name__ __module__ __qualname____annotations__H/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/tool.pyr r !s9 KKK!FFr*r cReZdZUdZded< ded< ded< ded< d Zd ed <d S) r!zA tool that wraps a function. In most cases, you should use the `function_tool` helpers to create a FunctionTool, as they let you easily wrap a Python function. strname descriptionzdict[str, Any]params_json_schemaz7Callable[[RunContextWrapper[Any], str], Awaitable[Any]]on_invoke_toolTboolstrict_json_schemaN)r%r&r'__doc__r(r3r)r*r+r!r!-sqIIIX9&&&&4KKKK $####==r*r!cveZdZUdZded< dZded< dZded < dZd ed < dZd ed < e dZ dS)FileSearchToolzA hosted tool that lets the LLM search through a vector store. Currently only supported with OpenAI models, using the Responses API. z list[str]vector_store_idsNz int | Nonemax_num_resultsFr2include_search_resultszRankingOptions | Noneranking_optionszFilters | NonefilterscdS)N file_searchr)selfs r+r.zFileSearchTool.nameas}r*) r%r&r'r4r(r8r9r:r;propertyr.r)r*r+r6r6Ls 1"&O&&&&2#(((((R-1O1111%"G""""5 Xr*r6cJeZdZUdZdZded< dZded< edZdS) WebSearchToolz~A hosted tool that lets the LLM search the web. Currently only supported with OpenAI models, using the Responses API. NzUserLocation | None user_locationmediumz Literal['low', 'medium', 'high']search_context_sizecdS)Nweb_search_previewr)r>s r+r.zWebSearchTool.namers##r*) r%r&r'r4rCr(rEr@r.r)r*r+rBrBfsd*.M----ds r+r.zComputerTool.names%%r*N)r%r&r'r4r(r@r.r)r*r+rIrIwsH==&&&&&&X&&&r*rIctxRunContextWrapper[Any]error Exceptionreturnr-c&dt|S)zLThe default tool error function, which just returns a generic error message.zCAn error occurred while running the tool. Please try again. Error: )r-)rMrOs r+default_tool_error_functionrSs ]QTUZQ[Q[ ] ]]r*T name_overridedescription_overridedocstring_styleuse_docstring_infofailure_error_function strict_modefuncToolFunction[...]rU str | NonerVrWDocstringStyle | NonerXr2rYToolErrorFunction | NonerZcdS)z6Overload for usage as @function_tool (no parentheses).Nr))r[rUrVrWrXrYrZs r+ function_toolras Cr*+Callable[[ToolFunction[...]], FunctionTool]cdS)z*Overload for usage as @function_tool(...).Nr)rTs r+raras Cr*ToolFunction[...] | None:FunctionTool | Callable[[ToolFunction[...]], FunctionTool]cjdfd t|r |Sd fd }|S) aC Decorator to create a FunctionTool from a function. By default, we will: 1. Parse the function signature to create a JSON schema for the tool's parameters. 2. Use the function's docstring to populate the tool's description. 3. Use the function's docstring to populate argument descriptions. The docstring style is detected automatically, but you can override it. If the function takes a `RunContextWrapper` as the first argument, it *must* match the context type of the agent that uses the tool. Args: func: The function to wrap. name_override: If provided, use this name for the tool instead of the function's name. description_override: If provided, use this description for the tool instead of the function's docstring. docstring_style: If provided, use this style for the tool's docstring. If not provided, we will attempt to auto-detect the style. use_docstring_info: If True, use the function's docstring to populate the tool's description and argument descriptions. failure_error_function: If provided, use this function to generate an error message when the tool call fails. The error message is sent to the LLM. If you pass None, then no error message will be sent and instead an Exception will be raised. strict_mode: Whether to enable strict mode for the tool's JSON schema. We *strongly* recommend setting this to True, as it increases the likelihood of correct JSON input. If False, it allows non-strict JSON schemas. For example, if a parameter has a default value, it will be optional, additional properties are allowed, etc. See here for more: https://platform.openai.com/docs/guides/structured-outputs?api-mode=responses#supported-schemas the_funcr\rQr!ct d fd d fd }tjjpd j| S) N)r[rUrVrWrXr3rMrNinputr-rQrcK |rtj|ni}nu#t$rh}tjrt jdjnt jdjd|tdjd||d}~wwxYwtjrt jdjnt jdjd| |r j d i|n }n-#t$r }tdjd||d}~wwxYw |\}}tjst jd|d|tj r)jr |g|Ri|d{V}n+ |i|d{V}njr  |g|Ri|}n |i|}tjrt jdjdnt jdjd ||S) NzInvalid JSON input for tool z: zInvoking tool z with input zTool call args: z , kwargs: zTool z completed.z returned r))jsonloadsrPrDONT_LOG_TOOL_DATArdebugr.rparams_pydantic_modelr to_call_argsinspectiscoroutinefunction takes_context) rMri json_dataeparsedargs kwargs_dictresultschemargs r+_on_invoke_tool_implzJfunction_tool.._create_function_tool.._on_invoke_tool_impls AF,NDJu,=,=,=B    ,XL!M !M!MNNNNL!V !V!Vu!V!VWWW(I6;II%II  ( P ;fk;;<<<< NfkNNuNNOOO c!80F0==9===5577 # c c c()Z )Z)ZWX)Z)Z[[abb c!' 3 3F ; ; D+, O MMM MMNNN*844 <'B#+8C#F$#F#F#F+#F#FFFFFFFFF#+8T#A[#A#AAAAAAAFF'<%Xc@D@@@K@@FF%Xt;{;;F( F =V[===>>>> DV[DDFDDEEEMs- BA#B  B#D D* D%%D*c 0K ||d{VS#t$rt}||}tj|r |d{VcYd}~Stjt djt|d|cYd}~Sd}~wwxYw)NzError running tool (non-fatal)) tool_namerO)messagedata)rPrq isawaitablerattach_error_to_current_spanrr.r-)rMriruryr{rYrzs r+_on_invoke_toolzEfunction_tool.._create_function_tool.._on_invoke_tools 11#u=========   )1//Q77&v..(!'<<<<<<''''''; @)/%(VV # s& B*B B9B BB)r.r/r0r1r3)rMrNrir-rQr)rr!r.r/r0) rgrr{rzrVrWrYrUrZrXs ` @@r+_create_function_toolz,function_tool.._create_function_tools '!5+1*    / / / / / / / b        ,*0b%8**     r* real_funcc|SNr))rrs r+ decoratorz function_tool..decorator3s$$Y///r*)rgr\rQr!)rr\rQr!)callable) r[rUrVrWrXrYrZrrs `````` @r+rarasNW W W W W W W W W W W t~~+$$T***000000 r*)rMrNrOrPrQr-)r[r\rUr]rVr]rWr^rXr2rYr_rZr2rQr!)rUr]rVr]rWr^rXr2rYr_rZr2rQrbr)r[rdrUr]rVr]rWr^rXr2rYr_rZr2rQre); __future__rrqrkcollections.abcr dataclassesrtypingrrrr r -openai.types.responses.file_search_tool_paramr r ,openai.types.responses.web_search_tool_paramr pydanticrtyping_extensionsrrrrrJrr exceptionsrrritemsrr run_contextrtracingrutilr util._typesrrToolFunctionWithoutContextToolFunctionWithContext ToolFunctionr r!r6rBrIToolrSrPr-ToolErrorFunctionrar)r*r+rs"""""" %%%%%%!!!!!!::::::::::::::QQQQQQQQEEEEEE$$$$$$44444444--------******<<<<<<<<****** '''''' Y| $ $ %j#o6";/@/Ez/Q#RTW#WX/ ;=TU_=``a  GGGGGGG G ======= =<  2  $ $ $ $ $ $ $  $   & & & & & & &  & \>=, FG*^^^^ /4i@.QTBUUV !%'+-1#7;        !%'+-1#7;       &*H!%'+-1#7RHHHHHHHHr*