update tool desc

Author: leleboys

src/asktable_mcp_server/schemas.py

@@ -2,7 +2,8 @@
 from pydantic import Field
 
 # 共享的参数类型定义
-QuestionParam = Annotated[str, Field(description="用户的自然语言查询描述。示例：生成查询昨天订单总金额的SQL、写一个SQL查询销售额前10的产品、帮我写一个统计各部门员工数量的SQL")]
+QuestionParamGenSQL = Annotated[str, Field(description="用户的自然语言查询描述。示例：生成查询昨天订单总金额的SQL、写一个SQL查询销售额前10的产品、帮我写一个统计各部门员工数量的SQL")]
+QuestionParamQuery = Annotated[str, Field(description="用户的自然语言查询描述。示例：查询昨天订单总金额、查询销售额前10的产品、统计各部门员工数量")]
 
 RoleIdParam = Annotated[Optional[str], Field(description="角色ID，精确控制用户对数据的访问权限，支持库/表/字段/行四级控制。示例：'role_123456'，可以为空（使用默认权限）。详见：https://docs.asktable.com/docs/role-and-permission-management/introduction")]
 

src/asktable_mcp_server/server.py

@@ -8,7 +8,8 @@
 )
 from asktable_mcp_server.sse_server import main as sse_main
 from asktable_mcp_server.schemas import (
-    QuestionParam,
+    QuestionParamQuery,
+    QuestionParamGenSQL,
     RoleIdParam,
     RoleVariablesParam,
     GEN_SQL_DESCRIPTION,
@@ -20,7 +21,7 @@
 
 @mcp.tool(name='使用 AskTable 查询数据')
 async def query(
-    question: QuestionParam,
+    question: QuestionParamQuery,
     role_id: RoleIdParam = None,
     role_variables: RoleVariablesParam = None
 ) -> dict:
@@ -44,7 +45,7 @@ async def query(
 
 @mcp.tool(name='使用 AskTable 生成 SQL')
 async def gen_sql(
-    question: QuestionParam,
+    question: QuestionParamGenSQL,
     role_id: RoleIdParam = None,
     role_variables: RoleVariablesParam = None
 ) -> dict:

src/asktable_mcp_server/sse_server.py

@@ -15,7 +15,8 @@
 )
 from asktable_mcp_server.version import __version__
 from asktable_mcp_server.schemas import (
-    QuestionParam,
+    QuestionParamQuery,
+    QuestionParamGenSQL,
     RoleIdParam,
     RoleVariablesParam,
     GEN_SQL_DESCRIPTION,
@@ -85,7 +86,7 @@ async def home(request: Request):
 
     @mcp.tool(name="使用 AskTable 查询数据")
     async def query(
-        question: QuestionParam,
+        question: QuestionParamQuery,
         role_id: RoleIdParam = None,
         role_variables: RoleVariablesParam = None
     ) -> dict:
@@ -115,7 +116,7 @@ async def query(
 
     @mcp.tool(name="使用 AskTable 生成 SQL")
     async def gen_sql(
-        question: QuestionParam,
+        question: QuestionParamGenSQL,
         role_id: RoleIdParam = None,
         role_variables: RoleVariablesParam = None
     ) -> dict:

src/asktable_mcp_server/version.py

@@ -1 +1 @@
-__version__ = "0.2.20" 
+__version__ = "0.2.21"