tj+dZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z mZmZddlmZerddlmZmZdd lmZdd l mZdd lmZGd d e eeefeeZdS)z%Tool emulator middleware for testing.) annotations) TYPE_CHECKINGAnyGeneric) BaseChatModel) HumanMessage ToolMessage)AgentMiddleware AgentStateContextT)init_chat_model) AwaitableCallable)Command)ToolCallRequest)BaseToolc<eZdZdZddddfd ZddZddZxZS)LLMToolEmulatora9Emulates 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-5.5", 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)toolsmodelrlist[str | BaseTool] | Nonerstr | BaseChatModel | NonereturnNonect|du|_t|_|jsV|T|D]Q}t |t r|j|2|j|jR|tdd|_ dSt |tr ||_ dSt|d|_ dS)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/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain/agents/middleware/tool_emulator.pyrzLLMToolEmulator.__init__Bs( !D=*-%% 9E$5 9 9dC((9)--d3333)--di8888 =()O]^___DJJJ } - - ?DJJJ(A>>>DJJJrequestrhandler7Callable[[ToolCallRequest], ToolMessage | Command[Any]]ToolMessage | Command[Any]cF|jd}|jp||jv}|s ||S|jd}|jr |jjnd}d|d|d|d}|jt|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. r&argsNo 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_callr r"r( descriptionrinvokerr r9 r'r,r- tool_nameshould_emulate tool_argstool_descriptionpromptresponses r*wrap_tool_callzLLMToolEmulator.wrap_tool_callms%f- )OY$:O-O $77## #%f- 7>|c7<33Ic 8 8 8, 8 8$ 8 8 8 :$$l6&:&:%;<<$ *40    r+BCallable[[ToolCallRequest], Awaitable[ToolMessage | Command[Any]]]cbK|jd}|jp||jv}|s||d{VS|jd}|jr |jjnd}d|d|d|d}|jt|gd{V}t|j |jd | S) 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. r&Nr1r2r3r4r5r6r7r8) r;r r"r(r<rainvokerr r9r>s r*awrap_tool_callzLLMToolEmulator.awrap_tool_calls"%f- )OY$:O-O * )))))))) )%f- 7>|c7<33Ic 8 8 8, 8 8$ 8 8 8 ++\&-A-A,BCCCCCCCC$ *40    r+)rrrrrr)r,rr-r.rr/)r,rr-rFrr/)__name__ __module__ __qualname____doc__rrErI __classcell__)r)s@r*rrs))\.2,0 )?)?)?)?)?)?)?)?V0 0 0 0 d2 2 2 2 2 2 2 2 r+rN)rM __future__rtypingrrr*langchain_core.language_models.chat_modelsrlangchain_core.messagesrr !langchain.agents.middleware.typesr r r langchain.chat_models.baser collections.abcrrlanggraph.typesrrlangchain.toolsrrr+r*rYs>++""""""..........DDDDDD========SSSSSSSSSS666666)33333333''''''AAAAAA(((((({ { { { { ojox&?@'(BS{ { { { { r+