@voltagent/core 0.1.50 → 0.1.51

package/dist/index.d.ts

@@ -1227,7 +1227,7 @@ type PublicGenerateOptions = Omit<CommonGenerateOptions, "historyEntryId" | "ope
 /**
  * Agent status information
  */
-type AgentStatus = "idle" | "working" | "tool_calling" | "error" | "completed";
+type AgentStatus = "idle" | "working" | "error" | "completed" | "cancelled";
 /**
  * Tool call definition
  */
@@ -1322,6 +1322,10 @@ type AgentHandoffOptions = {
      * Parent's operation context to merge SubAgent steps into
      */
     parentOperationContext?: OperationContext;
+    /**
+     * AbortSignal to cancel the handoff operation
+     */
+    signal?: AbortSignal;
 };
 /**
  * Result of a handoff to another agent
@@ -1380,6 +1384,8 @@ type OperationContext = {
     toolSpans?: Map<string, Span>;
     /** Conversation steps for building full message history including tool calls/results */
     conversationSteps?: StepWithContent[];
+    /** AbortSignal for cancelling the operation */
+    signal?: AbortSignal;
 };
 /**
  * Tool execution context passed to tool.execute method
@@ -3434,6 +3440,10 @@ declare class Agent<TProvider extends {
      */
     private _endOtelToolSpan;
     private publishTimelineEvent;
+    /**
+     * Sets up abort signal listener for cancellation handling
+     */
+    private setupAbortSignalListener;
     /**
      * Create an enhanced fullStream with real-time SubAgent event injection
      */

package/dist/index.js

@@ -1598,6 +1598,10 @@ app.openapi(streamRoute, async (c) => {
         temperature: 0.7
       }
     } = c.req.valid("json");
