diff --git a/nerve/tools/mcp/body.j2 b/nerve/tools/mcp/body.j2 index d04e456..451445a 100644 --- a/nerve/tools/mcp/body.j2 +++ b/nerve/tools/mcp/body.j2 @@ -2,7 +2,7 @@ from typing import * from pydantic import Field from loguru import logger -async def {{ tool.name }}({% for arg in arguments %}{{ arg.name }}: Annotated[{{ arg.type }}, "{{ arg.description }}"]{% if 'default' in arg %} = {{ arg.default }}{% endif %}{% if not loop.last %}, {% endif %}{% endfor %}) -> Any: +async def {{ tool.name }}({% for arg in arguments %}{{ arg.name }}: Annotated[{{ arg.type }}, "{{ arg.description }}"]{% if 'default' in arg %} = {% if arg.type == "str" %}"{{ arg.default }}"{% else %}{{ arg.default }}{% endif %}{% endif %}{% if not loop.last %}, {% endif %}{% endfor %}) -> Any: """{{ tool.description }}""" - return await globals().get("{{ client_name }}", None).call_tool("{{ tool.name }}", **locals()) \ No newline at end of file + return await globals().get("{{ client_name }}", None).call_tool("{{ tool.name }}", **locals())