fix: update Pydantic schema access (crewAIInc#337)

emmanuel-ferdman · web-flow · commit 94c173da3f21 · 2025-07-02T09:25:26.000-07:00
Signed-off-by: Emmanuel Ferdman &lt;emmanuelferdman@gmail.com&gt;
diff --git a/tests/base_tool_test.py b/tests/base_tool_test.py
@@ -16,7 +16,7 @@ def my_tool(question: str) -> str:
         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."
     )
-    assert my_tool.args_schema.schema()["properties"] == {
+    assert my_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -30,7 +30,7 @@ def my_tool(question: str) -> str:
         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."
     )
-    assert converted_tool.args_schema.schema()["properties"] == {
+    assert converted_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -54,7 +54,7 @@ def _run(self, question: str) -> str:
         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."
     )
-    assert my_tool.args_schema.schema()["properties"] == {
+    assert my_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -68,7 +68,7 @@ def _run(self, question: str) -> str:
         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."
     )
-    assert converted_tool.args_schema.schema()["properties"] == {
+    assert converted_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ def my_tool(question: str) -> str:`
`16`	`16`	`my_tool.description`
`17`	`17`	`== "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."`
`18`	`18`	`)`
`19`		`- assert my_tool.args_schema.schema()["properties"] == {`
	`19`	`+ assert my_tool.args_schema.model_json_schema()["properties"] == {`
`20`	`20`	`"question": {"title": "Question", "type": "string"}`
`21`	`21`	`}`
`22`	`22`	`assert (`
`@@ -30,7 +30,7 @@ def my_tool(question: str) -> str:`
`30`	`30`	`converted_tool.description`
`31`	`31`	`== "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."`
`32`	`32`	`)`
`33`		`- assert converted_tool.args_schema.schema()["properties"] == {`
	`33`	`+ assert converted_tool.args_schema.model_json_schema()["properties"] == {`
`34`	`34`	`"question": {"title": "Question", "type": "string"}`
`35`	`35`	`}`
`36`	`36`	`assert (`
`@@ -54,7 +54,7 @@ def _run(self, question: str) -> str:`
`54`	`54`	`my_tool.description`
`55`	`55`	`== "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."`
`56`	`56`	`)`
`57`		`- assert my_tool.args_schema.schema()["properties"] == {`
	`57`	`+ assert my_tool.args_schema.model_json_schema()["properties"] == {`
`58`	`58`	`"question": {"title": "Question", "type": "string"}`
`59`	`59`	`}`
`60`	`60`	`assert (`
`@@ -68,7 +68,7 @@ def _run(self, question: str) -> str:`
`68`	`68`	`converted_tool.description`
`69`	`69`	`== "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."`
`70`	`70`	`)`
`71`		`- assert converted_tool.args_schema.schema()["properties"] == {`
	`71`	`+ assert converted_tool.args_schema.model_json_schema()["properties"] == {`
`72`	`72`	`"question": {"title": "Question", "type": "string"}`
`73`	`73`	`}`
`74`	`74`	`assert (`