chat 4.16.0 → 4.17.0

package/dist/index.js

@@ -59,7 +59,7 @@ import {
   toModalElement,
   toPlainText,
   walkAst
-} from "./chunk-7S5DLTN2.js";
+} from "./chunk-Q32GJ2PR.js";
 
 // src/channel.ts
 import { WORKFLOW_DESERIALIZE as WORKFLOW_DESERIALIZE2, WORKFLOW_SERIALIZE as WORKFLOW_SERIALIZE2 } from "@workflow/serde";
@@ -613,6 +613,41 @@ function extractMessageContent(message) {
 // src/thread.ts
 import { WORKFLOW_DESERIALIZE as WORKFLOW_DESERIALIZE3, WORKFLOW_SERIALIZE as WORKFLOW_SERIALIZE3 } from "@workflow/serde";
 
+// src/from-full-stream.ts
+var STREAM_CHUNK_TYPES = /* @__PURE__ */ new Set([
+  "markdown_text",
+  "task_update",
+  "plan_update"
+]);
+async function* fromFullStream(stream) {
+  let needsSeparator = false;
+  let hasEmittedText = false;
+  for await (const event of stream) {
+    if (typeof event === "string") {
+      yield event;
+      continue;
+    }
+    if (event === null || typeof event !== "object" || !("type" in event)) {
+      continue;
+    }
+    const typed = event;
+    if (STREAM_CHUNK_TYPES.has(typed.type)) {
+      yield event;
+      continue;
+    }
+    if (typed.type === "text-delta" && typeof typed.textDelta === "string") {
+      if (needsSeparator && hasEmittedText) {
+        yield "\n\n";
+      }
+      needsSeparator = false;
+      hasEmittedText = true;
+      yield typed.textDelta;
+    } else if (typed.type === "step-finish") {
+      needsSeparator = true;
+    }
+  }
+}
+
 // src/streaming-markdown.ts
 import remend from "remend";
 var StreamingMarkdownRenderer = class {
@@ -1080,9 +1115,11 @@ var ThreadImpl = class _ThreadImpl {
   }
   /**
    * Handle streaming from an AsyncIterable.
-   * Uses adapter's native streaming if available, otherwise falls back to post+edit.
+   * Normalizes the stream (supports both textStream and fullStream from AI SDK),
+   * then uses adapter's native streaming if available, otherwise falls back to post+edit.
    */
-  async handleStream(textStream) {
+  async handleStream(rawStream) {
+    const textStream = fromFullStream(rawStream);
     const options = {};
     if (this._currentMessage) {
       options.recipientUserId = this._currentMessage.author.userId;
@@ -1098,7 +1135,12 @@ var ThreadImpl = class _ThreadImpl {
             async next() {
               const result = await iterator.next();
               if (!result.done) {
-                accumulated += result.value;
+                const value = result.value;
+                if (typeof value === "string") {
+                  accumulated += value;
+                } else if (value.type === "markdown_text") {
+                  accumulated += value.text;
+                }
               }
               return result;
             }
@@ -1112,7 +1154,29 @@ var ThreadImpl = class _ThreadImpl {
         raw.threadId
       );
     }
-    return this.fallbackStream(textStream, options);
+    const textOnlyStream = {
+      [Symbol.asyncIterator]: () => {
+        const iterator = textStream[Symbol.asyncIterator]();
+        return {
+          async next() {
+            while (true) {
+              const result = await iterator.next();
+              if (result.done) {
+                return { value: void 0, done: true };
+              }
+              const value = result.value;
+              if (typeof value === "string") {
+                return { value, done: false };
+              }
+              if (value.type === "markdown_text") {
+                return { value: value.text, done: false };
+              }
+            }
+          }
+        };
+      }
+    };
+    return this.fallbackStream(textOnlyStream, options);
   }
   async startTyping(status) {
     await this.adapter.startTyping(this.id, status);
@@ -2131,30 +2195,32 @@ var Chat = class {
           }
           modalElement = converted;
         }
-        const isEphemeralMessage = event.messageId?.startsWith("ephemeral:");
         let message;
-        if (isEphemeralMessage) {
-          const recentMessage = thread.recentMessages[0];
-          if (recentMessage && typeof recentMessage.toJSON === "function") {
-            message = recentMessage;
-          }
-        } else if (event.messageId && event.adapter.fetchMessage) {
-          const fetched = await event.adapter.fetchMessage(event.threadId, event.messageId).catch(() => null);
-          if (fetched) {
-            message = new Message(fetched);
-          } else {
+        if (thread) {
+          const isEphemeralMessage = event.messageId?.startsWith("ephemeral:");
+          if (isEphemeralMessage) {
             const recentMessage = thread.recentMessages[0];
             if (recentMessage && typeof recentMessage.toJSON === "function") {
               message = recentMessage;
             }
+          } else if (event.messageId && event.adapter.fetchMessage) {
+            const fetched = await event.adapter.fetchMessage(event.threadId, event.messageId).catch(() => null);
+            if (fetched) {
+              message = new Message(fetched);
+            } else {
+              const recentMessage = thread.recentMessages[0];
+              if (recentMessage && typeof recentMessage.toJSON === "function") {
+                message = recentMessage;
+              }
+            }
           }
         }
         const contextId = crypto.randomUUID();
-        const channel = thread.channel;
+        const channel = thread ? thread.channel : void 0;
         this.storeModalContext(
           event.adapter.name,
           contextId,
-          thread,
+          thread ? thread : void 0,
           message,
           channel
         );
@@ -2410,15 +2476,18 @@ var Chat = class {
       return;
     }
     const dedupeKey = `dedupe:${adapter.name}:${message.id}`;
-    const alreadyProcessed = await this._stateAdapter.get(dedupeKey);
-    if (alreadyProcessed) {
+    const isFirstProcess = await this._stateAdapter.setIfNotExists(
+      dedupeKey,
+      true,
+      this._dedupeTtlMs
+    );
+    if (!isFirstProcess) {
       this.logger.debug("Skipping duplicate message", {
         adapter: adapter.name,
         messageId: message.id
       });
       return;
     }
-    await this._stateAdapter.set(dedupeKey, true, this._dedupeTtlMs);
     const lock = await this._stateAdapter.acquireLock(
       threadId,
       DEFAULT_LOCK_TTL_MS
@@ -3005,6 +3074,7 @@ export {
   deriveChannelId,
   emoji,
   emphasis,
+  fromFullStream,
   fromReactElement2 as fromReactElement,
   fromReactModalElement2 as fromReactModalElement,
   getEmoji,