@stigmer/sdk 3.12.4 → 3.12.6

package/__tests__/mcpserver-connect.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=mcpserver-connect.test.d.ts.map

package/__tests__/mcpserver-connect.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpserver-connect.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/mcpserver-connect.test.ts"],"names":[],"mappings":""}

package/__tests__/mcpserver-connect.test.js

@@ -0,0 +1,146 @@
+// Pins the shared async-connect protocol (stigmer/stigmer#425): start + poll,
+// terminal-failure rehydration, the UNIMPLEMENTED blocking fallback, and the
+// still-running bound. Every SDK surface (CLI, React hooks, host apps) rides
+// this one implementation, so its contract is pinned here rather than
+// re-tested per consumer.
+import { describe, expect, it } from "vitest";
+import { create } from "@bufbuild/protobuf";
+import { Code, ConnectError } from "@connectrpc/connect";
+import { McpServerSchema } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/api_pb";
+import { ConnectInputSchema } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/io_pb";
+import { ConnectPhase } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/status_pb";
+import { StigmerError, wrapError } from "../gen/errors.js";
+import { connectAndWait, ConnectStillRunningError, } from "../mcpserver-connect.js";
+const INPUT = create(ConnectInputSchema, { mcpServerId: "mcp-1", org: "test-org" });
+function server(overrides) {
+    return create(McpServerSchema, {
+        metadata: { id: "mcp-1", name: "Test Server" },
+        status: {
+            ...(overrides?.toolName !== undefined
+                ? { discoveredCapabilities: { tools: [{ name: overrides.toolName }] } }
+                : {}),
+            connectStatus: {
+                phase: overrides?.phase ?? ConnectPhase.connecting,
+                workflowId: "stigmer/mcp-server/connect/mcp-1",
+                failureCode: overrides?.failureCode ?? "",
+                failureMessage: overrides?.failureMessage ?? "",
+                warning: overrides?.warning ?? "",
+            },
+        },
+    });
+}
+function unimplemented() {
+    return wrapError(new ConnectError("startConnect is not implemented", Code.Unimplemented));
+}
+const NEVER = () => new Promise(() => { });
+describe("connectAndWait", () => {
+    it("resolves with the settled resource once connect_status reports success", async () => {
+        const polled = [
+            server({ phase: ConnectPhase.connecting }),
+            server({ phase: ConnectPhase.succeeded, toolName: "search_code" }),
+        ];
+        const lane = {
+            startConnect: async () => server(),
+            connect: NEVER,
+            get: async () => polled.shift() ?? server({ phase: ConnectPhase.succeeded }),
+        };
+        const settled = await connectAndWait(lane, INPUT, { pollIntervalMs: 1 });
+        expect(settled.status?.connectStatus?.phase).toBe(ConnectPhase.succeeded);
+        expect(settled.status?.discoveredCapabilities?.tools[0]?.name).toBe("search_code");
+    });
+    it("surfaces the CONNECTING snapshot (dead-runner warning) via onStarted", async () => {
+        const lane = {
+            startConnect: async () => server({ warning: "no runner appears to be polling the task queue" }),
+            connect: NEVER,
+            get: async () => server({ phase: ConnectPhase.succeeded }),
+        };
+        let warning = "";
+        await connectAndWait(lane, INPUT, {
+            pollIntervalMs: 1,
+            onStarted: (started) => {
+                warning = started.status?.connectStatus?.warning ?? "";
+            },
+        });
+        expect(warning).toContain("no runner");
+    });
+    it("rehydrates a failed operation into the classification the blocking RPC would have thrown", async () => {
+        const lane = {
+            startConnect: async () => server(),
+            connect: NEVER,
+            get: async () => server({
+                phase: ConnectPhase.failed,
+                failureCode: "FailedPrecondition",
+                failureMessage: "connect failed for MCP server 'Test Server': missing credentials",
+            }),
+        };
+        const err = await connectAndWait(lane, INPUT, { pollIntervalMs: 1 }).catch((e) => e);
+        expect(err).toBeInstanceOf(StigmerError);
+        const stigmerErr = err;
+        // Byte-parity with the blocking lane: same code mapping, same raw message.
+        expect(stigmerErr.code).toBe("failed-precondition");
+        expect(stigmerErr.message).toBe("connect failed for MCP server 'Test Server': missing credentials");
+    });
+    it("maps an unknown failure_code to the unknown classification instead of crashing", async () => {
+        const lane = {
+            startConnect: async () => server(),
+            connect: NEVER,
+            get: async () => server({
+                phase: ConnectPhase.failed,
+                failureCode: "SomeFutureCode",
+                failureMessage: "boom",
+            }),
+        };
+        const err = await connectAndWait(lane, INPUT, { pollIntervalMs: 1 }).catch((e) => e);
+        expect(err.code).toBe("unknown");
+        expect(err.message).toBe("boom");
+    });
+    it("falls back to the blocking connect when the backend answers UNIMPLEMENTED", async () => {
+        let blockingCalled = false;
+        const lane = {
+            startConnect: async () => {
+                throw unimplemented();
+            },
+            connect: async () => {
+                blockingCalled = true;
+                return server({ phase: ConnectPhase.succeeded, toolName: "search_code" });
+            },
+            get: NEVER,
+        };
+        const settled = await connectAndWait(lane, INPUT);
+        expect(blockingCalled).toBe(true);
+        expect(settled.status?.discoveredCapabilities?.tools[0]?.name).toBe("search_code");
+    });
+    it("propagates non-UNIMPLEMENTED start failures untouched", async () => {
+        const startFailure = wrapError(new ConnectError("no such server", Code.NotFound));
+        const lane = {
+            startConnect: async () => {
+                throw startFailure;
+            },
+            connect: NEVER,
+            get: NEVER,
+        };
+        await expect(connectAndWait(lane, INPUT)).rejects.toBe(startFailure);
+    });
+    it("throws ConnectStillRunningError when the deadline elapses while CONNECTING", async () => {
+        const lane = {
+            startConnect: async () => server(),
+            connect: NEVER,
+            get: async () => server({ phase: ConnectPhase.connecting }),
+        };
+        const err = await connectAndWait(lane, INPUT, { pollIntervalMs: 1, deadlineMs: 20 }).catch((e) => e);
+        expect(err).toBeInstanceOf(ConnectStillRunningError);
+        expect(err.mcpServerId).toBe("mcp-1");
+    });
+    it("bounds the blocking fallback with the same still-running semantics (soft, non-cancelling)", async () => {
+        const lane = {
+            startConnect: async () => {
+                throw unimplemented();
+            },
+            connect: NEVER,
+            get: NEVER,
+        };
+        const err = await connectAndWait(lane, INPUT, { deadlineMs: 20 }).catch((e) => e);
+        expect(err).toBeInstanceOf(ConnectStillRunningError);
+    });
+});
+//# sourceMappingURL=mcpserver-connect.test.js.map

