Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion libs/core/langchain_core/utils/function_calling.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ def _convert_pydantic_to_openai_function(
The function description.
"""
if hasattr(model, "model_json_schema"):
schema = model.model_json_schema() # Pydantic 2
schema = model.model_json_schema(mode="serialization") # Pydantic 2
elif hasattr(model, "schema"):
schema = model.schema() # Pydantic 1
else:
Expand Down
48 changes: 48 additions & 0 deletions libs/core/tests/unit_tests/utils/test_function_calling.py
Original file line number Diff line number Diff line change
Expand Up @@ -1168,3 +1168,51 @@ class MyModel(BaseModel):
func = convert_to_openai_function(MyModel, strict=True)
actual = func["parameters"]["required"]
assert actual == expected


def test_pydantic_fieldinfo_serialization() -> None:
"""Test that FieldInfo objects are properly serialized in Pydantic v2 schemas.

This test ensures that using mode='serialization' prevents FieldInfo objects
from appearing in the generated JSON schema, which would cause serialization
errors in integrations.
"""

class MyModel(BaseModel):
"""Test model with various field configurations."""

required_field: str = Field(..., description="A required field")
optional_field: str | None = Field(None, description="An optional field")
default_field: int = Field(default=42, description="A field with default")

result = convert_to_openai_function(MyModel)

expected = {
"name": "MyModel",
"description": "Test model with various field configurations.",
"parameters": {
"type": "object",
"properties": {
"required_field": {
"type": "string",
"description": "A required field",
},
"optional_field": {
"anyOf": [
{"type": "string"},
{"type": "null"},
],
"default": None,
"description": "An optional field",
},
"default_field": {
"type": "integer",
"default": 42,
"description": "A field with default",
},
},
"required": ["required_field"],
},
}

assert result == expected
Loading