Merge pull request #75 from webflow/sai

Moving beta tools to GA

Author: viratatwebflow

package-lock.json

@@ -14,11 +14,11 @@
     "build:watch": "tsup src/index.ts --watch"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "1.24.0",
+    "@modelcontextprotocol/sdk": "1.25.2",
     "cors": "^2.8.5",
     "express": "^5.1.0",
     "socket.io": "^4.8.1",
-    "webflow-api": "3.1.1",
+    "webflow-api": "3.2.1",
     "zod": "^3.24.2"
   },
   "devDependencies": {

package.json

@@ -15,6 +15,8 @@ import {
   registerDEVariableTools,
   registerRulesTools,
   registerLocalDeMCPConnectionTools,
+  registerCommentsTools,
+  registerEnterpriseTools,
 } from "./tools";
 import { RPCType } from "./types/RPCType";
 
@@ -51,12 +53,11 @@ export function registerTools(
   registerPagesTools(server, getClient);
   registerScriptsTools(server, getClient);
   registerSiteTools(server, getClient);
+  registerCommentsTools(server, getClient);
+  registerEnterpriseTools(server, getClient);
 }
 
-export function registerDesignerTools(
-  server: McpServer,
-  rpc: RPCType
-) {
+export function registerDesignerTools(server: McpServer, rpc: RPCType) {
   registerDEAssetTools(server, rpc);
   registerDEComponentsTools(server, rpc);
   registerDEElementTools(server, rpc);
@@ -72,9 +73,6 @@ export function registerMiscTools(server: McpServer) {
 /**
  * IMPORTANT: registerLocalTools is only valid for OSS MCP Version
  */
-export function registerLocalTools(
-  server: McpServer,
-  rpc: RPCType
-) {
+export function registerLocalTools(server: McpServer, rpc: RPCType) {
   registerLocalDeMCPConnectionTools(server, rpc);
 }

src/mcp.ts

@@ -24,7 +24,7 @@ export const StaticFieldSchema = z.object({
       z.literal("PlainText"),
       z.literal("RichText"),
       z.literal("Switch"),
-      z.literal("Video"),
+      z.literal("VideoLink"),
     ])
     .describe("Type of the field. Choose of these appropriate field types."),
   displayName: z.string().describe("Name of the field."),

src/schemas/StaticFieldSchema.ts

@@ -9,9 +9,15 @@ export function registerAiChatTools(server: McpServer) {
   server.registerTool(
     "ask_webflow_ai",
     {
-      title: "Ask Webflow AI",
       description: "Ask Webflow AI about anything related to Webflow API.",
-      inputSchema: z.object({ message: z.string() }),
+      title: "Ask Webflow AI",
+      annotations: {
+        openWorldHint: true,
+        readOnlyHint: true,
+      },
+      inputSchema: {
+        message: z.string().describe("The message to ask Webflow AI about."),
+      },
     },
     async ({ message }) => {
       const result = await postChat(message);