+    const abortController = new AbortController();
+    c.req.raw.signal?.addEventListener("abort", () => {
+      abortController.abort();
+    });
     const stream = new ReadableStream({
       async start(controller) {
         try {
@@ -1632,7 +1636,9 @@ app.openapi(streamRoute, async (c) => {
               temperature: options.temperature
               // Note: No onError callback needed - tool errors are handled via fullStream
               // Stream errors are handled by try/catch blocks around fullStream iteration
-            }
+            },
+            // Pass the abort signal to the agent
+            signal: abortController.signal
           };
           const response = await agent.streamText(input, processedStreamOptions);
           try {
@@ -1908,6 +1914,11 @@ app.openapi(streamObjectRoute, async (c) => {
       options = {}
     } = c.req.valid("json");
     const schemaInZodObject = (0, import_zod_from_json_schema.convertJsonSchemaToZod)(schema);
+    const abortController = new AbortController();
+    c.req.raw.signal?.addEventListener("abort", () => {
+      console.log("\u{1F6D1} API: Client aborted object stream request, stopping agent stream...");
+      abortController.abort();
+    });
     const sseStream = new ReadableStream({
       async start(controller) {
         try {
@@ -1954,7 +1965,9 @@ app.openapi(streamObjectRoute, async (c) => {
                 safeEnqueue(errorMessage);
                 safeClose();
               }
-            }
+            },
+            // Pass the abort signal to the agent
+            signal: abortController.signal
           };
           const agentStream = await agent.streamObject(
             input,
@@ -7702,7 +7715,9 @@ Context: ${JSON.stringify(context, null, 2)}`;
         // @ts-expect-error - bad types
         parentAgentId: sourceAgent?.id || parentAgentId,
         parentHistoryEntryId,
-        parentOperationContext
+        parentOperationContext,
+        // Pass the abort signal from parent's operation context to subagent
+        signal: parentOperationContext?.signal
       });
       let finalText = "";
       if (streamResponse.fullStream) {
@@ -8469,7 +8484,9 @@ ${retrieverContext}`;
       parentHistoryEntryId: options.parentHistoryEntryId,
       otelSpan,
       // Use parent's conversationSteps if available (for SubAgents), otherwise create new array
-      conversationSteps: options.parentOperationContext?.conversationSteps || []
+      conversationSteps: options.parentOperationContext?.conversationSteps || [],
+      // Inherit signal from parent context or use provided signal
+      signal: options.parentOperationContext?.signal || options.signal
     };
     return opContext;
   }
@@ -8603,6 +8620,52 @@ ${retrieverContext}`;
       parentHistoryEntryId: operationContext.parentHistoryEntryId
     });
   }
+  /**
+   * Sets up abort signal listener for cancellation handling
+   */
+  setupAbortSignalListener(signal, operationContext, finalConversationId, agentStartEvent) {
+    if (!signal)
+      return;
+    signal.addEventListener("abort", async () => {
+      this.updateHistoryEntry(operationContext, {
+        status: "cancelled",
+        endTime: /* @__PURE__ */ new Date()
+      });
+      operationContext.isActive = false;
+      const cancellationError = new Error("Operation cancelled by user");
+      cancellationError.name = "AbortError";
+      const agentCancelledEvent = {
+        id: crypto.randomUUID(),
+        name: "agent:cancel",
+        type: "agent",
+        startTime: agentStartEvent.startTime,
+        endTime: (/* @__PURE__ */ new Date()).toISOString(),
+        level: "INFO",
+        input: null,
+        statusMessage: {
+          message: cancellationError.message,
+          code: "USER_CANCELLED",
+          stage: "cancelled"
+        },
+        status: "cancelled",
+        metadata: {
+          displayName: this.name,
+          id: this.id,
+          userContext: Object.fromEntries(operationContext.userContext.entries())
+        },
+        traceId: operationContext.historyEntry.id,
+        parentEventId: agentStartEvent.id
+      };
+      this.publishTimelineEvent(operationContext, agentCancelledEvent);
+      await this.hooks.onEnd?.({
+        agent: this,
+        output: void 0,
+        error: cancellationError,
+        conversationId: finalConversationId || "",
+        context: operationContext
+      });
+    });
+  }
   /**
    * Create an enhanced fullStream with real-time SubAgent event injection
    */
@@ -8659,7 +8722,8 @@ ${retrieverContext}`;
       parentHistoryEntryId,
       parentOperationContext,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -8668,7 +8732,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -8727,6 +8792,10 @@ ${retrieverContext}`;
       operationContext.userContext.set("agent_start_time", agentStartTime);
       operationContext.userContext.set("agent_start_event_id", agentStartEvent.id);
       this.publishTimelineEvent(operationContext, agentStartEvent);
+      this.setupAbortSignalListener(signal, operationContext, finalConversationId, {
+        id: agentStartEvent.id,
+        startTime: agentStartTime
+      });
       const onStepFinish = this.memoryManager.createStepFinishHandler(
         operationContext,
         userId,
@@ -8840,7 +8909,7 @@ ${retrieverContext}`;
                   id: crypto.randomUUID(),
                   name: "tool:success",
                   type: "tool",
-                  startTime: (/* @__PURE__ */ new Date()).toISOString(),
+                  startTime: toolStartInfo.startTime,
                   // Use the original start time
                   endTime: (/* @__PURE__ */ new Date()).toISOString(),
                   // Current time as end time
@@ -9015,7 +9084,8 @@ ${retrieverContext}`;
       parentHistoryEntryId,
       parentOperationContext,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -9024,7 +9094,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -9081,6 +9152,10 @@ ${retrieverContext}`;
     operationContext.userContext.set("agent_start_time", agentStartTime);
     operationContext.userContext.set("agent_start_event_id", agentStartEvent.id);
     this.publishTimelineEvent(operationContext, agentStartEvent);
+    this.setupAbortSignalListener(signal, operationContext, finalConversationId, {
+      id: agentStartEvent.id,
+      startTime: agentStartTime
+    });
     const onStepFinish = this.memoryManager.createStepFinishHandler(
       operationContext,
       userId,
@@ -9201,7 +9276,7 @@ ${retrieverContext}`;
                 id: crypto.randomUUID(),
                 name: "tool:error",
                 type: "tool",
-                startTime: (/* @__PURE__ */ new Date()).toISOString(),
+                startTime: toolStartInfo.startTime,
                 // Use the original start time
                 endTime: (/* @__PURE__ */ new Date()).toISOString(),
                 // Current time as end time
@@ -9225,7 +9300,7 @@ ${retrieverContext}`;
                 id: crypto.randomUUID(),
                 name: "tool:success",
                 type: "tool",
-                startTime: (/* @__PURE__ */ new Date()).toISOString(),
+                startTime: toolStartInfo.startTime,
                 // Use the original start time
                 endTime: (/* @__PURE__ */ new Date()).toISOString(),
                 // Current time as end time
@@ -9356,7 +9431,7 @@ ${retrieverContext}`;
               id: crypto.randomUUID(),
               name: "tool:error",
               type: "tool",
-              startTime: (/* @__PURE__ */ new Date()).toISOString(),
+              startTime: toolStartInfo.startTime,
               endTime: (/* @__PURE__ */ new Date()).toISOString(),
               status: "error",
               level: "ERROR",
@@ -9474,7 +9549,8 @@ ${retrieverContext}`;
       parentHistoryEntryId,
       parentOperationContext,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -9483,7 +9559,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -9542,6 +9619,10 @@ ${retrieverContext}`;
       operationContext.userContext.set("agent_start_time", agentStartTime);
       operationContext.userContext.set("agent_start_event_id", agentStartEvent.id);
       this.publishTimelineEvent(operationContext, agentStartEvent);
+      this.setupAbortSignalListener(signal, operationContext, finalConversationId, {
+        id: agentStartEvent.id,
+        startTime: agentStartTime
+      });
       const onStepFinish = this.memoryManager.createStepFinishHandler(
         operationContext,
         userId,
@@ -9716,7 +9797,8 @@ ${retrieverContext}`;
       parentOperationContext,
       provider,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -9725,7 +9807,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -9782,6 +9865,10 @@ ${retrieverContext}`;
     operationContext.userContext.set("agent_start_time", agentStartTime);
     operationContext.userContext.set("agent_start_event_id", agentStartEvent.id);
     this.publishTimelineEvent(operationContext, agentStartEvent);
+    this.setupAbortSignalListener(signal, operationContext, finalConversationId, {
+      id: agentStartEvent.id,
+      startTime: agentStartTime
+    });
     const onStepFinish = this.memoryManager.createStepFinishHandler(
       operationContext,
       userId,