test_base.py — langchain Source File

Architecture documentation for test_base.py, a python file in the langchain codebase. 21 imports, 0 dependents.

File python CoreAbstractions MessageSchema 21 imports 32 functions 11 classes

Entity Profile

CoreAbstractions→ MessageSchema→ test_base.py — langchain Source File

Dependency Diagram

graph LR
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3["test_base.py"]
  8dfa0cac_d802_3ccd_f710_43a5e70da3a5["uuid"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 8dfa0cac_d802_3ccd_f710_43a5e70da3a5
  0c635125_6987_b8b3_7ff7_d60249aecde7["warnings"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 0c635125_6987_b8b3_7ff7_d60249aecde7
  cfe2bde5_180e_e3b0_df2b_55b3ebaca8e7["collections.abc"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> cfe2bde5_180e_e3b0_df2b_55b3ebaca8e7
  8e2034b7_ceb8_963f_29fc_2ea6b50ef9b3["typing"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 8e2034b7_ceb8_963f_29fc_2ea6b50ef9b3
  120e2591_3e15_b895_72b6_cb26195e40a6["pytest"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 120e2591_3e15_b895_72b6_cb26195e40a6
  91721f45_4909_e489_8c1f_084f8bd87145["typing_extensions"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 91721f45_4909_e489_8c1f_084f8bd87145
  f3bc7443_c889_119d_0744_aacc3620d8d2["langchain_core.callbacks"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> f3bc7443_c889_119d_0744_aacc3620d8d2
  ba43b74d_3099_7e1c_aac3_cf594720469e["langchain_core.language_models"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> ba43b74d_3099_7e1c_aac3_cf594720469e
  87a7e59a_c218_0854_8192_d59035f1c4b0["langchain_core.language_models._utils"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 87a7e59a_c218_0854_8192_d59035f1c4b0
  2312f229_c199_ac88_c29f_62e2a2958404["langchain_core.language_models.chat_models"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 2312f229_c199_ac88_c29f_62e2a2958404
  833aeadc_c3e9_bfcf_db07_ecb37ad3ba24["langchain_core.language_models.fake_chat_models"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 833aeadc_c3e9_bfcf_db07_ecb37ad3ba24
  d758344f_537f_649e_f467_b9d7442e86df["langchain_core.messages"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> d758344f_537f_649e_f467_b9d7442e86df
  ac2a9b92_4484_491e_1b48_ec85e71e1d58["langchain_core.outputs"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> ac2a9b92_4484_491e_1b48_ec85e71e1d58
  884fdf32_1e49_8c1f_c69e_7f622e672b39["langchain_core.outputs.llm_result"]
  8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 --> 884fdf32_1e49_8c1f_c69e_7f622e672b39
  style 8830054d_ac1e_daa9_c6c5_ff55b10d0bf3 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

"""Test base chat model."""

import uuid
import warnings
from collections.abc import AsyncIterator, Iterator
from typing import TYPE_CHECKING, Any, Literal

import pytest
from typing_extensions import override

from langchain_core.callbacks import (
    CallbackManagerForLLMRun,
)
from langchain_core.language_models import (
    BaseChatModel,
    FakeListChatModel,
    ParrotFakeChatModel,
)
from langchain_core.language_models._utils import _normalize_messages
from langchain_core.language_models.chat_models import _generate_response_from_error
from langchain_core.language_models.fake_chat_models import (
    FakeListChatModelError,
    GenericFakeChatModel,
)
from langchain_core.messages import (
    AIMessage,
    AIMessageChunk,
    BaseMessage,
    HumanMessage,
    SystemMessage,
)
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
from langchain_core.outputs.llm_result import LLMResult
from langchain_core.tracers import LogStreamCallbackHandler
from langchain_core.tracers.base import BaseTracer
from langchain_core.tracers.context import collect_runs
from langchain_core.tracers.event_stream import _AstreamEventsCallbackHandler
from langchain_core.tracers.schemas import Run
from tests.unit_tests.fake.callbacks import (
    BaseFakeCallbackHandler,
    FakeAsyncCallbackHandler,
    FakeCallbackHandler,
)
from tests.unit_tests.stubs import _any_id_ai_message, _any_id_ai_message_chunk

if TYPE_CHECKING:
    from langchain_core.outputs.llm_result import LLMResult


def _content_blocks_equal_ignore_id(
    actual: str | list[Any], expected: str | list[Any]
) -> bool:
    """Compare content blocks, ignoring auto-generated `id` fields.

    Args:
        actual: Actual content from response (string or list of content blocks).
        expected: Expected content to compare against (string or list of blocks).

    Returns:
        True if content matches (excluding `id` fields), `False` otherwise.
// ... (1254 more lines)

Domain

CoreAbstractions

Subdomains

MessageSchema

Functions

Classes

Dependencies

collections.abc
langchain_core.callbacks
langchain_core.language_models
langchain_core.language_models._utils
langchain_core.language_models.chat_models
langchain_core.language_models.fake_chat_models
langchain_core.messages
langchain_core.outputs
langchain_core.outputs.llm_result
langchain_core.tracers
langchain_core.tracers.base
langchain_core.tracers.context
langchain_core.tracers.event_stream
langchain_core.tracers.schemas
pytest
tests.unit_tests.fake.callbacks
tests.unit_tests.stubs
typing
typing_extensions
uuid
warnings

Source

View on GitHub

Frequently Asked Questions

What does test_base.py do?

test_base.py is a source file in the langchain codebase, written in python. It belongs to the CoreAbstractions domain, MessageSchema subdomain.

What functions are defined in test_base.py?

test_base.py defines 32 function(s): _content_blocks_equal_ignore_id, langchain_core, messages, messages_2, test_astream_fallback_to_ainvoke, test_astream_implementation_fallback_to_stream, test_astream_implementation_uses_astream, test_async_batch_size, test_async_pass_run_id, test_batch_size, and 22 more.

What does test_base.py depend on?

test_base.py imports 21 module(s): collections.abc, langchain_core.callbacks, langchain_core.language_models, langchain_core.language_models._utils, langchain_core.language_models.chat_models, langchain_core.language_models.fake_chat_models, langchain_core.messages, langchain_core.outputs, and 13 more.

Where is test_base.py in the architecture?

test_base.py is located at libs/core/tests/unit_tests/language_models/chat_models/test_base.py (domain: CoreAbstractions, subdomain: MessageSchema, directory: libs/core/tests/unit_tests/language_models/chat_models).

Analyze Your Own Codebase

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

Try Supermodel Free