wrap_model_call() — langchain Function Reference

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

Function python LangChainCore LanguageModelBase calls 1 called by 2

Entity Profile

LangChainCore→ LanguageModelBase→ wrap_model_call() — langchain Function Reference

Dependency Diagram

graph TD
  54e11611_aa23_831b_505a_42eddfb4a8f7["wrap_model_call()"]
  a8d1868f_6ff2_5369_173c_c9fc7f2da6be["CustomMiddleware"]
  54e11611_aa23_831b_505a_42eddfb4a8f7 -->|defined in| a8d1868f_6ff2_5369_173c_c9fc7f2da6be
  19d59b52_7973_0844_eb57_2971a2ebf227["wrap_model_call()"]
  19d59b52_7973_0844_eb57_2971a2ebf227 -->|calls| 54e11611_aa23_831b_505a_42eddfb4a8f7
  52a9340b_bf14_42cd_68f6_292be68d8d1e["wrap_model_call()"]
  52a9340b_bf14_42cd_68f6_292be68d8d1e -->|calls| 54e11611_aa23_831b_505a_42eddfb4a8f7
  19d59b52_7973_0844_eb57_2971a2ebf227["wrap_model_call()"]
  54e11611_aa23_831b_505a_42eddfb4a8f7 -->|calls| 19d59b52_7973_0844_eb57_2971a2ebf227
  style 54e11611_aa23_831b_505a_42eddfb4a8f7 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_framework.py lines 392–398

        def wrap_model_call(
            self,
            request: ModelRequest,
            handler: Callable[[ModelRequest], ModelResponse],
        ) -> ModelCallResult:
            assert request.runtime is not None
            return handler(request)