update code

Author: LekoArts

integrations/mastra/typescript/src/copilotkit.ts

@@ -5,9 +5,10 @@ import {
   CopilotServiceAdapter,
   ExperimentalEmptyAdapter,
 } from "@copilotkit/runtime";
-import { RuntimeContext } from "@mastra/core/runtime-context";
+import { RequestContext } from "@mastra/core/request-context";
 import { registerApiRoute } from "@mastra/core/server";
 import { MastraAgent } from "./mastra";
+
 export function registerCopilotKit<T extends Record<string, any> | unknown = unknown>({
   path,
   resourceId,
@@ -19,25 +20,25 @@ export function registerCopilotKit<T extends Record<string, any> | unknown = unk
   resourceId: string;
   serviceAdapter?: CopilotServiceAdapter;
   agents?: Record<string, AbstractAgent>;
-  setContext?: (c: any, runtimeContext: RuntimeContext<T>) => void | Promise<void>;
+  setContext?: (c: any, requestContext: RequestContext<T>) => void | Promise<void>;
 }) {
   return registerApiRoute(path, {
     method: `ALL`,
     handler: async (c) => {
       const mastra = c.get("mastra");
 
-      const runtimeContext = new RuntimeContext<T>();
+      const requestContext = new RequestContext<T>();
 
       if (setContext) {
-        await setContext(c, runtimeContext);
+        await setContext(c, requestContext);
       }
 
       const aguiAgents =
         agents ||
         MastraAgent.getLocalAgents({
           resourceId,
           mastra,
-          runtimeContext,
+          requestContext,
         });
 
       const runtime = new CopilotRuntime({

integrations/mastra/typescript/src/mastra.ts

@@ -14,11 +14,10 @@ import type {
 import { AbstractAgent, EventType } from "@ag-ui/client";
 import type { StorageThreadType } from "@mastra/core";
 import { Agent as LocalMastraAgent } from "@mastra/core/agent";
-import { RuntimeContext } from "@mastra/core/runtime-context";
+import { RequestContext } from "@mastra/core/request-context";
 import { randomUUID } from "@ag-ui/client";
 import { Observable } from "rxjs";
 import { MastraClient } from "@mastra/client-js";
-type RemoteMastraAgent = ReturnType<MastraClient["getAgent"]>;
 import {
   convertAGUIMessagesToMastra,
   GetLocalAgentsOptions,
@@ -31,10 +30,12 @@ import {
   getNetwork,
 } from "./utils";
 
+type RemoteMastraAgent = ReturnType<MastraClient["getAgent"]>;
+
 export interface MastraAgentConfig extends AgentConfig {
   agent: LocalMastraAgent | RemoteMastraAgent;
   resourceId?: string;
-  runtimeContext?: RuntimeContext;
+  requestContext?: RequestContext;
 }
 
 interface MastraAgentStreamOptions {
@@ -53,14 +54,14 @@ interface MastraAgentStreamOptions {
 export class MastraAgent extends AbstractAgent {
   agent: LocalMastraAgent | RemoteMastraAgent;
   resourceId?: string;
-  runtimeContext?: RuntimeContext;
+  requestContext?: RequestContext;
 
   constructor(private config: MastraAgentConfig) {
-    const { agent, resourceId, runtimeContext, ...rest } = config;
+    const { agent, resourceId, requestContext, ...rest } = config;
     super(rest);
     this.agent = agent;
     this.resourceId = resourceId;
-    this.runtimeContext = runtimeContext ?? new RuntimeContext();
+    this.requestContext = requestContext ?? new RequestContext();
   }
 
   public clone() {
@@ -270,8 +271,8 @@ export class MastraAgent extends AbstractAgent {
     const resourceId = this.resourceId ?? threadId;
 
     const convertedMessages = convertAGUIMessagesToMastra(messages);
-    this.runtimeContext?.set("ag-ui", { context: inputContext });
-    const runtimeContext = this.runtimeContext;
+    this.requestContext?.set("ag-ui", { context: inputContext });
+    const requestContext = this.requestContext;
 
     if (this.isLocalMastraAgent(this.agent)) {
       // Local agent - use the agent's stream method directly
@@ -281,7 +282,7 @@ export class MastraAgent extends AbstractAgent {
           resourceId,
           runId,
           clientTools,
-          runtimeContext,
+          requestContext,
         });
 
         // For local agents, the response should already be a stream

integrations/mastra/typescript/src/utils.ts

@@ -3,7 +3,7 @@ import { AbstractAgent } from "@ag-ui/client";
 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 { RequestContext } from "@mastra/core/request-context";
 import { MastraAgent } from "./mastra";
 
 const toMastraTextContent = (content: Message["content"]): string => {
@@ -116,13 +116,13 @@ export async function getRemoteAgents({
 export interface GetLocalAgentsOptions {
   mastra: Mastra;
   resourceId?: string;
-  runtimeContext?: RuntimeContext;
+  requestContext?: RequestContext;
 }
 
 export function getLocalAgents({
   mastra,
   resourceId,
-  runtimeContext,
+  requestContext,
 }: GetLocalAgentsOptions): Record<string, AbstractAgent> {
   const agents = mastra.getAgents() || {};
 
@@ -132,7 +132,7 @@ export function getLocalAgents({
         agentId,
         agent,
         resourceId,
-        runtimeContext,
+        requestContext,
       });
       return acc;
     },
@@ -146,14 +146,14 @@ export interface GetLocalAgentOptions {
   mastra: Mastra;
   agentId: string;
   resourceId?: string;
-  runtimeContext?: RuntimeContext;
+  requestContext?: RequestContext;
 }
 
 export function getLocalAgent({
   mastra,
   agentId,
   resourceId,
-  runtimeContext,
+  requestContext,
 }: GetLocalAgentOptions) {
   const agent = mastra.getAgent(agentId);
   if (!agent) {
@@ -163,18 +163,18 @@ export function getLocalAgent({
     agentId,
     agent,
     resourceId,
-    runtimeContext,
+    requestContext,
   }) as AbstractAgent;
 }
 
 export interface GetNetworkOptions {
   mastra: Mastra;
   networkId: string;
   resourceId?: string;
-  runtimeContext?: RuntimeContext;
+  requestContext?: RequestContext;
 }
 
-export function getNetwork({ mastra, networkId, resourceId, runtimeContext }: GetNetworkOptions) {
+export function getNetwork({ mastra, networkId, resourceId, requestContext }: GetNetworkOptions) {
   const network = mastra.getAgent(networkId);
   if (!network) {
     throw new Error(`Network ${networkId} not found`);
@@ -183,6 +183,6 @@ export function getNetwork({ mastra, networkId, resourceId, runtimeContext }: Ge
     agentId: network.name!,
     agent: network as unknown as LocalMastraAgent,
     resourceId,
-    runtimeContext,
+    requestContext,
   }) as AbstractAgent;
 }