chore: schema fix

phil65 · Nov 23, 2024 · e1a233b · e1a233b
1 parent 4ca48bd
commit e1a233b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/llmling/tools/base.py b/src/llmling/tools/base.py
@@ -21,12 +21,13 @@ class BaseTool(ABC):
     # Class-level schema definition
     name: ClassVar[str]
     description: ClassVar[str]
-    parameters_schema: ClassVar[dict[str, Any]]
 
     @classmethod
     def get_schema(cls) -> py2openai.OpenAIFunctionTool:
         """Get the tool's schema for LLM function calling."""
-        return py2openai.create_schema(cls.execute).model_dump_openai()
+        schema = py2openai.create_schema(cls.execute).model_dump_openai()
+        schema["function"]["name"] = cls.name
+        return schema
 
     @abstractmethod
     async def execute(self, **params: Any) -> Any | Awaitable[Any]: