test_summarization.py — langchain Source File

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

File python LangChainCore ApiManagement 20 imports 32 functions 12 classes

Entity Profile

LangChainCore→ ApiManagement→ test_summarization.py — langchain Source File

Dependency Diagram

graph LR
  1911a463_b67d_0301_5ef1_5c535dafc14a["test_summarization.py"]
  2bf6d401_816d_d011_3b05_a6114f55ff58["collections.abc"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 2bf6d401_816d_d011_3b05_a6114f55ff58
  feec1ec4_6917_867b_d228_b134d0ff8099["typing"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> feec1ec4_6917_867b_d228_b134d0ff8099
  23cb242e_1754_041d_200a_553fcb8abe1b["unittest.mock"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 23cb242e_1754_041d_200a_553fcb8abe1b
  f69d6389_263d_68a4_7fbf_f14c0602a9ba["pytest"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> f69d6389_263d_68a4_7fbf_f14c0602a9ba
  17a62cb3_fefd_6320_b757_b53bb4a1c661["langchain_core.callbacks"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 17a62cb3_fefd_6320_b757_b53bb4a1c661
  e929cf21_6ab8_6ff3_3765_0d35a099a053["langchain_core.language_models"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> e929cf21_6ab8_6ff3_3765_0d35a099a053
  242b5354_4ca1_727c_dd7a_12de2b570803["langchain_core.language_models.base"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 242b5354_4ca1_727c_dd7a_12de2b570803
  f77bacd6_1087_c145_d5d5_68e6f3937f9d["langchain_core.language_models.chat_models"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> f77bacd6_1087_c145_d5d5_68e6f3937f9d
  9444498b_8066_55c7_b3a2_1d90c4162a32["langchain_core.messages"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 9444498b_8066_55c7_b3a2_1d90c4162a32
  a188dc39_6da2_34b5_82d2_3c9b8af6f61c["langchain_core.messages.utils"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> a188dc39_6da2_34b5_82d2_3c9b8af6f61c
  4382dc25_6fba_324a_49e2_e9742d579385["langchain_core.outputs"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 4382dc25_6fba_324a_49e2_e9742d579385
  31eab4ab_7281_1e6c_b17d_12e6ad9de07a["langchain_core.runnables"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> 31eab4ab_7281_1e6c_b17d_12e6ad9de07a
  a2ed5244_833f_36b3_ad87_753172a0ffd1["langgraph.graph.message"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> a2ed5244_833f_36b3_ad87_753172a0ffd1
  e07f6d54_afcc_052d_d33f_8ccdcc46f752["langgraph.runtime"]
  1911a463_b67d_0301_5ef1_5c535dafc14a --> e07f6d54_afcc_052d_d33f_8ccdcc46f752
  style 1911a463_b67d_0301_5ef1_5c535dafc14a fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

from collections.abc import Iterable
from typing import Any
from unittest.mock import patch

import pytest
from langchain_core.callbacks import AsyncCallbackManagerForLLMRun, CallbackManagerForLLMRun
from langchain_core.language_models import ModelProfile
from langchain_core.language_models.base import (
    LanguageModelInput,
)
from langchain_core.language_models.chat_models import BaseChatModel
from langchain_core.messages import (
    AIMessage,
    AnyMessage,
    BaseMessage,
    HumanMessage,
    MessageLikeRepresentation,
    RemoveMessage,
    ToolMessage,
)
from langchain_core.messages.utils import count_tokens_approximately, get_buffer_string
from langchain_core.outputs import ChatGeneration, ChatResult
from langchain_core.runnables import RunnableConfig
from langgraph.graph.message import REMOVE_ALL_MESSAGES
from langgraph.runtime import Runtime
from pydantic import Field
from typing_extensions import override

from langchain.agents import AgentState
from langchain.agents.middleware.summarization import SummarizationMiddleware
from langchain.chat_models import init_chat_model
from tests.unit_tests.agents.model import FakeToolCallingModel


class MockChatModel(BaseChatModel):
    """Mock chat model for testing."""

    @override
    def invoke(
        self,
        input: LanguageModelInput,
        config: RunnableConfig | None = None,
        *,
        stop: list[str] | None = None,
        **kwargs: Any,
    ) -> AIMessage:
        return AIMessage(content="Generated summary")

    @override
    def _generate(
        self,
        messages: list[BaseMessage],
        stop: list[str] | None = None,
        run_manager: CallbackManagerForLLMRun | None = None,
        **kwargs: Any,
    ) -> ChatResult:
        return ChatResult(generations=[ChatGeneration(message=AIMessage(content="Summary"))])

    @property
    def _llm_type(self) -> str:
// ... (1253 more lines)

Domain

LangChainCore

Subdomains

ApiManagement

Functions

Classes

Dependencies

collections.abc
langchain.agents
langchain.agents.middleware.summarization
langchain.chat_models
langchain_core.callbacks
langchain_core.language_models
langchain_core.language_models.base
langchain_core.language_models.chat_models
langchain_core.messages
langchain_core.messages.utils
langchain_core.outputs
langchain_core.runnables
langgraph.graph.message
langgraph.runtime
pydantic
pytest
tests.unit_tests.agents.model
typing
typing_extensions
unittest.mock

Source

View on GitHub

Frequently Asked Questions

What does test_summarization.py do?

test_summarization.py is a source file in the langchain codebase, written in python. It belongs to the LangChainCore domain, ApiManagement subdomain.

What functions are defined in test_summarization.py?

test_summarization.py defines 32 function(s): test_create_summary_passes_lc_source_metadata, test_create_summary_uses_get_buffer_string_format, test_summarization_adjust_token_counts, test_summarization_before_model_uses_unscaled_tokens_for_cutoff, test_summarization_cutoff_moves_backward_to_include_ai_message, test_summarization_middleware_async_error_handling, test_summarization_middleware_binary_search_edge_cases, test_summarization_middleware_cutoff_at_boundary, test_summarization_middleware_cutoff_at_start_of_tool_sequence, test_summarization_middleware_deprecated_parameters_with_defaults, and 22 more.

What does test_summarization.py depend on?

test_summarization.py imports 20 module(s): collections.abc, langchain.agents, langchain.agents.middleware.summarization, langchain.chat_models, langchain_core.callbacks, langchain_core.language_models, langchain_core.language_models.base, langchain_core.language_models.chat_models, and 12 more.

Where is test_summarization.py in the architecture?

test_summarization.py is located at libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_summarization.py (domain: LangChainCore, subdomain: ApiManagement, directory: libs/langchain_v1/tests/unit_tests/agents/middleware/implementations).

Analyze Your Own Codebase

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

Try Supermodel Free