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

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

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

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

@@ -246,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.
@@ -362,6 +370,7 @@ interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []> {
      * ```
      */
     onError?: CopilotErrorHandler;
+    onStopGeneration?: OnStopGenerationHandler;
 }
 declare class CopilotRuntime<const T extends Parameter[] | [] = []> {
     actions: ActionsConfiguration<T>;
@@ -370,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.1",
+  "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.13",
-    "@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.1"
+    "@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/lib/runtime/agui-action.ts

@@ -48,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);

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

@@ -144,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.
@@ -301,6 +310,8 @@ export interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []
    * ```
    */
   onError?: CopilotErrorHandler;
+
+  onStopGeneration?: OnStopGenerationHandler;
 }
 
 export class CopilotRuntime<const T extends Parameter[] | [] = []> {
@@ -310,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">[];
@@ -359,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;
@@ -488,6 +501,18 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
       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) => {

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

@@ -221,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,
     };
   }
 }