Update schema formater

llmio/agent.py

@@ -81,7 +81,12 @@ def function_definition(self) -> T.FunctionDefinition:
         """
         Returns the tool schema that is sent to the OpenAI API.
         """
-        schema = self.params.schema()
+        schema = self.params.model_json_schema()
+
+        schema.pop("title", None)
+        for prop in schema.get("properties", {}).values():
+            prop.pop("title", None)
+
         if self.strict:
             schema["additionalProperties"] = False
         return T.FunctionDefinition(
@@ -537,7 +542,7 @@ async def speak(
 
     @property
     def response_format(self) -> dict[str, Any]:
-        schema = self._response_format.schema()
+        schema = self._response_format.model_json_schema()
         schema["additionalProperties"] = False
         return {
             "type": "json_schema",

llmio/function_parser.py

@@ -30,11 +30,5 @@ def model_from_function(function: Callable) -> type[BaseModel]:
                 "Unable to parse function signature. Only named arguments supported."
             )
 
-    class Config:
-        @staticmethod
-        def json_schema_extra(schema: Dict[str, Any]) -> None:
-            schema.pop("title", None)
-            for prop in schema.get("properties", {}).values():
-                prop.pop("title", None)
-
-    return create_model(to_camel(function.__name__), __config__=Config, **fields)  # type: ignore
+    model = create_model(to_camel(function.__name__), **fields)  # type: ignore
+    return model