github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-17 16:39:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Chester Curme 2025-05-21 09:58:17 -04:00
parent 0f4a4c9f86
commit 6564669a14
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/standard-tests/langchain_tests/integration_tests/chat_models.py
View File

@ -6,7 +6,7 @@ from unittest.mock import MagicMock
import httpx import httpx
import pytest import pytest
import vcr import vcr # type: ignore[import-untyped]
from langchain_core._api import warn_deprecated from langchain_core._api import warn_deprecated
from langchain_core.callbacks import BaseCallbackHandler from langchain_core.callbacks import BaseCallbackHandler
from langchain_core.language_models import BaseChatModel, GenericFakeChatModel from langchain_core.language_models import BaseChatModel, GenericFakeChatModel
@ -29,7 +29,7 @@ from langchain_core.utils.function_calling import (
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from pydantic.v1 import BaseModel as BaseModelV1 from pydantic.v1 import BaseModel as BaseModelV1
from pydantic.v1 import Field as FieldV1 from pydantic.v1 import Field as FieldV1
from pytest_benchmark.fixture import BenchmarkFixture from pytest_benchmark.fixture import BenchmarkFixture # type: ignore[import-untyped]
from typing_extensions import Annotated, TypedDict from typing_extensions import Annotated, TypedDict
from langchain_tests.unit_tests.chat_models import ( from langchain_tests.unit_tests.chat_models import (
@ -2685,7 +2685,7 @@ class ChatModelIntegrationTests(ChatModelTests):
def test_stream_time( def test_stream_time(
self, model: BaseChatModel, benchmark: BenchmarkFixture, vcr: vcr.VCR self, model: BaseChatModel, benchmark: BenchmarkFixture, vcr: vcr.VCR
) -> None: ) -> None:
def _run(): def _run() -> None:
cassette_name = f"{self.__class__.__name__}_test_stream_time.yaml" cassette_name = f"{self.__class__.__name__}_test_stream_time.yaml"
with vcr.use_cassette(cassette_name, record_mode="once"): with vcr.use_cassette(cassette_name, record_mode="once"):
for _ in model.stream("Write a story about a cat."): for _ in model.stream("Write a story about a cat."):

2
libs/standard-tests/langchain_tests/unit_tests/chat_models.py
View File

@ -23,7 +23,7 @@ from pydantic.v1 import (
from pydantic.v1 import ( from pydantic.v1 import (
ValidationError as ValidationErrorV1, ValidationError as ValidationErrorV1,
) )
from pytest_benchmark.fixture import BenchmarkFixture from pytest_benchmark.fixture import BenchmarkFixture # type: ignore[import-untyped]
from syrupy import SnapshotAssertion from syrupy import SnapshotAssertion
from langchain_tests.base import BaseStandardTests from langchain_tests.base import BaseStandardTests