package/__tests__/mcpserver-connect.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpserver-connect.test.js","sourceRoot":"","sources":["../../src/__tests__/mcpserver-connect.test.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAC9E,6EAA6E;AAC7E,6EAA6E;AAC7E,sEAAsE;AACtE,0BAA0B;AAE1B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,wDAAwD,CAAC;AAEzF,OAAO,EAAE,kBAAkB,EAAE,MAAM,uDAAuD,CAAC;AAC3F,OAAO,EAAE,YAAY,EAAE,MAAM,2DAA2D,CAAC;AACzF,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EACL,cAAc,EACd,wBAAwB,GAEzB,MAAM,yBAAyB,CAAC;AAEjC,MAAM,KAAK,GAAG,MAAM,CAAC,kBAAkB,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,CAAC;AAEpF,SAAS,MAAM,CAAC,SAMf;IACC,OAAO,MAAM,CAAC,eAAe,EAAE;QAC7B,QAAQ,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE;QAC9C,MAAM,EAAE;YACN,GAAG,CAAC,SAAS,EAAE,QAAQ,KAAK,SAAS;gBACnC,CAAC,CAAC,EAAE,sBAAsB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE;gBACvE,CAAC,CAAC,EAAE,CAAC;YACP,aAAa,EAAE;gBACb,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,YAAY,CAAC,UAAU;gBAClD,UAAU,EAAE,kCAAkC;gBAC9C,WAAW,EAAE,SAAS,EAAE,WAAW,IAAI,EAAE;gBACzC,cAAc,EAAE,SAAS,EAAE,cAAc,IAAI,EAAE;gBAC/C,OAAO,EAAE,SAAS,EAAE,OAAO,IAAI,EAAE;aAClC;SACF;KACF,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa;IACpB,OAAO,SAAS,CAAC,IAAI,YAAY,CAAC,iCAAiC,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;AAC5F,CAAC;AAED,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,IAAI,OAAO,CAAY,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AAErD,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,MAAM,GAAG;YACb,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,CAAC;YAC1C,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;SACnE,CAAC;QACF,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,EAAE;YAClC,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,SAAS,EAAE,CAAC;SAC7E,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC;QACzE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1E,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,sBAAsB,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACrF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,gDAAgD,EAAE,CAAC;YAC/F,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,SAAS,EAAE,CAAC;SAC3D,CAAC;QAEF,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE;YAChC,cAAc,EAAE,CAAC;YACjB,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;gBACrB,OAAO,GAAG,OAAO,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,IAAI,EAAE,CAAC;YACzD,CAAC;SACF,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;QACxG,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,EAAE;YAClC,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK,IAAI,EAAE,CACd,MAAM,CAAC;gBACL,KAAK,EAAE,YAAY,CAAC,MAAM;gBAC1B,WAAW,EAAE,oBAAoB;gBACjC,cAAc,EAAE,kEAAkE;aACnF,CAAC;SACL,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9F,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,GAAmB,CAAC;QACvC,2EAA2E;QAC3E,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACpD,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;IACtG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,EAAE;YAClC,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK,IAAI,EAAE,CACd,MAAM,CAAC;gBACL,KAAK,EAAE,YAAY,CAAC,MAAM;gBAC1B,WAAW,EAAE,gBAAgB;gBAC7B,cAAc,EAAE,MAAM;aACvB,CAAC;SACL,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9F,MAAM,CAAE,GAAoB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,CAAE,GAAoB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE;gBACvB,MAAM,aAAa,EAAE,CAAC;YACxB,CAAC;YACD,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,cAAc,GAAG,IAAI,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC;YAC5E,CAAC;YACD,GAAG,EAAE,KAAK;SACX,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAClD,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,sBAAsB,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACrF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,YAAY,CAAC,gBAAgB,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAClF,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE;gBACvB,MAAM,YAAY,CAAC;YACrB,CAAC;YACD,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK;SACX,CAAC;QAEF,MAAM,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,EAAE;YAClC,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,CAAC;SAC5D,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,KAAK,CACxF,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAClB,CAAC;QACF,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAAC;QACrD,MAAM,CAAE,GAAgC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QACzG,MAAM,IAAI,GAAyB;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE;gBACvB,MAAM,aAAa,EAAE,CAAC;YACxB,CAAC;YACD,OAAO,EAAE,KAAK;YACd,GAAG,EAAE,KAAK;SACX,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3F,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/__tests__/update-input-roundtrip.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=update-input-roundtrip.test.d.ts.map

package/__tests__/update-input-roundtrip.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-input-roundtrip.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/update-input-roundtrip.test.ts"],"names":[],"mappings":""}