langchain Architecture Docs
Home / Class/ CustomErrorModel Class — langchain Architecture

CustomErrorModel Class — langchain Architecture

Architecture documentation for the CustomErrorModel class in test_model_retry.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  bf8b96fd_8b4d_404f_3c0c_6a8cbaf62b72["CustomErrorModel"]
  67d9a14e_be69_a367_fd3c_cdb7f78dfa76["FakeToolCallingModel"]
  bf8b96fd_8b4d_404f_3c0c_6a8cbaf62b72 -->|extends| 67d9a14e_be69_a367_fd3c_cdb7f78dfa76
  6a114635_155a_5b86_2f48_fb8ca99f7545["test_model_retry.py"]
  bf8b96fd_8b4d_404f_3c0c_6a8cbaf62b72 -->|defined in| 6a114635_155a_5b86_2f48_fb8ca99f7545
  93e48461_3897_ea11_773a_8838a5a28da2["_generate()"]
  bf8b96fd_8b4d_404f_3c0c_6a8cbaf62b72 -->|method| 93e48461_3897_ea11_773a_8838a5a28da2

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_retry.py lines 340–366

    class CustomErrorModel(FakeToolCallingModel):
        """Model that raises CustomError."""

        def _generate(
            self,
            messages: list[BaseMessage],
            stop: list[str] | None = None,
            run_manager: CallbackManagerForLLMRun | None = None,
            **kwargs: Any,
        ) -> ChatResult:
            """Execute the model and raise CustomError.

            Args:
                messages: Input messages.
                stop: Optional stop sequences.
                run_manager: Optional callback manager.
                **kwargs: Additional keyword arguments.

            Raises:
                CustomError: Always raises CustomError.
            """
            attempt_count["value"] += 1
            if attempt_count["value"] == 1:
                msg = "Retryable error"
                raise CustomError(msg, retry_me=True)
            msg = "Non-retryable error"
            raise CustomError(msg, retry_me=False)

Defined In

Extends

Source

Frequently Asked Questions

What is the CustomErrorModel class?
CustomErrorModel is a class in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_retry.py.
Where is CustomErrorModel defined?
CustomErrorModel is defined in libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_retry.py at line 340.
What does CustomErrorModel extend?
CustomErrorModel extends FakeToolCallingModel.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free