_AnotherFakeChatModel Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba["_AnotherFakeChatModel"]
  d009a608_c505_bd50_7200_0de8a69ba4b7["BaseChatModel"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|extends| d009a608_c505_bd50_7200_0de8a69ba4b7
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3["test_base.py"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|defined in| 8830054d_ac1e_daa9_c6c5_ff55b10d0bf3
  81e42e2c_b881_af7c_ae88_94ece0bc79d0["_llm_type()"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|method| 81e42e2c_b881_af7c_ae88_94ece0bc79d0
  dac28d45_baba_cfe3_92c5_a0506ff30a61["_generate()"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|method| dac28d45_baba_cfe3_92c5_a0506ff30a61
  a3259483_2d85_e90b_bd75_7bb97b0c0bc8["_agenerate()"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|method| a3259483_2d85_e90b_bd75_7bb97b0c0bc8
  a0206503_7ab1_e6fa_3c81_7f4c334be0f1["_stream()"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|method| a0206503_7ab1_e6fa_3c81_7f4c334be0f1
  3b4ca704_3700_3088_ed41_313175d54670["_astream()"]
  cf7d2d44_d1f8_20cc_397a_94eb1d099aba -->|method| 3b4ca704_3700_3088_ed41_313175d54670

Relationship Graph

Source Code

libs/core/tests/unit_tests/language_models/chat_models/test_base.py lines 923–962

class _AnotherFakeChatModel(BaseChatModel):
    responses: Iterator[AIMessage]
    """Responses for _generate."""

    chunks: Iterator[AIMessageChunk]
    """Responses for _stream."""

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

    def _generate(
        self,
        *_args: Any,
        **_kwargs: Any,
    ) -> ChatResult:
        return ChatResult(generations=[ChatGeneration(message=next(self.responses))])

    async def _agenerate(
        self,
        *_args: Any,
        **_kwargs: Any,
    ) -> ChatResult:
        return ChatResult(generations=[ChatGeneration(message=next(self.responses))])

    def _stream(
        self,
        *_args: Any,
        **_kwargs: Any,
    ) -> Iterator[ChatGenerationChunk]:
        for chunk in self.chunks:
            yield ChatGenerationChunk(message=chunk)

    async def _astream(
        self,
        *_args: Any,
        **_kwargs: Any,
    ) -> AsyncIterator[ChatGenerationChunk]:
        for chunk in self.chunks:
            yield ChatGenerationChunk(message=chunk)