@copilotkit/runtime 1.55.0-next.8 → 1.55.0

package/src/v2/runtime/__tests__/handle-connect.test.ts

@@ -2,10 +2,9 @@ import { Observable } from "rxjs";
 import { describe, it, expect, vi } from "vitest";
 import { BaseEvent } from "@ag-ui/client";
 import { handleConnectAgent } from "../handlers/handle-connect";
-import { CopilotRuntime } from "../runtime";
+import { CopilotRuntime } from "../core/runtime";
 import { AgentRunnerConnectRequest } from "../runner/agent-runner";
 import { IntelligenceAgentRunner } from "../runner/intelligence";
-import { CopilotKitIntelligence } from "../intelligence-platform/client";
 
 describe("handleConnectAgent", () => {
   const createMockRuntime = (
@@ -207,9 +206,12 @@ describe("handleConnectAgent", () => {
         }),
       });
 
-    const createIntelligenceRuntime = (
-      platform?: Partial<CopilotKitIntelligence>,
-    ) => {
+    /** Loose mock type for CopilotKitIntelligence — avoids `as any` while the class has private fields. */
+    interface MockIntelligencePlatform {
+      [key: string]: ((...args: any[]) => any) | undefined;
+    }
+
+    const createIntelligenceRuntime = (platform?: MockIntelligencePlatform) => {
       const runner = Object.create(IntelligenceAgentRunner.prototype);
       runner.connect = vi.fn(
         () =>
@@ -240,7 +242,7 @@ describe("handleConnectAgent", () => {
           events: [],
         }),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
 
       const response = await handleConnectAgent({
         runtime,
@@ -272,7 +274,7 @@ describe("handleConnectAgent", () => {
           events: [{ type: "MESSAGES_SNAPSHOT", messages: [] }],
         }),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
 
       const response = await handleConnectAgent({
         runtime,
@@ -294,7 +296,7 @@ describe("handleConnectAgent", () => {
         ɵconnectThread: vi.fn().mockResolvedValue(null),
         createThread: vi.fn(),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
 
       const response = await handleConnectAgent({
         runtime,
@@ -318,7 +320,7 @@ describe("handleConnectAgent", () => {
           .fn()
           .mockRejectedValue(new Error("No active connect plan")),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
 
       const response = await handleConnectAgent({
         runtime,
@@ -335,7 +337,7 @@ describe("handleConnectAgent", () => {
       const platform = {
         ɵconnectThread: vi.fn().mockResolvedValue(null),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
 
       const response = await handleConnectAgent({
         runtime,
@@ -356,7 +358,7 @@ describe("handleConnectAgent", () => {
         ɵconnectThread: vi.fn().mockResolvedValue(null),
       };
       const identifyUser = vi.fn().mockResolvedValue({ id: "resolved-user" });
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
       runtime.identifyUser = identifyUser;
       const request = createConnectRequest(
         { "X-User-Id": "legacy-user" },
@@ -383,7 +385,7 @@ describe("handleConnectAgent", () => {
       const platform = {
         ɵconnectThread: vi.fn(),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
       runtime.identifyUser = vi.fn().mockResolvedValue({ id: "" });
 
       const response = await handleConnectAgent({
@@ -400,7 +402,7 @@ describe("handleConnectAgent", () => {
       const platform = {
         ɵconnectThread: vi.fn(),
       };
-      const runtime = createIntelligenceRuntime(platform as any);
+      const runtime = createIntelligenceRuntime(platform);
       runtime.identifyUser = vi
         .fn()
         .mockRejectedValue(new Error("auth failed"));

package/src/v2/runtime/__tests__/handle-run.test.ts

@@ -3,9 +3,8 @@ import { describe, it, expect, vi } from "vitest";
 import { AbstractAgent, HttpAgent } from "@ag-ui/client";
 import { A2UIMiddleware } from "@ag-ui/a2ui-middleware";
 import { handleRunAgent } from "../handlers/handle-run";
-import { CopilotRuntime } from "../runtime";
+import { CopilotRuntime } from "../core/runtime";
 import { IntelligenceAgentRunner } from "../runner/intelligence";
-import { CopilotKitIntelligence } from "../intelligence-platform/client";
 
 describe("handleRunAgent", () => {
   const createMockRuntime = (
@@ -287,9 +286,14 @@ describe("handleRunAgent", () => {
   });
 
   describe("IntelligenceAgentRunner join code path", () => {
+    /** Loose mock type for CopilotKitIntelligence — avoids `as any` while the class has private fields. */
+    interface MockIntelligencePlatform {
+      [key: string]: ((...args: any[]) => any) | undefined;
+    }
+
     const createIntelligenceRuntime = (
       agent: AbstractAgent,
-      platform?: Partial<CopilotKitIntelligence>,
+      platform?: MockIntelligencePlatform,
       options?: {
         generateThreadNames?: boolean;
         identifyUser?: (
@@ -352,7 +356,7 @@ describe("handleRunAgent", () => {
           .fn()
           .mockResolvedValue({ joinToken: "jt-123", joinCode: "jc-123" }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
 
       const response = await handleRunAgent({
         runtime,
@@ -392,7 +396,7 @@ describe("handleRunAgent", () => {
           .mockResolvedValue({ joinToken: "jt-123", joinCode: "jc-123" }),
       };
       const identifyUser = vi.fn().mockResolvedValue({ id: "resolved-user" });
-      const runtime = createIntelligenceRuntime(agent, platform as any, {
+      const runtime = createIntelligenceRuntime(agent, platform, {
         identifyUser,
       });
       const request = createRunRequest({ "X-User-Id": "legacy-user" });
@@ -430,7 +434,7 @@ describe("handleRunAgent", () => {
           .fn()
           .mockResolvedValue({ joinToken: "jt-456", joinCode: "jc-456" }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
 
       await handleRunAgent({
         runtime,
@@ -453,7 +457,7 @@ describe("handleRunAgent", () => {
         getThreadMessages: vi.fn().mockResolvedValue({ messages: [] }),
         ɵacquireThreadLock: vi.fn().mockResolvedValue({ joinCode: "jc-789" }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
 
       const response = await handleRunAgent({
         runtime,
@@ -479,7 +483,7 @@ describe("handleRunAgent", () => {
           .fn()
           .mockRejectedValue(new Error("Thread is locked by another runner")),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
 
       const response = await handleRunAgent({
         runtime,
@@ -512,7 +516,7 @@ describe("handleRunAgent", () => {
           .fn()
           .mockResolvedValue({ joinToken: "jt-123", joinCode: "jc-123" }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
       const response = await handleRunAgent({
         runtime,
         request: new Request("https://example.com/agent/my-agent/run", {
@@ -572,7 +576,7 @@ describe("handleRunAgent", () => {
           .fn()
           .mockResolvedValue({ joinToken: "jt-123", joinCode: "jc-123" }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
 
       const response = await handleRunAgent({
         runtime,
@@ -599,7 +603,7 @@ describe("handleRunAgent", () => {
           joinCode: "jc-created",
         }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any);
+      const runtime = createIntelligenceRuntime(agent, platform);
 
       const response = await handleRunAgent({
         runtime,
@@ -670,7 +674,7 @@ describe("handleRunAgent", () => {
           joinCode: "jc-created",
         }),
       };
-      const runtime = createIntelligenceRuntime(baseAgent, platform as any, {
+      const runtime = createIntelligenceRuntime(baseAgent, platform, {
         generateThreadNames: true,
       });
       const response = await handleRunAgent({
@@ -725,7 +729,7 @@ describe("handleRunAgent", () => {
           joinCode: "jc-created",
         }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any, {
+      const runtime = createIntelligenceRuntime(agent, platform, {
         generateThreadNames: false,
       });
 
@@ -754,7 +758,7 @@ describe("handleRunAgent", () => {
           joinCode: "jc-created",
         }),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any, {
+      const runtime = createIntelligenceRuntime(agent, platform, {
         generateThreadNames: true,
       });
 
@@ -810,7 +814,7 @@ describe("handleRunAgent", () => {
           joinCode: "jc-created",
         }),
       };
-      const runtime = createIntelligenceRuntime(baseAgent, platform as any, {
+      const runtime = createIntelligenceRuntime(baseAgent, platform, {
         generateThreadNames: true,
       });
       const errorSpy = vi.spyOn(console, "error").mockImplementation(() => {});
@@ -865,7 +869,7 @@ describe("handleRunAgent", () => {
         getThreadMessages: vi.fn(),
         ɵacquireThreadLock: vi.fn(),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any, {
+      const runtime = createIntelligenceRuntime(agent, platform, {
         identifyUser: vi.fn().mockResolvedValue({ id: "" }),
       });
 
@@ -887,7 +891,7 @@ describe("handleRunAgent", () => {
         getThreadMessages: vi.fn(),
         ɵacquireThreadLock: vi.fn(),
       };
-      const runtime = createIntelligenceRuntime(agent, platform as any, {
+      const runtime = createIntelligenceRuntime(agent, platform, {
         identifyUser: vi.fn().mockRejectedValue(new Error("auth failed")),
       });
       const errorSpy = vi.spyOn(console, "error").mockImplementation(() => {});

package/src/v2/runtime/__tests__/handle-threads.test.ts

@@ -7,7 +7,7 @@ import {
   handleSubscribeToThreads,
   handleUpdateThread,
 } from "../handlers/handle-threads";
-import { CopilotRuntime } from "../runtime";
+import { CopilotRuntime } from "../core/runtime";
 
 describe("thread handlers", () => {
   const createIdentifyUser = () => vi.fn().mockResolvedValue({ id: "user-1" });

package/src/v2/runtime/__tests__/handle-transcribe.test.ts

@@ -1,5 +1,5 @@
 import { handleTranscribe } from "../handlers/handle-transcribe";
-import { CopilotRuntime } from "../runtime";
+import { CopilotRuntime } from "../core/runtime";
 import {
   TranscriptionService,
   TranscribeFileOptions,

package/src/v2/runtime/__tests__/hono-adapter.test.ts

@@ -0,0 +1,150 @@
+import { describe, it, expect, vi } from "vitest";
+import type { AbstractAgent } from "@ag-ui/client";
+
+import {
+  createCopilotEndpoint,
+  createCopilotEndpointSingleRoute,
+} from "../endpoints";
+import { CopilotRuntime } from "../core/runtime";
+
+describe("Hono adapter with hooks", () => {
+  const createMockAgent = () => {
+    const agent: unknown = {
+      execute: async () => ({ events: [] }),
+    };
+    (agent as { clone: () => unknown }).clone = () => createMockAgent();
+    return agent as AbstractAgent;
+  };
+
+  const createMockRuntime = (opts?: Partial<CopilotRuntime>) =>
+    new CopilotRuntime({
+      agents: {
+        default: createMockAgent(),
+        myAgent: createMockAgent(),
+        testAgent: createMockAgent(),
+      },
+      ...opts,
+    });
+
+  describe("createCopilotEndpoint", () => {
+    const createEndpoint = (
+      hooks?: Parameters<typeof createCopilotEndpoint>[0]["hooks"],
+      runtimeOpts?: Partial<CopilotRuntime>,
+    ) => {
+      const runtime = createMockRuntime(runtimeOpts);
+      const endpoint = createCopilotEndpoint({
+        runtime,
+        basePath: "/",
+        hooks,
+      });
+      return { endpoint, runtime };
+    };
+
+    it("routes GET /info correctly", async () => {
+      const { endpoint } = createEndpoint();
+      const response = await endpoint.fetch(
+        new Request("https://example.com/info"),
+      );
+
+      expect(response.status).toBe(200);
+      const body = await response.json();
+      expect(body).toHaveProperty("version");
+    });
+
+    it("routes POST /agent/:id/run correctly", async () => {
+      const { endpoint } = createEndpoint();
+      const response = await endpoint.fetch(
+        new Request("https://example.com/agent/myAgent/run", {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({
+            messages: [],
+            state: {},
+            threadId: "thread-1",
+          }),
+        }),
+      );
+
+      expect(response.status).not.toBe(404);
+    });
+
+    it("forwards hooks — onRequest hook is called", async () => {
+      const onRequest = vi.fn();
+      const { endpoint, runtime } = createEndpoint({ onRequest });
+
+      await endpoint.fetch(new Request("https://example.com/info"));
+
+      expect(onRequest).toHaveBeenCalledTimes(1);
+      expect(onRequest).toHaveBeenCalledWith(
+        expect.objectContaining({
+          request: expect.any(Request),
+          path: expect.stringContaining("/info"),
+          runtime,
+        }),
+      );
+    });
+
+    it("hooks can modify response headers via onResponse", async () => {
+      const { endpoint } = createEndpoint({
+        onResponse: ({ response }) => {
+          const headers = new Headers(response.headers);
+          headers.set("x-custom-header", "hello-from-hook");
+          return new Response(response.body, {
+            status: response.status,
+            headers,
+          });
+        },
+      });
+
+      const response = await endpoint.fetch(
+        new Request("https://example.com/info"),
+      );
+
+      expect(response.status).toBe(200);
+      expect(response.headers.get("x-custom-header")).toBe("hello-from-hook");
+    });
+
+    it("Hono CORS headers are present", async () => {
+      const { endpoint } = createEndpoint();
+      const response = await endpoint.fetch(
+        new Request("https://example.com/info", {
+          method: "OPTIONS",
+          headers: {
+            Origin: "https://somesite.com",
+            "Access-Control-Request-Method": "GET",
+          },
+        }),
+      );
+
+      expect(response.headers.get("Access-Control-Allow-Origin")).toBe("*");
+    });
+  });
+
+  describe("createCopilotEndpointSingleRoute", () => {
+    it("dispatches single-route with method envelope", async () => {
+      const runtime = createMockRuntime();
+      const endpoint = createCopilotEndpointSingleRoute({
+        runtime,
+        basePath: "/rpc",
+      });
+
+      const response = await endpoint.fetch(
+        new Request("https://example.com/rpc", {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({
+            method: "agent/run",
+            params: { agentId: "myAgent" },
+            body: {
+              messages: [],
+              state: {},
+              threadId: "thread-1",
+            },
+          }),
+        }),
+      );
+
+      expect(response.status).not.toBe(404);
+    });
+  });
+});