ModelWithGenerate Class — langchain Architecture

Architecture documentation for the ModelWithGenerate class in test_base.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  239a3d4d_82d5_1405_918c_8dafa29e1e08["ModelWithGenerate"]
  d009a608_c505_bd50_7200_0de8a69ba4b7["BaseChatModel"]
  239a3d4d_82d5_1405_918c_8dafa29e1e08 -->|extends| d009a608_c505_bd50_7200_0de8a69ba4b7
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3["test_base.py"]
  239a3d4d_82d5_1405_918c_8dafa29e1e08 -->|defined in| 8830054d_ac1e_daa9_c6c5_ff55b10d0bf3
  18272b46_7612_622b_e347_9f04654cf936["_generate()"]
  239a3d4d_82d5_1405_918c_8dafa29e1e08 -->|method| 18272b46_7612_622b_e347_9f04654cf936
  1a5dabf9_50c0_0db3_d938_c1e9c7d0fc1c["_llm_type()"]
  239a3d4d_82d5_1405_918c_8dafa29e1e08 -->|method| 1a5dabf9_50c0_0db3_d938_c1e9c7d0fc1c

Relationship Graph

Source Code

libs/core/tests/unit_tests/language_models/chat_models/test_base.py lines 186–202

    class ModelWithGenerate(BaseChatModel):
        @override
        def _generate(
            self,
            messages: list[BaseMessage],
            stop: list[str] | None = None,
            run_manager: CallbackManagerForLLMRun | None = None,
            **kwargs: Any,
        ) -> ChatResult:
            """Top Level call."""
            message = AIMessage(content="hello")
            generation = ChatGeneration(message=message)
            return ChatResult(generations=[generation])

        @property
        def _llm_type(self) -> str:
            return "fake-chat-model"