optimizeSql

Instructions

Implementation Reference

• src/alibabacloud_dms_mcp_server/server.py:559-580 (handler)

  Core handler function 'optimize_sql' that implements the SQL optimization logic. It creates a DMS client, builds an OptimizeSqlByMetaAgentRequest, calls the optimize_sql_by_meta_agent API, and returns the response as a dict.

  async def optimize_sql(
          database_id: str = Field(description="DMS databaseId"),
          question: Optional[str] = Field(default=None, description="Natural language question"),
          sql: str = Field(description="SQL statement"),
          model: Optional[str] = Field(default=None,
                                       description="Optional: if a specific model is desired, it can be specified here")
  ) -> Any:
      client = create_client()
      req = dms_enterprise_20181101_models.OptimizeSqlByMetaAgentRequest(db_id=database_id, query=question, sql=sql)
      # if mcp.state.real_login_uid:
      #     req.real_login_user_uid = mcp.state.real_login_uid
      if model:
          req.model = model
      try:
          resp = client.optimize_sql_by_meta_agent(req)
          if not resp or not resp.body:
              return None
          data = resp.body.to_map()
          return data
      except Exception as e:
          logger.error(f"Error in optimize_sql: {e}")
          raise

• src/alibabacloud_dms_mcp_server/server.py:559-565 (schema)

  Input schema for optimize_sql: database_id (str), question (Optional[str]), sql (str), model (Optional[str]). Return type is Any (dict from API response).

  async def optimize_sql(
          database_id: str = Field(description="DMS databaseId"),
          question: Optional[str] = Field(default=None, description="Natural language question"),
          sql: str = Field(description="SQL statement"),
          model: Optional[str] = Field(default=None,
                                       description="Optional: if a specific model is desired, it can be specified here")
  ) -> Any:

• src/alibabacloud_dms_mcp_server/server.py:709-720 (registration)

  Registration of 'optimizeSql' tool in the configured database toolset (_register_configured_db_toolset). Wraps the core handler with the pre-configured database_id.

  @self.mcp.tool(name="optimizeSql",
                 description="Analyze and optimize SQL performance based on the provided SQL statement",
                 annotations={"title": "SQL优化", "readOnlyHint": True, "destructiveHint": False})
  async def optimize_sql_configured(
          question: Optional[str] = Field(default=None, description="Natural language question"),
          sql: str = Field(description="SQL statement"),
          model: Optional[str] = Field(default=None,
                                       description="Optional: if a specific model is desired, it can be specified here")
  ) -> Any:
      result_obj = await optimize_sql(database_id=self.default_database_id, question=question, sql=sql,
                                      model=model)
      return result_obj

• src/alibabacloud_dms_mcp_server/server.py:800-803 (registration)

  Registration of 'optimizeSql' tool in the full toolset (_register_full_toolset). Directly registers the core optimize_sql function with description requiring a database_id.

  self.mcp.tool(name="optimizeSql", description=f"Analyze and optimize SQL performance "
                                                f"based on the provided SQL statement and database ID"
                                                f"{DATABASE_ID_DESCRIPTION}",
                annotations={"title": "SQL优化", "readOnlyHint": True})(optimize_sql)