@cydm/pie 2.6.11 → 2.7.0

package/dist/chunks/{chunk-PAC62XU4.js → chunk-TWILEFLD.js}

@@ -1,9 +1,10 @@
 import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
 import {
   EventStream,
+  disposeModelTransports,
   streamSimple,
   validateToolArguments
-} from "./chunk-7B4IIBVS.js";
+} from "./chunk-R5I2QLF2.js";
 
 // ../../packages/agent-core/src/agent-loop.ts
 function createLoopErrorMessage(error, config, signal) {
@@ -261,7 +262,15 @@ async function streamAssistantResponse(context, config, signal, stream, streamFn
       }
     }
   }
-  return await response.result();
+  const finalResult = await response.result();
+  if (signal?.aborted && finalResult.stopReason === "stop") {
+    const adjusted = { ...finalResult, stopReason: "aborted" };
+    if (addedPartial && context.messages.length > 0) {
+      context.messages[context.messages.length - 1] = adjusted;
+    }
+    return adjusted;
+  }
+  return finalResult;
 }
 async function executeToolCalls(tools, assistantMessage, signal, stream, getSteeringMessages, hooks) {
   const toolCalls = assistantMessage.content.filter((c) => c.type === "toolCall");
@@ -449,6 +458,7 @@ var Agent = class {
   followUpMode;
   streamFn;
   _sessionId;
+  transportOwner = {};
   _apiKey;
   getApiKey;
   runningPrompt;
@@ -496,12 +506,14 @@ var Agent = class {
     return this._sessionId;
   }
   set sessionId(value) {
+    if (this._sessionId !== value) this.releaseTransport();
     this._sessionId = value;
   }
   get apiKey() {
     return this._apiKey;
   }
   set apiKey(value) {
+    if (this._apiKey !== value) this.releaseTransport();
     this._apiKey = value;
   }
   get thinkingBudgets() {
@@ -534,6 +546,7 @@ var Agent = class {
     this._state.systemPrompt = v;
   }
   setModel(m) {
+    this.releaseTransport();
     this._state.model = m;
   }
   setThinkingLevel(l) {
@@ -546,12 +559,14 @@ var Agent = class {
     this.hooks = hooks;
   }
   replaceMessages(ms) {
+    this.releaseTransport();
     this._state.messages = ms.slice();
   }
   appendMessage(m) {
     this._state.messages = [...this._state.messages, m];
   }
   clearMessages() {
+    this.releaseTransport();
     this._state.messages = [];
   }
   steer(m) {
@@ -611,13 +626,18 @@ var Agent = class {
     this.followUpQueue = [];
     return followUp;
   }
+  releaseTransport() {
+    disposeModelTransports(this.transportOwner);
+  }
   abort() {
+    this.releaseTransport();
     this.abortController?.abort();
   }
   waitForIdle() {
     return this.runningPrompt ?? Promise.resolve();
   }
   reset() {
+    this.releaseTransport();
     this._state.messages = [];
     this._state.isStreaming = false;
     this._state.streamMessage = null;
@@ -763,6 +783,7 @@ var Agent = class {
       reasoning,
       apiKey: this._apiKey,
       sessionId: this._sessionId,
+      transportOwner: this.transportOwner,
       thinkingBudgets: this._thinkingBudgets,
       maxRetryDelayMs: this._maxRetryDelayMs,
       convertToLlm: this.convertToLlm,
@@ -833,7 +854,8 @@ var Agent = class {
           (c) => c.type === "thinking" && c.thinking?.trim().length > 0 || c.type === "text" && c.text?.trim().length > 0 || c.type === "toolCall" && c.name?.trim().length > 0
         );
         if (!onlyEmpty) {
-          this.appendMessage(partial);
+          const effective = this.abortController?.signal.aborted && partial.stopReason === "stop" ? { ...partial, stopReason: "aborted" } : partial;
+          this.appendMessage(effective);
         }
       }
     } catch (err) {

package/dist/chunks/{src-HPIBCE2I.js → src-E7NCXMU5.js}

@@ -6,8 +6,8 @@ import {
   bindToolCallerIdentity,
   createToolCallerIdentity,
   getToolCallerIdentity
-} from "./chunk-PAC62XU4.js";
-import "./chunk-7B4IIBVS.js";
+} from "./chunk-TWILEFLD.js";
+import "./chunk-R5I2QLF2.js";
 import "./chunk-XXGYUYXI.js";
 import "./chunk-LZQXQOPJ.js";
 export {

package/dist/chunks/{test-stream-Y7WXELAJ.js → test-stream-BWBJTFNI.js}

@@ -1,7 +1,7 @@
 import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
 import {
   EventStream
-} from "./chunk-7B4IIBVS.js";
+} from "./chunk-R5I2QLF2.js";
 import "./chunk-XXGYUYXI.js";
 import "./chunk-LZQXQOPJ.js";