diff --git a/tools/listmodels.py b/tools/listmodels.py index 2afe61b..621d4cd 100644 --- a/tools/listmodels.py +++ b/tools/listmodels.py @@ -45,13 +45,8 @@ class ListModelsTool(BaseTool): """Return the JSON schema for the tool's input""" return { "type": "object", - "properties": { - "model": { - "type": "string", - "description": "Model to use (ignored by listmodels tool)" - } - }, - "required": [] + "properties": {"model": {"type": "string", "description": "Model to use (ignored by listmodels tool)"}}, + "required": [], } def get_system_prompt(self) -> str: diff --git a/tools/version.py b/tools/version.py index e52158a..10886a9 100644 --- a/tools/version.py +++ b/tools/version.py @@ -149,13 +149,8 @@ class VersionTool(BaseTool): """Return the JSON schema for the tool's input""" return { "type": "object", - "properties": { - "model": { - "type": "string", - "description": "Model to use (ignored by version tool)" - } - }, - "required": [] + "properties": {"model": {"type": "string", "description": "Model to use (ignored by version tool)"}}, + "required": [], } def get_system_prompt(self) -> str: