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

x

Browse Source
This commit is contained in:
Eugene Yurtsev 2024-11-22 16:24:12 -05:00
parent 557f85d7fd
commit 7ea38acf2d
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/core/tests/unit_tests/runnables/test_runnable.py
View File

@ -87,6 +87,7 @@ from langchain_core.tracers import (
RunLogPatch, RunLogPatch,
) )
from langchain_core.tracers.context import collect_runs from langchain_core.tracers.context import collect_runs
from langchain_core.utils.pydantic import PYDANTIC_MAJOR_VERSION, PYDANTIC_MINOR_VERSION
from tests.unit_tests.pydantic_utils import _normalize_schema, _schema from tests.unit_tests.pydantic_utils import _normalize_schema, _schema
from tests.unit_tests.stubs import AnyStr, _any_id_ai_message, _any_id_ai_message_chunk from tests.unit_tests.stubs import AnyStr, _any_id_ai_message, _any_id_ai_message_chunk
@ -223,6 +224,13 @@ class FakeRetriever(BaseRetriever):
return [Document(page_content="foo"), Document(page_content="bar")] return [Document(page_content="foo"), Document(page_content="bar")]
@pytest.mark.skipif(
(PYDANTIC_MAJOR_VERSION, PYDANTIC_MINOR_VERSION) >= (2, 10),
reason=(
"Only test with most recent version of pydantic. "
"Pydantic introduced small fixes to generated JSONSchema on minor versions."
),
)
def test_schemas(snapshot: SnapshotAssertion) -> None: def test_schemas(snapshot: SnapshotAssertion) -> None:
fake = FakeRunnable() # str -> int fake = FakeRunnable() # str -> int