core[patch]: Compat pydantic 2.10 (#28308)

pydantic 2.10 compat for langchain-core
langchain-ai · Nov 23, 2024 · a813d11 · a813d11
1 parent ed84d48
commit a813d11
Show file tree

Hide file tree

Showing 9 changed files with 199 additions and 113 deletions.
diff --git a/libs/core/langchain_core/runnables/base.py b/libs/core/langchain_core/runnables/base.py
@@ -5217,7 +5217,7 @@ class RunnableBindingBase(RunnableSerializable[Input, Output]):
     kwargs.
     """
 
-    config: RunnableConfig = Field(default_factory=dict)
+    config: RunnableConfig = Field(default_factory=RunnableConfig)  # type: ignore
     """The config to bind to the underlying Runnable."""
 
     config_factories: list[Callable[[RunnableConfig], RunnableConfig]] = Field(

diff --git a/libs/core/langchain_core/utils/pydantic.py b/libs/core/langchain_core/utils/pydantic.py
@@ -48,7 +48,18 @@ def get_pydantic_major_version() -> int:
         return 0
 
 
+def _get_pydantic_minor_version() -> int:
+    """Get the minor version of Pydantic."""
+    try:
+        import pydantic
+
+        return int(pydantic.__version__.split(".")[1])
+    except ImportError:
+        return 0
+
+
 PYDANTIC_MAJOR_VERSION = get_pydantic_major_version()
+PYDANTIC_MINOR_VERSION = _get_pydantic_minor_version()
 
 
 if PYDANTIC_MAJOR_VERSION == 1:
@@ -200,7 +211,7 @@ def wrapper(cls: type[BaseModel], values: dict[str, Any]) -> dict[str, Any]:
                     name not in values or values[name] is None
                 ) and not field_info.is_required():
                     if field_info.default_factory is not None:
-                        values[name] = field_info.default_factory()
+                        values[name] = field_info.default_factory()  # type: ignore
                     else:
                         values[name] = field_info.default
 

diff --git a/libs/core/poetry.lock b/libs/core/poetry.lock
diff --git a/libs/core/tests/unit_tests/prompts/__snapshots__/test_chat.ambr b/libs/core/tests/unit_tests/prompts/__snapshots__/test_chat.ambr
@@ -94,6 +94,7 @@
             'const': 'ai',
             'default': 'ai',
             'title': 'Type',
+            'type': 'string',
           }),
           'usage_metadata': dict({
             'anyOf': list([
@@ -206,6 +207,7 @@
             'const': 'AIMessageChunk',
             'default': 'AIMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
           'usage_metadata': dict({
             'anyOf': list([
@@ -290,6 +292,7 @@
             'const': 'chat',
             'default': 'chat',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -364,6 +367,7 @@
             'const': 'ChatMessageChunk',
             'default': 'ChatMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -435,6 +439,7 @@
             'const': 'function',
             'default': 'function',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -497,6 +502,7 @@
             'const': 'FunctionMessageChunk',
             'default': 'FunctionMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -595,6 +601,7 @@
             'const': 'human',
             'default': 'human',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -669,6 +676,7 @@
             'const': 'HumanMessageChunk',
             'default': 'HumanMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -767,6 +775,7 @@
           'type': dict({
             'const': 'invalid_tool_call',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -892,6 +901,7 @@
             'const': 'system',
             'default': 'system',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -961,6 +971,7 @@
             'const': 'SystemMessageChunk',
             'default': 'SystemMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1009,6 +1020,7 @@
           'type': dict({
             'const': 'tool_call',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1087,6 +1099,7 @@
           'type': dict({
             'const': 'tool_call_chunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1209,6 +1222,7 @@
             'const': 'tool',
             'default': 'tool',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1290,6 +1304,7 @@
             'const': 'ToolMessageChunk',
             'default': 'ToolMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1509,6 +1524,7 @@
             'const': 'ai',
             'default': 'ai',
             'title': 'Type',
+            'type': 'string',
           }),
           'usage_metadata': dict({
             'anyOf': list([
@@ -1621,6 +1637,7 @@
             'const': 'AIMessageChunk',
             'default': 'AIMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
           'usage_metadata': dict({
             'anyOf': list([
@@ -1705,6 +1722,7 @@
             'const': 'chat',
             'default': 'chat',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1779,6 +1797,7 @@
             'const': 'ChatMessageChunk',
             'default': 'ChatMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1850,6 +1869,7 @@
             'const': 'function',
             'default': 'function',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -1912,6 +1932,7 @@
             'const': 'FunctionMessageChunk',
             'default': 'FunctionMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2010,6 +2031,7 @@
             'const': 'human',
             'default': 'human',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2084,6 +2106,7 @@
             'const': 'HumanMessageChunk',
             'default': 'HumanMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2182,6 +2205,7 @@
           'type': dict({
             'const': 'invalid_tool_call',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2307,6 +2331,7 @@
             'const': 'system',
             'default': 'system',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2376,6 +2401,7 @@
             'const': 'SystemMessageChunk',
             'default': 'SystemMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2424,6 +2450,7 @@
           'type': dict({
             'const': 'tool_call',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2502,6 +2529,7 @@
           'type': dict({
             'const': 'tool_call_chunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2624,6 +2652,7 @@
             'const': 'tool',
             'default': 'tool',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([
@@ -2705,6 +2734,7 @@
             'const': 'ToolMessageChunk',
             'default': 'ToolMessageChunk',
             'title': 'Type',
+            'type': 'string',
           }),
         }),
         'required': list([

diff --git a/libs/core/tests/unit_tests/prompts/test_chat.py b/libs/core/tests/unit_tests/prompts/test_chat.py
@@ -33,6 +33,7 @@
     _convert_to_message,
 )
 from langchain_core.prompts.string import PromptTemplateFormat
+from langchain_core.utils.pydantic import PYDANTIC_MAJOR_VERSION, PYDANTIC_MINOR_VERSION
 from tests.unit_tests.pydantic_utils import _normalize_schema
 
 
@@ -852,18 +853,22 @@ def test_chat_input_schema(snapshot: SnapshotAssertion) -> None:
     assert prompt_all_required.optional_variables == []
     with pytest.raises(ValidationError):
         prompt_all_required.input_schema(input="")
-    assert _normalize_schema(prompt_all_required.get_input_jsonschema()) == snapshot(
-        name="required"
-    )
+
+    if (PYDANTIC_MAJOR_VERSION, PYDANTIC_MINOR_VERSION) >= (2, 10):
+        assert _normalize_schema(
+            prompt_all_required.get_input_jsonschema()
+        ) == snapshot(name="required")
     prompt_optional = ChatPromptTemplate(
         messages=[MessagesPlaceholder("history", optional=True), ("user", "${input}")]
     )
     # input variables only lists required variables
     assert set(prompt_optional.input_variables) == {"input"}
     prompt_optional.input_schema(input="")  # won't raise error
-    assert _normalize_schema(prompt_optional.get_input_jsonschema()) == snapshot(
-        name="partial"
-    )
+
+    if (PYDANTIC_MAJOR_VERSION, PYDANTIC_MINOR_VERSION) >= (2, 10):
+        assert _normalize_schema(prompt_optional.get_input_jsonschema()) == snapshot(
+            name="partial"
+        )
 
 
 def test_chat_prompt_w_msgs_placeholder_ser_des(snapshot: SnapshotAssertion) -> None: