UnreliableModel Class — langchain Architecture

Architecture documentation for the UnreliableModel class in test_wrap_model_call.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  ce9e9b65_f780_1a57_3312_ab80c1efea9a["UnreliableModel"]
  0f829642_e361_67a1_cbca_f82839bbbd31["GenericFakeChatModel"]
  ce9e9b65_f780_1a57_3312_ab80c1efea9a -->|extends| 0f829642_e361_67a1_cbca_f82839bbbd31
  573473e5_eb93_a7d7_84ba_c975071c09af["test_wrap_model_call.py"]
  ce9e9b65_f780_1a57_3312_ab80c1efea9a -->|defined in| 573473e5_eb93_a7d7_84ba_c975071c09af
  493c1571_a532_76e9_d5e5_71508277b22c["_generate()"]
  ce9e9b65_f780_1a57_3312_ab80c1efea9a -->|method| 493c1571_a532_76e9_d5e5_71508277b22c

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/core/test_wrap_model_call.py lines 1308–1321

        class UnreliableModel(GenericFakeChatModel):
            @override
            def _generate(
                self,
                messages: list[BaseMessage],
                stop: list[str] | None = None,
                run_manager: CallbackManagerForLLMRun | None = None,
                **kwargs: Any,
            ) -> ChatResult:
                call_count["value"] += 1
                if call_count["value"] <= 2:
                    msg = f"Attempt {call_count['value']} failed"
                    raise ValueError(msg)
                return super()._generate(messages, **kwargs)