@witqq/agent-sdk 0.8.0 → 0.9.0

package/dist/chat/server.cjs

@@ -195,16 +195,22 @@ var SSEChatTransport = class {
 };
 async function streamToTransport(events, transport) {
   try {
-    let accumulatedText = "";
+    const textChunks = [];
+    let finishReason;
     for await (const event of events) {
       if (!transport.isOpen) break;
+      if (event.type === "done") {
+        finishReason = event.finishReason;
+        continue;
+      }
       transport.send(event);
       if (event.type === "message:delta") {
-        accumulatedText += event.text;
+        textChunks.push(event.text);
       }
     }
     if (transport.isOpen) {
-      transport.send({ type: "done", finalOutput: accumulatedText || void 0 });
+      const finalOutput = textChunks.length > 0 ? textChunks.join("") : void 0;
+      transport.send({ type: "done", finalOutput, finishReason });
     }
     transport.close();
   } catch (err) {
@@ -1394,6 +1400,7 @@ var ChatRuntime = class {
   _backends;
   _sessionStore;
   _contextConfig;
+  _ctxManager;
   _middleware;
   _tools = /* @__PURE__ */ new Map();
   _retryConfig;
@@ -1413,6 +1420,9 @@ var ChatRuntime = class {
     this._defaultBackend = options.defaultBackend;
     this._sessionStore = options.sessionStore;
     this._contextConfig = options.context;
+    if (this._contextConfig) {
+      this._ctxManager = new ContextWindowManager(this._contextConfig);
+    }
     this._middleware = [...options.middleware ?? []];
     this._retryConfig = options.retryConfig;
     this._onContextTrimmed = options.onContextTrimmed;
@@ -1593,8 +1603,8 @@ var ChatRuntime = class {
   }
   /** Stage 5: Auto-trim context window if configured. Returns session snapshot for adapter. */
   async trimSessionContext(cid, session, model) {
-    if (!this._contextConfig) return session;
-    const ctxManager = new ContextWindowManager(this._contextConfig);
+    if (!this._ctxManager) return session;
+    const ctxManager = this._ctxManager;
     const lastUsage = this._sessionUsage.get(cid);
     const modelContextWindow = model ? this._modelContextWindows.get(model) : void 0;
     if (lastUsage && modelContextWindow) {