langchain Architecture Docs
Home / Class/ MockChatModel Class — langchain Architecture

MockChatModel Class — langchain Architecture

Architecture documentation for the MockChatModel class in test_summarization.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  0046c1d9_5751_2342_39c7_f6d868ce1c43["MockChatModel"]
  48aa29b8_65e7_522f_a445_a441eeb6baff["BaseChatModel"]
  0046c1d9_5751_2342_39c7_f6d868ce1c43 -->|extends| 48aa29b8_65e7_522f_a445_a441eeb6baff
  1911a463_b67d_0301_5ef1_5c535dafc14a["test_summarization.py"]
  0046c1d9_5751_2342_39c7_f6d868ce1c43 -->|defined in| 1911a463_b67d_0301_5ef1_5c535dafc14a
  e2eebc26_55e4_d579_e031_4afa56f0e222["invoke()"]
  0046c1d9_5751_2342_39c7_f6d868ce1c43 -->|method| e2eebc26_55e4_d579_e031_4afa56f0e222
  58d9edd6_f9c8_25b5_3fb4_0f2d1b04e08c["_generate()"]
  0046c1d9_5751_2342_39c7_f6d868ce1c43 -->|method| 58d9edd6_f9c8_25b5_3fb4_0f2d1b04e08c
  ff94def4_5ea2_903f_e9d8_d2114213ec7f["_llm_type()"]
  0046c1d9_5751_2342_39c7_f6d868ce1c43 -->|method| ff94def4_5ea2_903f_e9d8_d2114213ec7f

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_summarization.py lines 35–61

class MockChatModel(BaseChatModel):
    """Mock chat model for testing."""

    @override
    def invoke(
        self,
        input: LanguageModelInput,
        config: RunnableConfig | None = None,
        *,
        stop: list[str] | None = None,
        **kwargs: Any,
    ) -> AIMessage:
        return AIMessage(content="Generated summary")

    @override
    def _generate(
        self,
        messages: list[BaseMessage],
        stop: list[str] | None = None,
        run_manager: CallbackManagerForLLMRun | None = None,
        **kwargs: Any,
    ) -> ChatResult:
        return ChatResult(generations=[ChatGeneration(message=AIMessage(content="Summary"))])

    @property
    def _llm_type(self) -> str:
        return "mock"

Defined In

Extends

Source

Frequently Asked Questions

What is the MockChatModel class?
MockChatModel is a class in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_summarization.py.
Where is MockChatModel defined?
MockChatModel is defined in libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_summarization.py at line 35.
What does MockChatModel extend?
MockChatModel extends BaseChatModel.

Analyze Your Own Codebase

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

Try Supermodel Free