@copilotkit/runtime 1.10.5-next.0 → 1.10.5-next.10

package/dist/lib/integrations/node-http/index.mjs

@@ -1,6 +1,6 @@
 import {
   copilotRuntimeNodeHttpEndpoint
-} from "../../../chunk-HZWVUP7K.mjs";
+} from "../../../chunk-NFB4H6NT.mjs";
 import "../../../chunk-SHBDMA63.mjs";
 import "../../../chunk-F7IANE7Z.mjs";
 import "../../../chunk-XWBDEXDA.mjs";

package/dist/service-adapters/index.d.ts

@@ -1,10 +1,10 @@
-import { b as CopilotServiceAdapter, C as CopilotRuntimeChatCompletionRequest, a as CopilotRuntimeChatCompletionResponse } from '../langserve-3f99bf2e.js';
-export { c as RemoteChain, R as RemoteChainParameters } from '../langserve-3f99bf2e.js';
+import { b as CopilotServiceAdapter, C as CopilotRuntimeChatCompletionRequest, a as CopilotRuntimeChatCompletionResponse } from '../langserve-0c6100e3.js';
+export { c as RemoteChain, R as RemoteChainParameters } from '../langserve-0c6100e3.js';
 export { convertServiceAdapterError } from './shared/index.js';
-import { L as LangChainAdapter } from '../groq-adapter-4ba7b2f0.js';
-export { G as GoogleGenerativeAIAdapter, f as GroqAdapter, e as GroqAdapterParams, a as OpenAIAdapter, O as OpenAIAdapterParams, c as OpenAIAssistantAdapter, b as OpenAIAssistantAdapterParams, d as UnifyAdapter, U as UnifyAdapterParams } from '../groq-adapter-4ba7b2f0.js';
+import { L as LangChainAdapter } from '../groq-adapter-c8aec5c5.js';
+export { G as GoogleGenerativeAIAdapter, f as GroqAdapter, e as GroqAdapterParams, a as OpenAIAdapter, O as OpenAIAdapterParams, c as OpenAIAssistantAdapter, b as OpenAIAssistantAdapterParams, d as UnifyAdapter, U as UnifyAdapterParams } from '../groq-adapter-c8aec5c5.js';
 import Anthropic from '@anthropic-ai/sdk';
-import '../index-d4614f9b.js';
+import '../index-96b330da.js';
 import '../graphql/types/base/index.js';
 import 'rxjs';
 import '@copilotkit/shared';

package/dist/{shared-26958730.d.ts → shared-0a7346ce.d.ts}

@@ -3,8 +3,8 @@ import * as createPinoLogger from 'pino';
 import createPinoLogger__default from 'pino';
 import { YogaInitialContext, createYoga } from 'graphql-yoga';
 import { Parameter, Action, CopilotErrorHandler } from '@copilotkit/shared';
-import { b as CopilotServiceAdapter, A as ActionInput, d as AgentSessionInput, e as AgentStateInput, F as ForwardedParametersInput, E as ExtensionsInput, R as RemoteChainParameters, f as RuntimeEventSource, g as ExtensionsResponse } from './langserve-3f99bf2e.js';
-import { M as MessageInput, a as Message } from './index-d4614f9b.js';
+import { b as CopilotServiceAdapter, A as ActionInput, d as AgentSessionInput, e as AgentStateInput, F as ForwardedParametersInput, E as ExtensionsInput, R as RemoteChainParameters, f as RuntimeEventSource, g as ExtensionsResponse } from './langserve-0c6100e3.js';
+import { b as MessageInput, M as Message } from './index-96b330da.js';
 import { CopilotCloudOptions } from './lib/cloud/index.js';
 import { AbstractAgent } from '@ag-ui/client';
 
@@ -20,6 +20,11 @@ declare class MetaEventInput {
     messages?: MessageInput[];
 }
 
