Break out mastra copilot dependency to subpackage (#312)

* refactor: move copilot stuff to its own file

* update package.json to accomodate separate import

Author: alecf

typescript-sdk/integrations/mastra/package.json

@@ -14,6 +14,25 @@
     "dist/**",
     "README.md"
   ],
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    },
+    "./copilotkit": {
+      "types": "./dist/copilotkit.d.ts",
+      "import": "./dist/copilotkit.mjs",
+      "require": "./dist/copilotkit.js"
+    }
+  },
+  "typesVersions": {
+    "*": {
+      "copilotkit": [
+        "dist/copilotkit.d.ts"
+      ]
+    }
+  },
   "scripts": {
     "build": "tsup",
     "dev": "tsup --watch",
@@ -23,6 +42,20 @@
     "link:global": "pnpm link --global",
     "unlink:global": "pnpm unlink --global"
   },
+  "tsup": {
+    "entry": {
+      "index": "src/index.ts",
+      "copilotkit": "src/copilotkit.ts"
+    },
+    "dts": true,
+    "format": [
+      "cjs",
+      "esm"
+    ],
+    "splitting": false,
+    "sourcemap": true,
+    "clean": true
+  },
   "dependencies": {
     "@ai-sdk/ui-utils": "^1.1.19",
     "@mastra/client-js": "^0.10.18",

typescript-sdk/integrations/mastra/src/copilotkit.ts

@@ -0,0 +1,56 @@
+import { AbstractAgent } from "@ag-ui/client";
+import {
+  CopilotRuntime,
+  copilotRuntimeNodeHttpEndpoint,
+  CopilotServiceAdapter,
+  ExperimentalEmptyAdapter,
+} from "@copilotkit/runtime";
+import { RuntimeContext } from "@mastra/core/runtime-context";
+import { registerApiRoute } from "@mastra/core/server";
+import { MastraAgent } from "./mastra";
+export function registerCopilotKit<T extends Record<string, any> | unknown = unknown>({
+  path,
+  resourceId,
+  serviceAdapter = new ExperimentalEmptyAdapter(),
+  agents,
+  setContext,
+}: {
+  path: string;
+  resourceId: string;
+  serviceAdapter?: CopilotServiceAdapter;
+  agents?: Record<string, AbstractAgent>;
+  setContext?: (c: any, runtimeContext: RuntimeContext<T>) => void | Promise<void>;
+}) {
+  return registerApiRoute(path, {
+    method: `ALL`,
+    handler: async (c) => {
+      const mastra = c.get("mastra");
+
+      const runtimeContext = new RuntimeContext<T>();
+
+      if (setContext) {
+        await setContext(c, runtimeContext);
+      }
+
+      const aguiAgents =
+        agents ||
+        MastraAgent.getLocalAgents({
+          resourceId,
+          mastra,
+          runtimeContext,
+        });
+
+      const runtime = new CopilotRuntime({
+        agents: aguiAgents,
+      });
+
+      const handler = copilotRuntimeNodeHttpEndpoint({
+        endpoint: path,
+        runtime,
+        serviceAdapter,
+      });
+
+      return handler.handle(c.req.raw, {});
+    },
+  });
+}

typescript-sdk/integrations/mastra/src/utils.ts

@@ -1,17 +1,9 @@
 import type { Message } from "@ag-ui/client";
 import { AbstractAgent } from "@ag-ui/client";
-import {
-  CopilotRuntime,
-  copilotRuntimeNodeHttpEndpoint,
-  CopilotServiceAdapter,
-  ExperimentalEmptyAdapter,
-} from "@copilotkit/runtime";
-import type { CoreMessage } from "@mastra/core";
-import { registerApiRoute } from "@mastra/core/server";
-import type { Mastra } from "@mastra/core";
+import { MastraClient } from "@mastra/client-js";
+import type { CoreMessage, Mastra } from "@mastra/core";
 import { Agent as LocalMastraAgent } from "@mastra/core/agent";
 import { RuntimeContext } from "@mastra/core/runtime-context";
-import { MastraClient } from "@mastra/client-js";
 import { MastraAgent } from "./mastra";
 
 export function convertAGUIMessagesToMastra(messages: Message[]): CoreMessage[] {
@@ -66,53 +58,6 @@ export function convertAGUIMessagesToMastra(messages: Message[]): CoreMessage[]
   return result;
 }
 
-export function registerCopilotKit<T extends Record<string, any> | unknown = unknown>({
-  path,
-  resourceId,
-  serviceAdapter = new ExperimentalEmptyAdapter(),
-  agents,
-  setContext,
-}: {
-  path: string;
-  resourceId: string;
-  serviceAdapter?: CopilotServiceAdapter;
-  agents?: Record<string, AbstractAgent>;
-  setContext?: (c: any, runtimeContext: RuntimeContext<T>) => void | Promise<void>;
-}) {
-  return registerApiRoute(path, {
-    method: `ALL`,
-    handler: async (c) => {
-      const mastra = c.get("mastra");
-
-      const runtimeContext = new RuntimeContext<T>();
-
-      if (setContext) {
-        await setContext(c, runtimeContext);
-      }
-
-      const aguiAgents =
-        agents ||
-        MastraAgent.getLocalAgents({
-          resourceId,
-          mastra,
-          runtimeContext,
-        });
-
-      const runtime = new CopilotRuntime({
-        agents: aguiAgents,
-      });
-
-      const handler = copilotRuntimeNodeHttpEndpoint({
-        endpoint: path,
-        runtime,
-        serviceAdapter,
-      });
-
-      return handler.handle(c.req.raw, {});
-    },
-  });
-}
-
 export interface GetRemoteAgentsOptions {
   mastraClient: MastraClient;
   resourceId?: string;