j i dZddlZddlmZmZddlmZddlmZdede fdede e e efffd Z d edee fdedee e e efffd Z d ed edefdZdS)z.Custom tool decorator for OpenAI custom tools.N) AwaitableCallable)Any)toolfunc.returncjdtdttttfffd }|S)Nxrc"d|dgSNcustom_tool_call_output)typeoutput)r rs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain_openai/tools/custom_tool.pywrappedz#_make_wrapped_func..wrapped s2dd1ggFFGG)strlistdictr)rrs` r_make_wrapped_funcr sKH3H4S#X/HHHHHH Nr coroutinec xdtdtdttttfffd }|S)Nargskwargsrc0K|i|d{V}d|dgSr r)rrresultrs rrz(_make_wrapped_coroutine..wrappeds> y$1&111111112fEEFFr)rrrr)rrs` r_make_wrapped_coroutinersSGSGCGDc3h4HGGGGGG Nrrrcdtdtfdtffd }|r(t|drs||dS|S)aDecorator to create an OpenAI custom tool. Custom tools allow for tools with (potentially long) freeform string inputs. See below for an example using LangGraph: ```python @custom_tool def execute_code(code: str) -> str: """Execute python code.""" return "27" model = ChatOpenAI(model="gpt-5", output_version="responses/v1") agent = create_react_agent(model, [execute_code]) input_message = {"role": "user", "content": "Use the tool to calculate 3^3."} for step in agent.stream( {"messages": [input_message]}, stream_mode="values", ): step["messages"][-1].pretty_print() ``` You can also specify a format for a corresponding context-free grammar using the `format` kwarg: ```python from langchain_openai import ChatOpenAI, custom_tool from langgraph.prebuilt import create_react_agent grammar = """ start: expr expr: term (SP ADD SP term)* -> add | term term: factor (SP MUL SP factor)* -> mul | factor factor: INT SP: " " ADD: "+" MUL: "*" %import common.INT """ format = {"type": "grammar", "syntax": "lark", "definition": grammar} # highlight-next-line @custom_tool(format=format) def do_math(input_string: str) -> str: """Do a mathematical operation.""" return "27" model = ChatOpenAI(model="gpt-5", output_version="responses/v1") agent = create_react_agent(model, [do_math]) input_message = {"role": "user", "content": "Use the tool to calculate 3^3."} for step in agent.stream( {"messages": [input_message]}, stream_mode="values", ): step["messages"][-1].pretty_print() ``` r.rcddi}dvrd|d<tdddi|}||_|j|_t j|rt||_nt||_ |S)Nr custom_toolformat infer_schemaFr) poprmetadata__doc__ descriptioninspectiscoroutinefunctionrrrr)rr%tool_objrs r decoratorzcustom_tool..decorator_sM* v  !'H!5!5HX 5455U5f55d;;$#|  &t , , 5!8!>!>H  .t44HMrr)rrcallable)rrr+s ` rr!r!swH c* s       "a!!"&"ya!!! r)r&r(collections.abcrrtypingrlangchain_core.toolsrrrrrrr!rrrr0s44////////%%%%%%Xc3h/HS$tCQTH~BV=V4WYs^+, c9T$sCx.1223TsTcTcTTTTTTr