FakeChatModelStartTracer Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  6f69cef3_734e_618d_7b6a_fead428f256a["FakeChatModelStartTracer"]
  e5bf61e0_74ec_5f37_3118_e41c1debe1c6["FakeTracer"]
  6f69cef3_734e_618d_7b6a_fead428f256a -->|extends| e5bf61e0_74ec_5f37_3118_e41c1debe1c6
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3["test_base.py"]
  6f69cef3_734e_618d_7b6a_fead428f256a -->|defined in| 8830054d_ac1e_daa9_c6c5_ff55b10d0bf3
  05e85b8a_0de9_d1f4_6afb_e4753fcc12fe["__init__()"]
  6f69cef3_734e_618d_7b6a_fead428f256a -->|method| 05e85b8a_0de9_d1f4_6afb_e4753fcc12fe
  847620a3_5d68_a2b2_c32f_881e478a4e62["on_chat_model_start()"]
  6f69cef3_734e_618d_7b6a_fead428f256a -->|method| 847620a3_5d68_a2b2_c32f_881e478a4e62

Relationship Graph

Source Code

libs/core/tests/unit_tests/language_models/chat_models/test_base.py lines 474–485

class FakeChatModelStartTracer(FakeTracer):
    def __init__(self) -> None:
        super().__init__()
        self.messages: list[list[list[BaseMessage]]] = []

    def on_chat_model_start(self, *args: Any, **kwargs: Any) -> Run:
        _, messages = args
        self.messages.append(messages)
        return super().on_chat_model_start(
            *args,
            **kwargs,
        )