fix(mastra): update dependencies and fix integration

Author: roaminro

typescript-sdk/integrations/mastra/package.json

@@ -26,16 +26,16 @@
   "dependencies": {
     "@ag-ui/client": "workspace:*",
     "@ai-sdk/ui-utils": "^1.1.19",
-    "@mastra/client-js": "^0.10.9",
+    "@mastra/client-js": "^0.10.18",
     "rxjs": "7.8.1"
   },
   "peerDependencies": {
-    "@copilotkit/runtime": "^1.8.13",
-    "@mastra/core": "^0.10.10 || ^0.11.1",
+    "@copilotkit/runtime": "^1.9.3",
+    "@mastra/core": "^0.11.1 || ^0.12.1",
     "zod": "^3.25.67"
   },
   "devDependencies": {
-    "@mastra/core": "^0.11.1",
+    "@mastra/core": "^0.12.1",
     "@types/jest": "^29.5.14",
     "@types/node": "^20.11.19",
     "jest": "^29.7.0",

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

@@ -73,9 +73,8 @@ export class MastraAgent extends AbstractAgent {
         subscriber.next(runStartedEvent);
 
         // Handle local agent memory management (from Mastra implementation)
-        if ("metrics" in this.agent) {
-          // @ts-ignore
-          const memory = this.agent.getMemory();
+        if (this.isLocalMastraAgent(this.agent)) {
+          const memory = await this.agent.getMemory();
 
           if (memory && input.state && Object.keys(input.state || {}).length > 0) {
             let thread: StorageThreadType | null = await memory.getThreadById({
@@ -163,14 +162,12 @@ export class MastraAgent extends AbstractAgent {
               subscriber.error(error);
             },
             onRunFinished: async () => {
-              if ("metrics" in this.agent) {
+              if (this.isLocalMastraAgent(this.agent)) {
                 try {
-                  // @ts-ignore
-                  const memory = this.agent.getMemory();
+                  const memory = await this.agent.getMemory();
                   if (memory) {
                     const workingMemory = await memory.getWorkingMemory({
                       threadId: input.threadId,
-                      // @ts-ignore
                       memoryConfig: {
                         workingMemory: {
                           enabled: true,
@@ -219,6 +216,10 @@ export class MastraAgent extends AbstractAgent {
     });
   }
 
+  isLocalMastraAgent(agent: LocalMastraAgent | RemoteMastraAgent): agent is LocalMastraAgent {
+    return "getMemory" in agent;
+  }
+
   /**
    * Streams in process or remote mastra agent.
    * @param input - The input for the mastra agent.
@@ -251,13 +252,7 @@ export class MastraAgent extends AbstractAgent {
     const convertedMessages = convertAGUIMessagesToMastra(messages);
     const runtimeContext = this.runtimeContext;
 
-    function isLocalMastraAgent(
-      agent: LocalMastraAgent | RemoteMastraAgent,
-    ): agent is LocalMastraAgent {
-      return "metrics" in agent;
-    }
-
-    if (isLocalMastraAgent(this.agent)) {
+    if (this.isLocalMastraAgent(this.agent)) {
       // Local agent - use the agent's stream method directly
       try {
         const response = await this.agent.stream(convertedMessages, {