Home / File/ test_tool_emulator.py — langchain Source File

test_tool_emulator.py — langchain Source File

Architecture documentation for test_tool_emulator.py, a python file in the langchain codebase. 14 imports, 0 dependents.

File python LangChainCore Runnables 14 imports 3 functions 6 classes

Entity Profile

Dependency Diagram

graph LR
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1["test_tool_emulator.py"]
  2bf6d401_816d_d011_3b05_a6114f55ff58["collections.abc"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 2bf6d401_816d_d011_3b05_a6114f55ff58
  3741f36e_1d35_fcc3_9884_1c5e40d99797["itertools"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 3741f36e_1d35_fcc3_9884_1c5e40d99797
  feec1ec4_6917_867b_d228_b134d0ff8099["typing"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> feec1ec4_6917_867b_d228_b134d0ff8099
  e929cf21_6ab8_6ff3_3765_0d35a099a053["langchain_core.language_models"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> e929cf21_6ab8_6ff3_3765_0d35a099a053
  f77bacd6_1087_c145_d5d5_68e6f3937f9d["langchain_core.language_models.chat_models"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> f77bacd6_1087_c145_d5d5_68e6f3937f9d
  00fc047e_6208_0289_ebf0_f5b91b72ee90["langchain_core.language_models.fake_chat_models"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 00fc047e_6208_0289_ebf0_f5b91b72ee90
  9444498b_8066_55c7_b3a2_1d90c4162a32["langchain_core.messages"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 9444498b_8066_55c7_b3a2_1d90c4162a32
  4382dc25_6fba_324a_49e2_e9742d579385["langchain_core.outputs"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 4382dc25_6fba_324a_49e2_e9742d579385
  31eab4ab_7281_1e6c_b17d_12e6ad9de07a["langchain_core.runnables"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 31eab4ab_7281_1e6c_b17d_12e6ad9de07a
  121262a1_0bd6_d637_bce3_307ab6b3ecd4["langchain_core.tools"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 121262a1_0bd6_d637_bce3_307ab6b3ecd4
  dd5e7909_a646_84f1_497b_cae69735550e["pydantic"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> dd5e7909_a646_84f1_497b_cae69735550e
  d9a6942a_c37a_07f8_ed13_74d0fdc117be["langchain.agents"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> d9a6942a_c37a_07f8_ed13_74d0fdc117be
  598be07d_58e8_b830_8ebe_d631441aa2fe["langchain.agents.middleware"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 598be07d_58e8_b830_8ebe_d631441aa2fe
  397e32fd_cfd9_f9c7_de38_41991c05a78b["langchain.messages"]
  5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 --> 397e32fd_cfd9_f9c7_de38_41991c05a78b
  style 5b00cf78_0f18_c9ad_fc5f_85cc0378daf1 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

"""Unit tests for tool emulator middleware."""

from collections.abc import Callable, Sequence
from itertools import cycle
from typing import Any, Literal

from langchain_core.language_models import LanguageModelInput
from langchain_core.language_models.chat_models import BaseChatModel
from langchain_core.language_models.fake_chat_models import GenericFakeChatModel
from langchain_core.messages import BaseMessage, HumanMessage
from langchain_core.outputs import ChatGeneration, ChatResult
from langchain_core.runnables import Runnable
from langchain_core.tools import BaseTool, tool
from pydantic import BaseModel

from langchain.agents import create_agent
from langchain.agents.middleware import LLMToolEmulator
from langchain.messages import AIMessage


@tool
def get_weather(location: str) -> str:
    """Get current weather for a location."""
    msg = "This tool should be emulated"
    raise NotImplementedError(msg)


@tool
def search_web(query: str) -> str:
    """Search the web for information."""
    msg = "This tool should be emulated"
    raise NotImplementedError(msg)


@tool
def calculator(expression: str) -> str:
    """Perform mathematical calculations."""
    # This tool executes normally (not emulated)
    return f"Result: {eval(expression)}"  # noqa: S307


class FakeModel(GenericFakeChatModel):
    """Fake model that supports bind_tools."""

    tool_style: Literal["openai", "anthropic"] = "openai"

    def bind_tools(
        self,
        tools: Sequence[dict[str, Any] | type[BaseModel] | Callable[..., Any] | BaseTool],
        **_kwargs: Any,
    ) -> Runnable[LanguageModelInput, AIMessage]:
        if len(tools) == 0:
            msg = "Must provide at least one tool"
            raise ValueError(msg)

        tool_dicts = []
        for tool_ in tools:
            if isinstance(tool_, dict):
                tool_dicts.append(tool_)
                continue
// ... (564 more lines)

Domain

Subdomains

Dependencies

  • collections.abc
  • itertools
  • langchain.agents
  • langchain.agents.middleware
  • langchain.messages
  • langchain_core.language_models
  • langchain_core.language_models.chat_models
  • langchain_core.language_models.fake_chat_models
  • langchain_core.messages
  • langchain_core.outputs
  • langchain_core.runnables
  • langchain_core.tools
  • pydantic
  • typing

Frequently Asked Questions

What does test_tool_emulator.py do?
test_tool_emulator.py is a source file in the langchain codebase, written in python. It belongs to the LangChainCore domain, Runnables subdomain.
What functions are defined in test_tool_emulator.py?
test_tool_emulator.py defines 3 function(s): calculator, get_weather, search_web.
What does test_tool_emulator.py depend on?
test_tool_emulator.py imports 14 module(s): collections.abc, itertools, langchain.agents, langchain.agents.middleware, langchain.messages, langchain_core.language_models, langchain_core.language_models.chat_models, langchain_core.language_models.fake_chat_models, and 6 more.
Where is test_tool_emulator.py in the architecture?
test_tool_emulator.py is located at libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_tool_emulator.py (domain: LangChainCore, subdomain: Runnables, directory: libs/langchain_v1/tests/unit_tests/agents/middleware/implementations).

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free