viewgate-mcp 1.0.45 → 1.0.46

package/dist/index.js

@@ -75,8 +75,8 @@ function createMcpServer(apiKey, personalKey) {
             case "planning":
                 return "planning";
             case "sync_endpoints":
-            case "get_synced_endpoints":
                 return "sync";
+            case "get_synced_endpoints":
             case "get_ai_resolved_tickets":
                 return "idle";
             default:
@@ -89,15 +89,14 @@ function createMcpServer(apiKey, personalKey) {
             resetGuard();
         }
         const desiredFlow = getToolFlow(toolName);
-        if (toolName === "get_ai_resolved_tickets") {
-            if (guard.flow !== "idle") {
-                throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
-            }
+        // Neutral tools (idle flow) are always allowed and don't interrupt active flows
+        if (desiredFlow === "idle") {
             guard.lastTool = toolName;
             guard.lastActivityAt = now;
             return;
         }
-        if (guard.flow === "idle") {
+        if (guard.flow === "idle" || desiredFlow !== guard.flow) {
+            // Implicit flow reset if we are starting a new flow with a "start tool"
             switch (toolName) {
                 case "get_ui_components":
                     guard.flow = "ui_components";
@@ -123,16 +122,18 @@ function createMcpServer(apiKey, personalKey) {
                     guard.step = 1;
                     break;
                 default:
-                    throw new Error("TOOL_CALL_BLOCKED: tool not allowed in idle");
+                    if (guard.flow === "idle") {
+                        throw new Error("TOOL_CALL_BLOCKED: tool not allowed in idle");
+                    }
+                    else {
+                        throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
+                    }
             }
             guard.startedAt = now;
             guard.lastTool = toolName;
             guard.lastActivityAt = now;
             return;
         }
-        if (desiredFlow !== guard.flow) {
-            throw new Error("TOOL_CALL_BLOCKED: tool not allowed in active flow");
-        }
         if (guard.flow === "ui_components") {
             if (toolName === "get_ui_components") {
                 if (guard.step > 2)
@@ -155,18 +156,17 @@ function createMcpServer(apiKey, personalKey) {
         }
         else if (guard.flow === "annotations") {
             if (toolName === "get_annotations") {
-                if (guard.step > 2)
-                    throw new Error("TOOL_CALL_BLOCKED: cannot restart flow at this step");
+                // Allow re-fetching at any time during the flow
                 guard.step = 1;
             }
             else if (toolName === "mark_annotation_ready") {
                 if (guard.step !== 1)
                     throw new Error("TOOL_CALL_BLOCKED: unexpected step");
-                guard.step = 2;
+                // Stop here and reset flow as per USER_REQUEST (MCP only reaches 'applied')
+                resetGuard();
             }
             else if (toolName === "mark_annotations_as_live") {
-                if (guard.step !== 2)
-                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                // Optional step if the model decides to use it, but no longer part of the required chain
                 resetGuard();
             }
             else {
@@ -175,18 +175,14 @@ function createMcpServer(apiKey, personalKey) {
         }
         else if (guard.flow === "ui_improvements") {
             if (toolName === "get_ui_improvements") {
-                if (guard.step > 2)
-                    throw new Error("TOOL_CALL_BLOCKED: cannot restart flow at this step");
                 guard.step = 1;
             }
             else if (toolName === "mark_annotation_ready") {
                 if (guard.step !== 1)
                     throw new Error("TOOL_CALL_BLOCKED: unexpected step");
-                guard.step = 2;
+                resetGuard();
             }
             else if (toolName === "mark_annotations_as_live") {
-                if (guard.step !== 2)
-                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
                 resetGuard();
             }
             else {
@@ -216,8 +212,7 @@ function createMcpServer(apiKey, personalKey) {
                 guard.step = 2;
             }
             else if (toolName === "get_synced_endpoints") {
-                if (guard.step !== 2)
-                    throw new Error("TOOL_CALL_BLOCKED: unexpected step");
+                // Actually get_synced_endpoints is now neutral, but if called as part of flow, reset.
                 resetGuard();
             }
             else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viewgate-mcp",
-  "version": "1.0.45",
+  "version": "1.0.46",
   "main": "dist/index.js",
   "bin": {
     "viewgate-mcp": "./dist/index.js"