g3fidZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z erddlmZmZdd lmZdd l mZdd lmZGd d e Zy)z%Tool emulator middleware for testing.) annotations) TYPE_CHECKING) BaseChatModel) HumanMessage ToolMessage)AgentMiddleware)init_chat_model) AwaitableCallable)Command)ToolCallRequest)BaseToolc^eZdZdZddd dfdZ ddZ d dZxZS) LLMToolEmulatora8Emulates specified tools using an LLM instead of executing them. This middleware allows selective emulation of tools for testing purposes. By default (when `tools=None`), all tools are emulated. You can specify which tools to emulate by passing a list of tool names or `BaseTool` instances. Examples: !!! example "Emulate all tools (default behavior)" ```python from langchain.agents.middleware import LLMToolEmulator middleware = LLMToolEmulator() agent = create_agent( model="openai:gpt-4o", tools=[get_weather, get_user_location, calculator], middleware=[middleware], ) ``` !!! example "Emulate specific tools by name" ```python middleware = LLMToolEmulator(tools=["get_weather", "get_user_location"]) ``` !!! example "Use a custom model for emulation" ```python middleware = LLMToolEmulator( tools=["get_weather"], model="anthropic:claude-sonnet-4-5-20250929" ) ``` !!! example "Emulate specific tools by passing tool instances" ```python middleware = LLMToolEmulator(tools=[get_weather, get_user_location]) ``` N)toolsmodelct||du|_t|_|jsZ|X|D]S}t |t r|jj|/|jj|jU|tdd|_ yt |tr||_ yt|d|_ y)aInitialize the tool emulator. Args: tools: List of tool names (`str`) or `BaseTool` instances to emulate. If `None`, ALL tools will be emulated. If empty list, no tools will be emulated. model: Model to use for emulation. Defaults to `'anthropic:claude-sonnet-4-5-20250929'`. Can be a model identifier string or `BaseChatModel` instance. Nz$anthropic:claude-sonnet-4-5-20250929) temperature) super__init__ emulate_allsettools_to_emulate isinstancestraddnamer rr)selfrrtool __class__s g/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain/agents/middleware/tool_emulator.pyrzLLMToolEmulator.__init__Bs( !D=*-%E$5 9dC())--d3))--dii8  9 =()O]^_DJ } -DJ(A>DJc~|jd}|jxs||jv}|s||S|jd}|jr|jjnd}d|d|d|d}|j j t|g}t|j|jd| S) aEmulate tool execution using LLM if tool should be emulated. Args: request: Tool call request to potentially emulate. handler: Callback to execute the tool (can be called multiple times). Returns: ToolMessage with emulated response if tool should be emulated, otherwise calls handler for normal execution. rargsNo description available;You are emulating a tool call for testing purposes. Tool:  Description: Arguments: Generate a realistic response that this tool would return given these arguments. Return ONLY the tool's output, no explanation or preamble. Introduce variation into your responses.idcontent tool_call_idr) tool_callrrr descriptionrinvokerrr- rrequesthandler tool_nameshould_emulate tool_argstool_descriptionpromptresponses r"wrap_tool_callzLLMToolEmulator.wrap_tool_callms%%f- ))OY$:O:O-O7# #%%f- 7>||7<<33IcK ,-.#%7 8 ::$$l6&:%;<$$ **40  r#cK|jd}|jxs||jv}|s||d{S|jd}|jr|jjnd}d|d|d|d}|j j t|gd{}t|j|jd | S77+w) aAsync version of `wrap_tool_call`. Emulate tool execution using LLM if tool should be emulated. Args: request: Tool call request to potentially emulate. handler: Async callback to execute the tool (can be called multiple times). Returns: ToolMessage with emulated response if tool should be emulated, otherwise calls handler for normal execution. rNr%r&r'r(r)r*r+r,) r/rrr r0rainvokerrr-r2s r"awrap_tool_callzLLMToolEmulator.awrap_tool_calls"%%f- ))OY$:O:O-O )) )%%f- 7>||7<<33IcK ,-.#%7 8 ++\&-A,BCC$$ **40  -*&Ds"9CCA+C'C(*CC)rzlist[str | BaseTool] | Nonerzstr | BaseChatModel | NonereturnNone)r3r r4z2Callable[[ToolCallRequest], ToolMessage | Command]r?ToolMessage | Command)r3r r4z=Callable[[ToolCallRequest], Awaitable[ToolMessage | Command]]r?rA)__name__ __module__ __qualname____doc__rr;r> __classcell__)r!s@r"rrsz)\.2,0 )?+)?* )?  )?V0  0 D0   0 d2  2 O2   2 r#rN)rE __future__rtypingr*langchain_core.language_models.chat_modelsrlangchain_core.messagesrr!langchain.agents.middleware.typesrlangchain.chat_models.baser collections.abcr r langgraph.typesr r langchain.toolsrrr#r"rQs7+" D==63'A({ o{ r#