@cloudbase/agent-agents 0.0.3 → 0.0.7

package/dist/index.mjs

@@ -1873,6 +1873,11 @@ var AgentExecutor = class {
     const customHandler = this.config.controlFlow?.customHandler;
     while (maxIterations === void 0 || iteration < maxIterations) {
       iteration++;
+      if (options?.signal?.aborted) {
+        throw new Error(
+          `Run aborted: ${options.signal.reason || "AbortSignal was aborted"}`
+        );
+      }
       const controlResult = await this.checkControlFlow(
         customHandler,
         context,
@@ -1937,7 +1942,8 @@ var AgentExecutor = class {
     );
     const streamResult = await this.processStreamingResponse(
       llmResponse,
-      eventEmitter
+      eventEmitter,
+      options
     );
     const assistantMessage = await this.saveAssistantMessage(
       streamResult.content,
@@ -2309,6 +2315,9 @@ var AgentExecutor = class {
       requestParams.tools = toolDefinitions;
       requestParams.tool_choice = "auto";
     }
+    if (options?.signal) {
+      requestParams.signal = options.signal;
+    }
     return this.modelProvider.chat.completions.stream(requestParams);
   }
   /**
@@ -4100,6 +4109,17 @@ var Agent = class {
         }
       }
     };
+    if (options?.signal?.aborted) {
+      throw new ExecutionError(
+        "Run aborted before start",
+        { reason: options.signal.reason || "AbortSignal was already aborted" }
+      );
+    }
+    const abortHandler = () => {
+      this.isRunning = false;
+      this.state.status = "idle";
+    };
+    options?.signal?.addEventListener("abort", abortHandler);
     eventEmitter.emit("RUN_STARTED" /* RUN_STARTED */, {
       type: "RUN_STARTED" /* RUN_STARTED */,
       threadId: conversationId,
@@ -4113,6 +4133,12 @@ var Agent = class {
       this.state.status = "running";
       this.state.runId = runId;
       this.state.conversationId = conversationId;
+      if (options?.signal?.aborted) {
+        throw new ExecutionError(
+          "Run aborted",
+          { reason: options.signal.reason || "AbortSignal was aborted" }
+        );
+      }
       if (options?.resume) {
         try {
           const resumePayload = JSON.parse(options.resume.payload);
@@ -4132,12 +4158,24 @@ var Agent = class {
         state: this.state
       });
       const context = await this.createRunContext(input, state, options);
+      if (options?.signal?.aborted) {
+        throw new ExecutionError(
+          "Run aborted",
+          { reason: options.signal.reason || "AbortSignal was aborted" }
+        );
+      }
       const result = await this.executeAgentCore(
         eventEmitter,
         context,
         validatedOptions,
         true
       );
+      if (options?.signal?.aborted) {
+        throw new ExecutionError(
+          "Run aborted",
+          { reason: options.signal.reason || "AbortSignal was aborted" }
+        );
+      }
       if (result && result.metadata && result.metadata.outcome !== "interrupt") {
         eventEmitter.emit("RUN_FINISHED" /* RUN_FINISHED */, {
           type: "RUN_FINISHED" /* RUN_FINISHED */,
@@ -4166,6 +4204,10 @@ var Agent = class {
       this.isRunning = false;
       this.state.status = "error";
       throw agentError;
+    } finally {
+      if (options?.signal) {
+        options.signal.removeEventListener("abort", abortHandler);
+      }
     }
   }
   // Streaming run method - Returns AsyncIterator (no RxJS dependency)
@@ -4977,8 +5019,12 @@ var OpenAIProvider = class extends BaseModelProvider {
   async createCompletion(params) {
     try {
       const openaiParams = this.convertToOpenAIParams(params);
+      const requestOptions = {};
+      if (params.signal) {
+        requestOptions.signal = params.signal;
+      }
       const response = await this.withRetry(async () => {
-        return await this.client.chat.completions.create(openaiParams);
+        return await this.client.chat.completions.create(openaiParams, requestOptions);
       });
       return this.convertFromOpenAIResponse(response);
     } catch (error) {
@@ -4992,10 +5038,14 @@ var OpenAIProvider = class extends BaseModelProvider {
   async *createStream(params) {
     try {
       const openaiParams = this.convertToOpenAIParams(params);
+      const requestOptions = {};
+      if (params.signal) {
+        requestOptions.signal = params.signal;
+      }
       const stream = await this.client.chat.completions.create({
         ...openaiParams,
         stream: true
-      });
+      }, requestOptions);
       if (Symbol.asyncIterator in stream) {
         for await (const chunk of stream) {
           yield this.convertFromOpenAIChunk(chunk);
@@ -5062,6 +5112,9 @@ var OpenAIProvider = class extends BaseModelProvider {
   // Handle OpenAI-specific errors
   handleOpenAIError(error) {
     if (error instanceof Error) {
+      if (error.name === "AbortError" || error.message.includes("abort")) {
+        return this.createError("Request aborted", "aborted", error);
+      }
       if (error.message.includes("401") || error.message.includes("Unauthorized")) {
         return this.createError("Invalid API key", "authentication", error);
       }
@@ -5223,8 +5276,12 @@ var AnthropicProvider = class extends BaseModelProvider {
   async createCompletion(params) {
     try {
       const anthropicParams = this.convertToAnthropicParams(params);
+      const requestOptions = {};
+      if (params.signal) {
+        requestOptions.signal = params.signal;
+      }
       const response = await this.withRetry(async () => {
-        return await this.client.messages.create(anthropicParams);
+        return await this.client.messages.create(anthropicParams, requestOptions);
       });
       return this.convertFromAnthropicResponse(response, params.model);
     } catch (error) {
@@ -5239,7 +5296,11 @@ var AnthropicProvider = class extends BaseModelProvider {
     try {
       const anthropicParams = this.convertToAnthropicParams(params);
       anthropicParams.stream = true;
-      const stream = await this.client.messages.create(anthropicParams);
+      const requestOptions = {};
+      if (params.signal) {
+        requestOptions.signal = params.signal;
+      }
+      const stream = await this.client.messages.create(anthropicParams, requestOptions);
       if (Symbol.asyncIterator in stream) {
         for await (const chunk of stream) {
           yield this.convertFromAnthropicChunk(chunk, params.model);
@@ -5345,6 +5406,9 @@ var AnthropicProvider = class extends BaseModelProvider {
   // Handle Anthropic-specific errors
   handleAnthropicError(error) {
     if (error instanceof Error) {
+      if (error.name === "AbortError" || error.message.includes("abort")) {
+        return this.createError("Request aborted", "aborted", error);
+      }
       if (error.message.includes("401") || error.message.includes("Unauthorized")) {
         return this.createError("Invalid API key", "authentication", error);
       }