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

fmt

Browse Source
This commit is contained in:
Bagatur 2024-09-04 11:12:11 -07:00
parent 576574c82c
commit fdf6fbde18
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/standard-tests/langchain_standard_tests/unit_tests/chat_models.py
View File

@ -5,13 +5,12 @@ from typing import Any, List, Literal, Optional, Tuple, Type
from unittest import mock from unittest import mock
import pytest import pytest
from syrupy import SnapshotAssertion
from langchain_core.language_models import BaseChatModel from langchain_core.language_models import BaseChatModel
from langchain_core.load import dumpd, load from langchain_core.load import dumpd, load
from langchain_core.pydantic_v1 import BaseModel, Field, SecretStr from langchain_core.pydantic_v1 import BaseModel, Field, SecretStr
from langchain_core.runnables import RunnableBinding from langchain_core.runnables import RunnableBinding
from langchain_core.tools import tool from langchain_core.tools import tool
from syrupy import SnapshotAssertion
from langchain_standard_tests.base import BaseStandardTests from langchain_standard_tests.base import BaseStandardTests
from langchain_standard_tests.utils.pydantic import PYDANTIC_MAJOR_VERSION from langchain_standard_tests.utils.pydantic import PYDANTIC_MAJOR_VERSION
@ -231,6 +230,8 @@ class ChatModelUnitTests(ChatModelTests):
def test_serdes(self, model: BaseChatModel, snapshot: SnapshotAssertion) -> None: def test_serdes(self, model: BaseChatModel, snapshot: SnapshotAssertion) -> None:
if not self.chat_model_class.is_lc_serializable(): if not self.chat_model_class.is_lc_serializable():
return return
ser = dumpd(model) env_params, model_params, expected_attrs = self.init_from_env_params
assert ser == snapshot(name="serialized") with mock.patch.dict(os.environ, env_params):
assert model.dict() == load(dumpd(model)).dict() ser = dumpd(model)
assert ser == snapshot(name="serialized")
assert model.dict() == load(dumpd(model)).dict()