Skip to content

Commit c55294e

Browse files
authored
chore(core): add test for nested pydantic fields in schemas (#32285)
1 parent 7a26c3d commit c55294e

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

libs/core/tests/unit_tests/test_tools.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2586,6 +2586,18 @@ def extract_data(extracted_data: dict[str, Any]) -> dict[str, Any]:
25862586
}
25872587

25882588

2589+
def test_nested_pydantic_fields() -> None:
2590+
class Address(BaseModel):
2591+
street: str
2592+
2593+
class Person(BaseModel):
2594+
name: str
2595+
address: Address = Field(description="Home address")
2596+
2597+
result = convert_to_openai_tool(Person)
2598+
assert len(result["function"]["parameters"]["properties"]) == 2
2599+
2600+
25892601
async def test_tool_ainvoke_does_not_mutate_inputs() -> None:
25902602
"""Verify that the inputs are not mutated when invoking a tool asynchronously."""
25912603

0 commit comments

Comments
 (0)