andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 15:22:37 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: update Pydantic schema access (#337)

Browse Source 
Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
Emmanuel Ferdman
2025-07-02 19:25:26 +03:00
committed by GitHub
parent 26652e5e24
commit d53e96fcd7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/base_tool_test.py
View File

@@ -16,7 +16,7 @@ def test_creating_a_tool_using_annotation():
my_tool.description my_tool.description
== "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it." == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
) )
assert my_tool.args_schema.schema()["properties"] == { assert my_tool.args_schema.model_json_schema()["properties"] == {
"question": {"title": "Question", "type": "string"} "question": {"title": "Question", "type": "string"}
} }
assert ( assert (
@@ -30,7 +30,7 @@ def test_creating_a_tool_using_annotation():
converted_tool.description converted_tool.description
== "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it." == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
) )
assert converted_tool.args_schema.schema()["properties"] == { assert converted_tool.args_schema.model_json_schema()["properties"] == {
"question": {"title": "Question", "type": "string"} "question": {"title": "Question", "type": "string"}
} }
assert ( assert (
@@ -54,7 +54,7 @@ def test_creating_a_tool_using_baseclass():
my_tool.description my_tool.description
== "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it." == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
) )
assert my_tool.args_schema.schema()["properties"] == { assert my_tool.args_schema.model_json_schema()["properties"] == {
"question": {"title": "Question", "type": "string"} "question": {"title": "Question", "type": "string"}
} }
assert ( assert (
@@ -68,7 +68,7 @@ def test_creating_a_tool_using_baseclass():
converted_tool.description converted_tool.description
== "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it." == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
) )
assert converted_tool.args_schema.schema()["properties"] == { assert converted_tool.args_schema.model_json_schema()["properties"] == {
"question": {"title": "Question", "type": "string"} "question": {"title": "Question", "type": "string"}
} }
assert ( assert (