github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-20 13:54:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

infra: xfail pydantic v2 arg to py function (#25686)

Browse Source 
Issue to track: #25687
This commit is contained in:
Bagatur 2024-08-22 16:52:57 -07:00 committed by GitHub
parent ee98da4f4e
commit b35ee09b3f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 38 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
libs/core/tests/unit_tests/utils/test_function_calling.py
View File

@ -296,6 +296,21 @@ def test_convert_to_openai_function(
assert actual == runnable_expected
@pytest.mark.xfail(reason="Direct pydantic v2 models not yet supported")
def test_convert_to_openai_function_nested_v2() -> None:
class NestedV2(BaseModelV2Maybe):
nested_v2_arg1: int = FieldV2Maybe(..., description="foo")
nested_v2_arg2: Literal["bar", "baz"] = FieldV2Maybe(
..., description="one of 'bar', 'baz'"
)
def my_function(arg1: NestedV2) -> None:
"""dummy function"""
pass
convert_to_openai_function(my_function)
def test_convert_to_openai_function_nested() -> None:
class Nested(BaseModel):
nested_arg1: int = Field(..., description="foo")
@ -303,13 +318,7 @@ def test_convert_to_openai_function_nested() -> None:
..., description="one of 'bar', 'baz'"
)
class NestedV2(BaseModelV2Maybe):
nested_v2_arg1: int = FieldV2Maybe(..., description="foo")
nested_v2_arg2: Literal["bar", "baz"] = FieldV2Maybe(
..., description="one of 'bar', 'baz'"
)
def my_function(arg1: Nested, arg2: NestedV2) -> None:
def my_function(arg1: Nested) -> None:
"""dummy function"""
pass
@ -331,20 +340,10 @@ def test_convert_to_openai_function_nested() -> None:
},
"required": ["nested_arg1", "nested_arg2"],
},
"arg2": {
"type": "object",
"properties": {
"nested_v2_arg1": {"type": "integer", "description": "foo"},
"nested_v2_arg2": {
"type": "string",
"enum": ["bar", "baz"],
"description": "one of 'bar', 'baz'",
},
},
"required": ["nested_v2_arg1", "nested_v2_arg2"],
},
},
"required": ["arg1", "arg2"],
"required": [
"arg1",
],
},
}