langchain Architecture Docs
Home / File/ test_tool_selection.py — langchain Source File

test_tool_selection.py — langchain Source File

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

File python LangChainCore ApiManagement 14 imports 5 functions 6 classes

Entity Profile

Dependency Diagram

graph LR
  f96107c8_8757_b550_ad6f_b1dcda0da20b["test_tool_selection.py"]
  2bf6d401_816d_d011_3b05_a6114f55ff58["collections.abc"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 2bf6d401_816d_d011_3b05_a6114f55ff58
  3741f36e_1d35_fcc3_9884_1c5e40d99797["itertools"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 3741f36e_1d35_fcc3_9884_1c5e40d99797
  feec1ec4_6917_867b_d228_b134d0ff8099["typing"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> feec1ec4_6917_867b_d228_b134d0ff8099
  f69d6389_263d_68a4_7fbf_f14c0602a9ba["pytest"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> f69d6389_263d_68a4_7fbf_f14c0602a9ba
  e929cf21_6ab8_6ff3_3765_0d35a099a053["langchain_core.language_models"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> e929cf21_6ab8_6ff3_3765_0d35a099a053
  00fc047e_6208_0289_ebf0_f5b91b72ee90["langchain_core.language_models.fake_chat_models"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 00fc047e_6208_0289_ebf0_f5b91b72ee90
  9444498b_8066_55c7_b3a2_1d90c4162a32["langchain_core.messages"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 9444498b_8066_55c7_b3a2_1d90c4162a32
  31eab4ab_7281_1e6c_b17d_12e6ad9de07a["langchain_core.runnables"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 31eab4ab_7281_1e6c_b17d_12e6ad9de07a
  121262a1_0bd6_d637_bce3_307ab6b3ecd4["langchain_core.tools"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 121262a1_0bd6_d637_bce3_307ab6b3ecd4
  dd5e7909_a646_84f1_497b_cae69735550e["pydantic"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> dd5e7909_a646_84f1_497b_cae69735550e
  d9a6942a_c37a_07f8_ed13_74d0fdc117be["langchain.agents"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> d9a6942a_c37a_07f8_ed13_74d0fdc117be
  598be07d_58e8_b830_8ebe_d631441aa2fe["langchain.agents.middleware"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 598be07d_58e8_b830_8ebe_d631441aa2fe
  55a5e8fe_1e74_0826_8313_3d7262cb8bd5["langchain.agents.middleware.tool_selection"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 55a5e8fe_1e74_0826_8313_3d7262cb8bd5
  397e32fd_cfd9_f9c7_de38_41991c05a78b["langchain.messages"]
  f96107c8_8757_b550_ad6f_b1dcda0da20b --> 397e32fd_cfd9_f9c7_de38_41991c05a78b
  style f96107c8_8757_b550_ad6f_b1dcda0da20b fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

"""Unit tests for LLM tool selection middleware."""

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

import pytest
from langchain_core.language_models import LanguageModelInput
from langchain_core.language_models.fake_chat_models import GenericFakeChatModel
from langchain_core.messages import HumanMessage
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 (
    LLMToolSelectorMiddleware,
    ModelRequest,
    ModelResponse,
    wrap_model_call,
)
from langchain.agents.middleware.tool_selection import _create_tool_selection_response
from langchain.messages import AIMessage


@tool
def get_weather(location: str) -> str:
    """Get current weather for a location."""
    return f"Weather in {location}: 72°F, sunny"


@tool
def search_web(query: str) -> str:
    """Search the web for information."""
    return f"Search results for: {query}"


@tool
def calculate(expression: str) -> str:
    """Perform mathematical calculations."""
    return f"Result of {expression}: 42"


@tool
def send_email(to: str, subject: str) -> str:
    """Send an email to someone."""
    return f"Email sent to {to}"


@tool
def get_stock_price(symbol: str) -> str:
    """Get current stock price for a symbol."""
    return f"Stock price for {symbol}: $150.25"


class FakeModel(GenericFakeChatModel):
    tool_style: Literal["openai", "anthropic"] = "openai"

    def bind_tools(
        self,
// ... (582 more lines)

Domain

Subdomains

Functions

Classes

Dependencies

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

Source

Frequently Asked Questions

What does test_tool_selection.py do?
test_tool_selection.py is a source file in the langchain codebase, written in python. It belongs to the LangChainCore domain, ApiManagement subdomain.
What functions are defined in test_tool_selection.py?
test_tool_selection.py defines 5 function(s): calculate, get_stock_price, get_weather, search_web, send_email.
What does test_tool_selection.py depend on?
test_tool_selection.py imports 14 module(s): collections.abc, itertools, langchain.agents, langchain.agents.middleware, langchain.agents.middleware.tool_selection, langchain.messages, langchain_core.language_models, langchain_core.language_models.fake_chat_models, and 6 more.
Where is test_tool_selection.py in the architecture?
test_tool_selection.py is located at libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_tool_selection.py (domain: LangChainCore, subdomain: ApiManagement, 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