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
  18dfd43b_adae_9f79_6df8_8c59d3080176["ModelWithGenerate"]
  76008743_94c2_792b_9e4f_e1278ae47565["BaseLLM"]
  18dfd43b_adae_9f79_6df8_8c59d3080176 -->|extends| 76008743_94c2_792b_9e4f_e1278ae47565
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3["test_base.py"]
  18dfd43b_adae_9f79_6df8_8c59d3080176 -->|defined in| 4404310a_b16c_e9e6_1de2_90e5ea6f3ce3
  6fde93f4_513d_1a3f_701f_1a4d762ee752["_generate()"]
  18dfd43b_adae_9f79_6df8_8c59d3080176 -->|method| 6fde93f4_513d_1a3f_701f_1a4d762ee752
  5ddb6fca_46b7_b6c6_10aa_c263c14caa90["_llm_type()"]
  18dfd43b_adae_9f79_6df8_8c59d3080176 -->|method| 5ddb6fca_46b7_b6c6_10aa_c263c14caa90

Relationship Graph

Source Code

libs/core/tests/unit_tests/language_models/llms/test_base.py lines 140–154

    class ModelWithGenerate(BaseLLM):
        @override
        def _generate(
            self,
            prompts: list[str],
            stop: list[str] | None = None,
            run_manager: CallbackManagerForLLMRun | None = None,
            **kwargs: Any,
        ) -> LLMResult:
            generations = [Generation(text="hello")]
            return LLMResult(generations=[generations])

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