Update Pydantic Example

shashi-stackone · shashi-stackone · commit 933cd1020096 · 2026-04-16T23:47:00.000+01:00
diff --git a/examples/pydantic_ai_integration.py b/examples/pydantic_ai_integration.py
@@ -27,15 +27,17 @@
 
 
 def _to_pydantic_ai_tool(stackone_tool) -> Tool:
-    """Convert a StackOneTool to a Pydantic AI Tool using the public API."""
+    """Convert a StackOneTool to a Pydantic AI Tool with the proper JSON schema."""
+    params_schema = stackone_tool.to_openai_function()["function"]["parameters"]
 
     def execute(**kwargs: object) -> str:
         return json.dumps(stackone_tool.execute(kwargs))
 
-    return Tool(
+    return Tool.from_schema(
         execute,
         name=stackone_tool.name,
         description=stackone_tool.description,
+        json_schema=params_schema,
     )
 
 
