wrap_model_call() — langchain Function Reference

Architecture documentation for the wrap_model_call() function in test_tools.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ wrap_model_call() — langchain Function Reference

Dependency Diagram

graph TD
  4a265de5_1026_9487_8bca_c3af5c5ec900["wrap_model_call()"]
  13463169_9081_23c5_190b_eb716b4ed9a0["ConditionalToolMiddleware"]
  4a265de5_1026_9487_8bca_c3af5c5ec900 -->|defined in| 13463169_9081_23c5_190b_eb716b4ed9a0
  style 4a265de5_1026_9487_8bca_c3af5c5ec900 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_tools.py lines 181–194

        def wrap_model_call(
            self,
            request: ModelRequest,
            handler: Callable[[ModelRequest], ModelResponse],
        ) -> ModelCallResult:
            # Remove admin_tool if not admin
            if not request.state.get("is_admin", False):
                filtered_tools: list[BaseTool | dict[str, Any]] = []
                for t in request.tools:
                    assert isinstance(t, BaseTool)
                    if t.name != "admin_tool":
                        filtered_tools.append(t)
                request = request.override(tools=filtered_tools)
            return handler(request)