Home / File/ chat_models.py — langchain Source File

chat_models.py — langchain Source File

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

File python CoreAbstractions MessageSchema 22 imports 8 functions 5 classes

Entity Profile

Dependency Diagram

graph LR
  872af98b_38b0_5d04_99e4_7f348acb4cfb["chat_models.py"]
  66c6348c_7716_027c_42d7_71449bc64eeb["base64"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 66c6348c_7716_027c_42d7_71449bc64eeb
  7025b240_fdc3_cf68_b72f_f41dac94566b["json"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 7025b240_fdc3_cf68_b72f_f41dac94566b
  9e98f0a7_ec6e_708f_4f1b_e9428b316e1c["os"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 9e98f0a7_ec6e_708f_4f1b_e9428b316e1c
  0c635125_6987_b8b3_7ff7_d60249aecde7["warnings"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 0c635125_6987_b8b3_7ff7_d60249aecde7
  8e2034b7_ceb8_963f_29fc_2ea6b50ef9b3["typing"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 8e2034b7_ceb8_963f_29fc_2ea6b50ef9b3
  525a7d6f_f455_56e3_854a_c8a7da4a1417["unittest.mock"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 525a7d6f_f455_56e3_854a_c8a7da4a1417
  1803c8c1_a347_1256_1454_9f04c3553d93["httpx"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 1803c8c1_a347_1256_1454_9f04c3553d93
  120e2591_3e15_b895_72b6_cb26195e40a6["pytest"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 120e2591_3e15_b895_72b6_cb26195e40a6
  f3bc7443_c889_119d_0744_aacc3620d8d2["langchain_core.callbacks"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> f3bc7443_c889_119d_0744_aacc3620d8d2
  ba43b74d_3099_7e1c_aac3_cf594720469e["langchain_core.language_models"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> ba43b74d_3099_7e1c_aac3_cf594720469e
  d758344f_537f_649e_f467_b9d7442e86df["langchain_core.messages"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> d758344f_537f_649e_f467_b9d7442e86df
  83d7c7fd_1989_762c_9cf3_cecb50ada22b["langchain_core.output_parsers"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 83d7c7fd_1989_762c_9cf3_cecb50ada22b
  e6b4f61e_7b98_6666_3641_26b069517d4a["langchain_core.prompts"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> e6b4f61e_7b98_6666_3641_26b069517d4a
  43d88577_548b_2248_b01b_7987bae85dcc["langchain_core.tools"]
  872af98b_38b0_5d04_99e4_7f348acb4cfb --> 43d88577_548b_2248_b01b_7987bae85dcc
  style 872af98b_38b0_5d04_99e4_7f348acb4cfb fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

"""Integration tests for chat models."""

from __future__ import annotations

import base64
import json
import os
import warnings
from typing import TYPE_CHECKING, Annotated, Any, Literal
from unittest.mock import MagicMock

import httpx
import pytest
from langchain_core.callbacks import BaseCallbackHandler
from langchain_core.language_models import BaseChatModel, GenericFakeChatModel
from langchain_core.messages import (
    AIMessage,
    AIMessageChunk,
    BaseMessage,
    HumanMessage,
    SystemMessage,
    ToolMessage,
)
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.tools import BaseTool, tool
from langchain_core.utils.function_calling import (
    convert_to_json_schema,
    tool_example_to_messages,
)
from pydantic import BaseModel, Field
from pydantic.v1 import BaseModel as BaseModelV1
from pydantic.v1 import Field as FieldV1
from typing_extensions import TypedDict, override

from langchain_tests.unit_tests.chat_models import ChatModelTests
from langchain_tests.utils.pydantic import PYDANTIC_MAJOR_VERSION

if TYPE_CHECKING:
    from pytest_benchmark.fixture import (
        BenchmarkFixture,
    )
    from vcr.cassette import Cassette


def _get_joke_class(  # noqa: RET503
    schema_type: Literal["pydantic", "typeddict", "json_schema"],
) -> Any:
    class Joke(BaseModel):
        """Joke to tell user."""

        setup: str = Field(description="question to set up a joke")
        punchline: str = Field(description="answer to resolve the joke")

    def validate_joke(result: Any) -> bool:
        return isinstance(result, Joke)

    class JokeDict(TypedDict):
        """Joke to tell user."""

// ... (3396 more lines)

Subdomains

Dependencies

  • base64
  • httpx
  • json
  • langchain_core.callbacks
  • langchain_core.language_models
  • langchain_core.messages
  • langchain_core.output_parsers
  • langchain_core.prompts
  • langchain_core.tools
  • langchain_core.utils.function_calling
  • langchain_tests.unit_tests.chat_models
  • langchain_tests.utils.pydantic
  • os
  • pydantic
  • pydantic.v1
  • pytest
  • pytest_benchmark.fixture
  • typing
  • typing_extensions
  • unittest.mock
  • vcr.cassette
  • warnings

Frequently Asked Questions

What does chat_models.py do?
chat_models.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 chat_models.py?
chat_models.py defines 8 function(s): _get_base64_from_url, _get_joke_class, _validate_tool_call_message, _validate_tool_call_message_no_args, magic_function, magic_function_no_args, pytest_benchmark, unicode_customer.
What does chat_models.py depend on?
chat_models.py imports 22 module(s): base64, httpx, json, langchain_core.callbacks, langchain_core.language_models, langchain_core.messages, langchain_core.output_parsers, langchain_core.prompts, and 14 more.
Where is chat_models.py in the architecture?
chat_models.py is located at libs/standard-tests/langchain_tests/integration_tests/chat_models.py (domain: CoreAbstractions, subdomain: MessageSchema, directory: libs/standard-tests/langchain_tests/integration_tests).

Analyze Your Own Codebase

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

Try Supermodel Free