agents 0.14.3 → 0.14.5

package/dist/chat/index.d.ts

@@ -677,6 +677,13 @@ declare class ResumableStream {
   private _isFlushingChunks;
   private _lastCleanupTime;
   constructor(sql: SqlTaggedTemplate);
+  /**
+   * Add the #1691 recovery column to the metadata table for rows created before
+   * it existed. Inspects the current schema and only runs `alter table` when the
+   * column is absent — idempotent across Durable Object restarts, with no
+   * error-driven control flow.
+   */
+  private _migrateMetadataColumns;
   get activeStreamId(): string | null;
   get activeRequestId(): string | null;
   hasActiveStream(): boolean;
@@ -691,7 +698,19 @@ declare class ResumableStream {
    * @param requestId - The unique ID of the chat request
    * @returns The generated stream ID
    */
-  start(requestId: string): string;
+  start(
+    requestId: string,
+    options?: {
+      messageId?: string;
+    }
+  ): string;
+  /**
+   * The assistant message id an orphaned stream was producing — the same id the
+   * live path persists under, so recovery re-associates reconstructed chunks
+   * with the correct message (#1691). Returns null when the row is missing or
+   * is a legacy row written before the `message_id` column existed.
+   */
+  getStreamMessageId(streamId: string): string | null;
   /**
    * Mark a stream as completed and flush any pending chunks.
    * @param streamId - The stream to mark as completed

package/dist/chat/index.js

@@ -606,10 +606,22 @@ var ResumableStream = class ResumableStream {
       created_at integer not null,
       completed_at integer
     )`;
+		this._migrateMetadataColumns();
 		this.sql`create index if not exists idx_stream_chunks_stream_id 
       on cf_ai_chat_stream_chunks(stream_id, chunk_index)`;
 		this.restore();
 	}
+	/**
+	* Add the #1691 recovery column to the metadata table for rows created before
+	* it existed. Inspects the current schema and only runs `alter table` when the
+	* column is absent — idempotent across Durable Object restarts, with no
+	* error-driven control flow.
+	*/
+	_migrateMetadataColumns() {
+		if (!(this.sql`
+        select name from pragma_table_info('cf_ai_chat_stream_metadata')
+      ` ?? []).some((column) => column.name === "message_id")) this.sql`alter table cf_ai_chat_stream_metadata add column message_id text`;
+	}
 	get activeStreamId() {
 		return this._activeStreamId;
 	}
@@ -632,20 +644,35 @@ var ResumableStream = class ResumableStream {
 	* @param requestId - The unique ID of the chat request
 	* @returns The generated stream ID
 	*/
-	start(requestId) {
+	start(requestId, options = {}) {
 		this.flushBuffer();
 		const streamId = nanoid();
 		this._activeStreamId = streamId;
 		this._activeRequestId = requestId;
 		this._segmentIndex = 0;
 		this._isLive = true;
+		const messageId = options.messageId ?? null;
 		this.sql`
-      insert into cf_ai_chat_stream_metadata (id, request_id, status, created_at)
-      values (${streamId}, ${requestId}, 'streaming', ${Date.now()})
+      insert into cf_ai_chat_stream_metadata (id, request_id, status, created_at, message_id)
+      values (${streamId}, ${requestId}, 'streaming', ${Date.now()}, ${messageId})
     `;
 		return streamId;
 	}
 	/**
+	* The assistant message id an orphaned stream was producing — the same id the
+	* live path persists under, so recovery re-associates reconstructed chunks
+	* with the correct message (#1691). Returns null when the row is missing or
+	* is a legacy row written before the `message_id` column existed.
+	*/
+	getStreamMessageId(streamId) {
+		const rows = this.sql`
+      select message_id from cf_ai_chat_stream_metadata
+      where id = ${streamId}
+    `;
+		if (!rows || rows.length === 0) return null;
+		return rows[0].message_id ?? null;
+	}
+	/**
 	* Mark a stream as completed and flush any pending chunks.
 	* @param streamId - The stream to mark as completed
 	*/