FakeChatModelWithResponseMetadata Class — langchain Architecture

Architecture documentation for the FakeChatModelWithResponseMetadata class in test_usage_callback.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  0a450e20_5672_bfa5_c351_794cc6a4e8aa["FakeChatModelWithResponseMetadata"]
  edf81759_05d2_fa5d_d2a5_784f18a911cc["GenericFakeChatModel"]
  0a450e20_5672_bfa5_c351_794cc6a4e8aa -->|extends| edf81759_05d2_fa5d_d2a5_784f18a911cc
  b9fdf1fe_41aa_aa38_11e1_ae02e5bde19d["test_usage_callback.py"]
  0a450e20_5672_bfa5_c351_794cc6a4e8aa -->|defined in| b9fdf1fe_41aa_aa38_11e1_ae02e5bde19d
  4f031f25_a3f1_5362_4531_1a19253a70eb["_generate()"]
  0a450e20_5672_bfa5_c351_794cc6a4e8aa -->|method| 4f031f25_a3f1_5362_4531_1a19253a70eb

Relationship Graph

Source Code

libs/core/tests/unit_tests/callbacks/test_usage_callback.py lines 49–57

class FakeChatModelWithResponseMetadata(GenericFakeChatModel):
    model_name: str

    def _generate(self, *args: Any, **kwargs: Any) -> ChatResult:
        result = super()._generate(*args, **kwargs)
        result.generations[0].message.response_metadata = {
            "model_name": self.model_name
        }
        return result