+declare class CopilotContextInput {
+    description: string;
+    value: string;
+}
+
 type EndpointDefinition = CopilotKitEndpoint | LangGraphPlatformEndpoint;
 declare enum EndpointType {
     CopilotKit = "copilotKit",
@@ -194,6 +199,7 @@ declare function generateMcpToolInstructions(toolsMap: Record<string, MCPTool>):
  */
 
 type CreateMCPClientFunction = (config: MCPEndpointConfig) => Promise<MCPClient>;
+
 interface CopilotRuntimeRequest {
     serviceAdapter: CopilotServiceAdapter;
     messages: MessageInput[];
@@ -209,6 +215,7 @@ interface CopilotRuntimeRequest {
     url?: string;
     extensions?: ExtensionsInput;
     metaEvents?: MetaEventInput[];
+    context?: CopilotContextInput[];
 }
 interface CopilotRuntimeResponse {
     threadId: string;
@@ -239,6 +246,14 @@ interface OnAfterRequestOptions {
     url?: string;
 }
 type OnAfterRequestHandler = (options: OnAfterRequestOptions) => void | Promise<void>;
+interface OnStopGenerationOptions {
+    threadId: string;
+    runId?: string;
+    url?: string;
+    agentName?: string;
+    lastMessage: MessageInput;
+}
+type OnStopGenerationHandler = (options: OnStopGenerationOptions) => void | Promise<void>;
 interface Middleware {
     /**
      * A function that is called before the request is processed.
@@ -355,6 +370,7 @@ interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []> {
      * ```
      */
     onError?: CopilotErrorHandler;
+    onStopGeneration?: OnStopGenerationHandler;
 }
 declare class CopilotRuntime<const T extends Parameter[] | [] = []> {
     actions: ActionsConfiguration<T>;
@@ -363,6 +379,7 @@ declare class CopilotRuntime<const T extends Parameter[] | [] = []> {
     private langserve;
     private onBeforeRequest?;
     private onAfterRequest?;
+    private onStopGeneration?;
     private delegateAgentProcessingToServiceAdapter;
     private observability?;
     private availableAgents;

package/package.json

@@ -9,7 +9,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.10.5-next.0",
+  "version": "1.10.5-next.10",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -37,11 +37,11 @@
     "tsconfig": "1.4.6"
   },
   "dependencies": {
-    "@ag-ui/client": "^0.0.37",
-    "@ag-ui/core": "^0.0.37",
-    "@ag-ui/encoder": "^0.0.37",
-    "@ag-ui/langgraph": "^0.0.12",
-    "@ag-ui/proto": "^0.0.37",
+    "@ag-ui/client": "^0.0.39",
+    "@ag-ui/core": "^0.0.39",
+    "@ag-ui/encoder": "^0.0.39",
+    "@ag-ui/langgraph": "^0.0.17",
+    "@ag-ui/proto": "^0.0.39",
     "@anthropic-ai/sdk": "^0.57.0",
     "@graphql-yoga/plugin-defer-stream": "^3.3.1",
     "@langchain/aws": "^0.1.9",
@@ -67,13 +67,13 @@
     "rxjs": "7.8.1",
     "type-graphql": "2.0.0-rc.1",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.10.5-next.0"
+    "@copilotkit/shared": "1.10.5-next.10"
   },
   "peerDependencies": {
-    "@ag-ui/client": ">=0.0.37",
-    "@ag-ui/core": ">=0.0.37",
-    "@ag-ui/encoder": ">=0.0.37",
-    "@ag-ui/proto": ">=0.0.37"
+    "@ag-ui/client": ">=0.0.39",
+    "@ag-ui/core": ">=0.0.39",
+    "@ag-ui/encoder": ">=0.0.39",
+    "@ag-ui/proto": ">=0.0.39"
   },
   "keywords": [
     "copilotkit",

package/src/graphql/inputs/copilot-context.input.ts

@@ -0,0 +1,10 @@
+import { Field, InputType } from "type-graphql";
+
+@InputType()
+export class CopilotContextInput {
+  @Field(() => String)
+  description: string;
+
+  @Field(() => String)
+  value: string;
+}

package/src/graphql/inputs/generate-copilot-response.input.ts

@@ -8,6 +8,7 @@ import { AgentSessionInput } from "./agent-session.input";
 import { AgentStateInput } from "./agent-state.input";
 import { ExtensionsInput } from "./extensions.input";
 import { MetaEventInput } from "./meta-event.input";
+import { CopilotContextInput } from "./copilot-context.input";
 
 @InputType()
 export class GenerateCopilotResponseMetadataInput {
@@ -52,4 +53,7 @@ export class GenerateCopilotResponseInput {
 
   @Field(() => [MetaEventInput], { nullable: true })
   metaEvents?: MetaEventInput[];
+
+  @Field(() => [CopilotContextInput], { nullable: true })
+  context?: CopilotContextInput[];
 }

package/src/graphql/resolvers/copilot.resolver.ts

@@ -257,6 +257,7 @@ export class CopilotResolver {
         url: data.frontend.url,
         extensions: data.extensions,
         metaEvents: data.metaEvents,
+        context: data.context,
       });
     } catch (error) {
       // Catch structured CopilotKit errors at the main mutation level and re-throw as GraphQL errors

package/src/lib/runtime/agui-action.ts

@@ -16,6 +16,7 @@ import { AbstractAgent } from "@ag-ui/client";
 import { CopilotKitError, CopilotKitErrorCode, parseJson } from "@copilotkit/shared";
 import { MetaEventInput } from "../../graphql/inputs/meta-event.input";
 import { GraphQLContext } from "../integrations/shared";
+import { CopilotContextInput } from "../../graphql/inputs/copilot-context.input";
 
 export function constructAGUIRemoteAction({
   logger,
@@ -25,6 +26,7 @@ export function constructAGUIRemoteAction({
   metaEvents,
   threadMetadata,
   nodeName,
+  context,
   graphqlContext,
 }: {
   logger: Logger;
@@ -34,6 +36,7 @@ export function constructAGUIRemoteAction({
   metaEvents?: MetaEventInput[];
   threadMetadata?: Record<string, any>;
   nodeName?: string;
+  context?: CopilotContextInput[];
   graphqlContext: GraphQLContext;
 }) {
   const action = {
@@ -45,6 +48,13 @@ export function constructAGUIRemoteAction({
       actionInputsWithoutAgents,
       threadId,
     }: RemoteAgentHandlerParams): Promise<Observable<RuntimeEvent>> => {
+      graphqlContext.request.signal.addEventListener(
+        "abort",
+        () => {
+          agent.abortRun();
+        },
+        { once: true }, // optional: fire only once
+      );
       logger.debug({ actionName: agent.agentId }, "Executing remote agent");
 
       const agentWireMessages = convertMessagesToAGUIMessage(messages);
@@ -92,6 +102,7 @@ export function constructAGUIRemoteAction({
         agent.legacy_to_be_removed_runAgentBridged({
           tools,
           forwardedProps,
+          context,
         }) as Observable<RuntimeEvent>
       ).pipe(
         mergeMap((event) => {

package/src/lib/runtime/copilot-runtime.ts

@@ -90,6 +90,7 @@ type CreateMCPClientFunction = (config: MCPEndpointConfig) => Promise<MCPClient>
 // --- MCP Imports ---
 
 import { generateHelpfulErrorMessage } from "../streaming";
+import { CopilotContextInput } from "../../graphql/inputs/copilot-context.input";
 
 export interface CopilotRuntimeRequest {
   serviceAdapter: CopilotServiceAdapter;
@@ -106,6 +107,7 @@ export interface CopilotRuntimeRequest {
   url?: string;
   extensions?: ExtensionsInput;
   metaEvents?: MetaEventInput[];
+  context?: CopilotContextInput[];
 }
 
 interface CopilotRuntimeResponse {
@@ -142,6 +144,15 @@ interface OnAfterRequestOptions {
 
 type OnAfterRequestHandler = (options: OnAfterRequestOptions) => void | Promise<void>;
 
+interface OnStopGenerationOptions {
+  threadId: string;
+  runId?: string;
+  url?: string;
+  agentName?: string;
+  lastMessage: MessageInput;
+}
+type OnStopGenerationHandler = (options: OnStopGenerationOptions) => void | Promise<void>;
+
 interface Middleware {
   /**
    * A function that is called before the request is processed.
@@ -299,6 +310,8 @@ export interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []
    * ```
    */
   onError?: CopilotErrorHandler;
+
+  onStopGeneration?: OnStopGenerationHandler;
 }
 
 export class CopilotRuntime<const T extends Parameter[] | [] = []> {
@@ -308,6 +321,7 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
   private langserve: Promise<Action<any>>[] = [];
   private onBeforeRequest?: OnBeforeRequestHandler;
   private onAfterRequest?: OnAfterRequestHandler;
+  private onStopGeneration?: OnStopGenerationHandler;
   private delegateAgentProcessingToServiceAdapter: boolean;
   private observability?: CopilotObservabilityConfig;
   private availableAgents: Pick<AgentWithEndpoint, "name" | "id">[];
@@ -357,6 +371,7 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
 
     this.onBeforeRequest = params?.middleware?.onBeforeRequest;
     this.onAfterRequest = params?.middleware?.onAfterRequest;
+    this.onStopGeneration = params?.onStopGeneration;
     this.delegateAgentProcessingToServiceAdapter =
       params?.delegateAgentProcessingToServiceAdapter || false;
     this.observability = params?.observability_c;
@@ -484,7 +499,20 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
       agentSession,
       agentStates,
       publicApiKey,
+      context,
     } = request;
+    graphqlContext.request.signal.addEventListener(
+      "abort",
+      () =>
+        this.onStopGeneration?.({
+          threadId,
+          runId,
+          url,
+          agentName: agentSession?.agentName,
+          lastMessage: rawMessages[rawMessages.length - 1],
+        }),
+      { once: true }, // optional: fire only once
+    );
 
     const eventSource = new RuntimeEventSource({
       errorHandler: async (error, context) => {
@@ -1025,6 +1053,7 @@ please use an LLM adapter instead.`,
       metaEvents,
       publicApiKey,
       forwardedParameters,
+      context,
     } = request;
     const { agentName, nodeName } = agentSession;
 
@@ -1408,6 +1437,7 @@ please use an LLM adapter instead.`,
       agents: this.agents,
       metaEvents: request.metaEvents,
       nodeName: request.agentSession?.nodeName,
+      context: request.context,
     });
 
     const configuredActions =

package/src/lib/runtime/langgraph/langgraph-agent.ts

@@ -19,10 +19,18 @@ import {
   ProcessedEvents,
   SchemaKeys,
   type State,
+  StateEnrichment,
 } from "@ag-ui/langgraph";
 import { Message as LangGraphMessage } from "@langchain/langgraph-sdk/dist/types.messages";
 import { ThreadState } from "@langchain/langgraph-sdk";
 
+interface CopilotKitStateEnrichment {
+  copilotkit: {
+    actions: StateEnrichment["ag-ui"]["tools"];
+    context: StateEnrichment["ag-ui"]["context"];
+  };
+}
+
 export interface PredictStateTool {
   tool: string;
   state_key: string;
@@ -179,17 +187,29 @@ export class LangGraphAgent extends AGUILangGraphAgent {
     );
   }
 
-  langGraphDefaultMergeState(state: State, messages: LangGraphMessage[], tools: any): State {
-    const { tools: returnedTools, ...rest } = super.langGraphDefaultMergeState(
-      state,
-      messages,
-      tools,
+  langGraphDefaultMergeState(
+    state: State,
+    messages: LangGraphMessage[],
+    input: RunAgentInput,
+  ): State<StateEnrichment & CopilotKitStateEnrichment> {
+    const aguiMergedState = super.langGraphDefaultMergeState(state, messages, input);
+    const { tools: returnedTools, "ag-ui": agui } = aguiMergedState;
+    // tolerate undefined and de-duplicate by stable key (id | name | key)
+    const rawCombinedTools = [
+      ...((returnedTools as any[]) ?? []),
+      ...((agui?.tools as any[]) ?? []),
+    ];
+    const combinedTools = Array.from(
+      new Map(
+        rawCombinedTools.map((t: any) => [t?.id ?? t?.name ?? t?.key ?? JSON.stringify(t), t]),
+      ).values(),
     );
 
     return {
-      ...rest,
+      ...aguiMergedState,
       copilotkit: {
-        actions: returnedTools,
+        actions: combinedTools,
+        context: agui?.context ?? [],
       },
     };
   }
@@ -201,6 +221,7 @@ export class LangGraphAgent extends AGUILangGraphAgent {
       config: schemaKeys.config,
       input: schemaKeys.input ? [...schemaKeys.input, ...CONSTANT_KEYS] : null,
       output: schemaKeys.output ? [...schemaKeys.output, ...CONSTANT_KEYS] : null,
+      context: schemaKeys.context ? [...schemaKeys.context, ...CONSTANT_KEYS] : null,
     };
   }
 }

package/src/lib/runtime/remote-actions.ts

@@ -19,6 +19,7 @@ import {
 import { MetaEventInput } from "../../graphql/inputs/meta-event.input";
 import { AbstractAgent } from "@ag-ui/client";
 import { constructAGUIRemoteAction } from "./agui-action";
+import { CopilotContextInput } from "../../graphql/inputs/copilot-context.input";
 
 export type EndpointDefinition = CopilotKitEndpoint | LangGraphPlatformEndpoint;
 
@@ -132,6 +133,7 @@ export async function setupRemoteActions({
   agents,
   metaEvents,
   nodeName,
+  context,
 }: {
   remoteEndpointDefinitions: EndpointDefinition[];
   graphqlContext: GraphQLContext;
@@ -141,6 +143,7 @@ export async function setupRemoteActions({
   agents: Record<string, AbstractAgent>;
   metaEvents?: MetaEventInput[];
   nodeName?: string;
+  context?: CopilotContextInput[];
 }): Promise<Action[]> {
   const logger = graphqlContext.logger.child({ component: "remote-actions.fetchRemoteActions" });
   logger.debug({ remoteEndpointDefinitions }, "Fetching from remote endpoints");
@@ -211,6 +214,7 @@ export async function setupRemoteActions({
         threadMetadata,
         nodeName,
         graphqlContext,
+        context,
       }),
     );
   }