@voltagent/core 0.1.49 → 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

@@ -401,7 +401,6 @@ var _AgentEventEmitter = class extends import_node_events.EventEmitter {
     const { agentId, historyId, event, skipPropagation = false, parentHistoryEntryId } = params;
     const agent = AgentRegistry.getInstance().getAgent(agentId);
     if (!agent) {
-      import_dev2.devLogger.warn("Agent not found: ", agentId);
       return void 0;
     }
     const historyManager = agent.getHistoryManager();
@@ -1599,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 {
@@ -1633,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 {
@@ -1909,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 {
@@ -1955,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,
@@ -3502,9 +3514,7 @@ var LibSQLStorage = class {
     }
     try {
       const migrationResult = await this.migrateAgentHistorySchema();
-      if (migrationResult.success) {
-        import_dev6.devLogger.info("Agent history schema migration completed successfully");
-      } else {
+      if (!migrationResult.success) {
         import_dev6.devLogger.error("Agent history schema migration error:", migrationResult.error);
       }
     } catch (error) {
@@ -7705,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) {
@@ -8472,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;
   }
@@ -8606,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
    */
@@ -8662,7 +8722,8 @@ ${retrieverContext}`;
       parentHistoryEntryId,
       parentOperationContext,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -8671,7 +8732,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -8730,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,
@@ -8843,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
@@ -8922,17 +8988,21 @@ ${retrieverContext}`;
         status: "completed"
       });
       operationContext.isActive = false;
-      const extendedResponse = {
+      const initialResponse = {
         ...response,
         userContext: new Map(operationContext.userContext)
       };
       await this.hooks.onEnd?.({
         conversationId: finalConversationId,
         agent: this,
-        output: extendedResponse,
+        output: initialResponse,
         error: void 0,
         context: operationContext
       });
+      const extendedResponse = {
+        ...response,
+        userContext: new Map(operationContext.userContext)
+      };
       this.updateHistoryEntry(operationContext, {
         output: response.text,
         usage: response.usage,
@@ -9014,7 +9084,8 @@ ${retrieverContext}`;
       parentHistoryEntryId,
       parentOperationContext,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -9023,7 +9094,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -9080,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,
@@ -9200,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
@@ -9224,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
@@ -9325,17 +9401,21 @@ ${retrieverContext}`;
           }
         });
         operationContext.isActive = false;
-        const resultWithContext = {
+        const initialResult = {
           ...result,
           userContext: new Map(operationContext.userContext)
         };
         await this.hooks.onEnd?.({
           agent: this,
-          output: resultWithContext,
+          output: initialResult,
           error: void 0,
           conversationId: finalConversationId,
           context: operationContext
         });
+        const resultWithContext = {
+          ...result,
+          userContext: new Map(operationContext.userContext)
+        };
         if (internalOptions.provider?.onFinish) {
           await internalOptions.provider.onFinish(
             resultWithContext
@@ -9351,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",
@@ -9469,7 +9549,8 @@ ${retrieverContext}`;
       parentHistoryEntryId,
       parentOperationContext,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -9478,7 +9559,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -9537,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,
@@ -9620,17 +9706,21 @@ ${retrieverContext}`;
         endTime: /* @__PURE__ */ new Date(),
         status: "completed"
       });
-      const extendedResponse = {
+      const initialResponse = {
         ...response,
         userContext: new Map(operationContext.userContext)
       };
       await this.hooks.onEnd?.({
         agent: this,
-        output: extendedResponse,
+        output: initialResponse,
         error: void 0,
         conversationId: finalConversationId,
         context: operationContext
       });
+      const extendedResponse = {
+        ...response,
+        userContext: new Map(operationContext.userContext)
+      };
       return extendedResponse;
     } catch (error) {
       const voltagentError = error;
@@ -9707,7 +9797,8 @@ ${retrieverContext}`;
       parentOperationContext,
       provider,
       contextLimit = 10,
-      userContext
+      userContext,
+      signal
     } = internalOptions;
     const operationContext = await this.initializeHistory(input, "working", {
       parentAgentId,
@@ -9716,7 +9807,8 @@ ${retrieverContext}`;
       userContext,
       userId,
       conversationId: initialConversationId,
-      parentOperationContext
+      parentOperationContext,
+      signal
     });
     const { messages: contextMessages, conversationId: finalConversationId } = await this.memoryManager.prepareConversationContext(
       operationContext,
@@ -9773,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,
@@ -9864,17 +9960,21 @@ ${retrieverContext}`;
             status: "completed"
           });
           operationContext.isActive = false;
-          const resultWithContext = {
+          const initialResult = {
             ...result,
             userContext: new Map(operationContext.userContext)
           };
           await this.hooks.onEnd?.({
             agent: this,
-            output: resultWithContext,
+            output: initialResult,
             error: void 0,
             conversationId: finalConversationId,
             context: operationContext
           });
+          const resultWithContext = {
+            ...result,
+            userContext: new Map(operationContext.userContext)
+          };
           if (provider?.onFinish) {
             await provider.onFinish(
               resultWithContext