mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-17 16:39:52 +00:00
lint
This commit is contained in:
parent
0f4a4c9f86
commit
6564669a14
@ -6,7 +6,7 @@ from unittest.mock import MagicMock
|
||||
|
||||
import httpx
|
||||
import pytest
|
||||
import vcr
|
||||
import vcr # type: ignore[import-untyped]
|
||||
from langchain_core._api import warn_deprecated
|
||||
from langchain_core.callbacks import BaseCallbackHandler
|
||||
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.v1 import BaseModel as BaseModelV1
|
||||
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 langchain_tests.unit_tests.chat_models import (
|
||||
@ -2685,7 +2685,7 @@ class ChatModelIntegrationTests(ChatModelTests):
|
||||
def test_stream_time(
|
||||
self, model: BaseChatModel, benchmark: BenchmarkFixture, vcr: vcr.VCR
|
||||
) -> None:
|
||||
def _run():
|
||||
def _run() -> None:
|
||||
cassette_name = f"{self.__class__.__name__}_test_stream_time.yaml"
|
||||
with vcr.use_cassette(cassette_name, record_mode="once"):
|
||||
for _ in model.stream("Write a story about a cat."):
|
||||
|
@ -23,7 +23,7 @@ from pydantic.v1 import (
|
||||
from pydantic.v1 import (
|
||||
ValidationError as ValidationErrorV1,
|
||||
)
|
||||
from pytest_benchmark.fixture import BenchmarkFixture
|
||||
from pytest_benchmark.fixture import BenchmarkFixture # type: ignore[import-untyped]
|
||||
from syrupy import SnapshotAssertion
|
||||
|
||||
from langchain_tests.base import BaseStandardTests
|
||||
|
Loading…
Reference in New Issue
Block a user