simple-ai-sdk 1.0.11 → 1.0.14

package/dist/client/chat-session-utils.d.ts

@@ -1,31 +1,17 @@
-import type { Message, JSONValue } from "../shared/index.js";
+import { type JSONValue } from "../shared/index.js";
 export type PrefetchChatSessionOptions = {
-    initialMessages: Message[];
     api: string;
     body?: Record<string, JSONValue>;
-    startStream?: boolean;
+    /**
+     * ここにcontentを指定すると呼び出し時に即座に送信される
+     */
+    streamMessageContent?: string;
 };
 /**
  * ページ遷移前にセッションを事前に初期化し、オプションでストリーミングを開始するフック
  *
  * @returns prefetchChatSession - セッションの初期化とストリーミング開始を行う関数
  *
- * @example
- * ```tsx
- * const prefetch = usePrefetchChatSession();
- *
- * // セッション初期化のみ
- * prefetch('session-1', {
- *   initialMessages: [{ id: '1', role: 'user', content: 'Hello' }]
- * });
- *
- * // セッション初期化とストリーミング開始（完了を待たない）
- * prefetch('session-2', {
- *   initialMessages: [{ id: '1', role: 'user', content: 'Hello' }],
- *   startStream: true,
- *   api: '/api/chat'
- * });
- * ```
  */
 export declare function usePrefetchChatSession(): (sessionId: string, options: PrefetchChatSessionOptions) => void;
 //# sourceMappingURL=chat-session-utils.d.ts.map

package/dist/client/chat-session-utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-session-utils.d.ts","sourceRoot":"","sources":["../../src/client/chat-session-utils.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAG7D,MAAM,MAAM,0BAA0B,GAAG;IACvC,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACjC,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,sBAAsB,gBAItB,MAAM,WAAW,0BAA0B,UA2B1D"}
+{"version":3,"file":"chat-session-utils.d.ts","sourceRoot":"","sources":["../../src/client/chat-session-utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,SAAS,EAAc,MAAM,oBAAoB,CAAC;AAGhE,MAAM,MAAM,0BAA0B,GAAG;IACvC,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACjC;;OAEG;IACH,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B,CAAC;AAEF;;;;;GAKG;AACH,wBAAgB,sBAAsB,gBAItB,MAAM,WAAW,0BAA0B,UAqC1D"}

package/dist/client/chat-session-utils.js

@@ -1,38 +1,29 @@
 import { useCallback } from "react";
+import { generateId } from "../shared/index.js";
 import { useAIStreamManager } from "./context.js";
 /**
  * ページ遷移前にセッションを事前に初期化し、オプションでストリーミングを開始するフック
  *
  * @returns prefetchChatSession - セッションの初期化とストリーミング開始を行う関数
  *
- * @example
- * ```tsx
- * const prefetch = usePrefetchChatSession();
- *
- * // セッション初期化のみ
- * prefetch('session-1', {
- *   initialMessages: [{ id: '1', role: 'user', content: 'Hello' }]
- * });
- *
- * // セッション初期化とストリーミング開始（完了を待たない）
- * prefetch('session-2', {
- *   initialMessages: [{ id: '1', role: 'user', content: 'Hello' }],
- *   startStream: true,
- *   api: '/api/chat'
- * });
- * ```
  */
 export function usePrefetchChatSession() {
     const manager = useAIStreamManager();
     const prefetchChatSession = useCallback((sessionId, options) => {
         // セッションの初期化（すでに存在する場合はスキップされる）
-        manager.initSession(sessionId, options.initialMessages);
+        manager.initSession(sessionId);
         // startStreamがtrueの場合、ストリーミングを開始（完了を待たない）
-        if (options.startStream) {
+        if (options.streamMessageContent) {
             // streamChatを呼び出すが、awaitせずに即座に返す
             // これによりストリーミングはバックグラウンドで継続される
             manager
-                .streamChat(sessionId, options.initialMessages, {
+                .streamChat(sessionId, [
+                {
+                    id: generateId({ prefix: "msg-" }),
+                    role: "user",
+                    content: options.streamMessageContent,
+                },
+            ], {
                 api: options.api,
                 body: options.body,
             })

package/dist/client/manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/client/manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,qBAAqB,EACrB,kBAAkB,EACnB,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAI7D,MAAM,WAAW,cAAc;IAC7B,SAAS,CAAC,QAAQ,EAAE,qBAAqB,GAAG,MAAM,IAAI,CAAC;IACvD,WAAW,IAAI,kBAAkB,CAAC;IAClC,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,CAAC;IACvD,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAClE;;;;OAIG;IACH,kBAAkB,CAChB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC,GAAG,SAAS,GACnD,IAAI,CAAC;IACR;;OAEG;IACH,oBAAoB,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,KAAK,IAAI,CAAC,GAAG,SAAS,GACnE,IAAI,CAAC;IACR;;OAEG;IACH,iBAAiB,CACf,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,GAC5C,IAAI,CAAC;IACR;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACvC;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACxC;;;OAGG;IACH,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACtC,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC;IACtE,UAAU,CACR,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE;QACP,GAAG,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KAClC,GACA,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;OAGG;IACH,OAAO,IAAI,IAAI,CAAC;CACjB;AAED,MAAM,MAAM,oBAAoB,GAAG;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,cAAM,aAAc,YAAW,cAAc;IAC3C,OAAO,CAAC,KAAK,CAEX;IAEF,OAAO,CAAC,SAAS,CAAoC;IAGrD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,KAAK,CAAuB;IACpC,OAAO,CAAC,gBAAgB,CAA8C;IAEtE,OAAO,CAAC,MAAM;IAMd,OAAO,CAAC,MAAM;IAMd,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,KAAK,CAAU;IACvB,OAAO,CAAC,KAAK,CAAC,CAAS;IACvB,OAAO,CAAC,UAAU,CAA+C;IAEjE,OAAO,CAAC,iBAAiB,CAA6B;IAEtD,OAAO,CAAC,cAAc,CAAoD;IAE1E,OAAO,CAAC,gBAAgB,CAGpB;IAEJ,OAAO,CAAC,aAAa,CAA6C;IAElE,OAAO,CAAC,SAAS,CAA6B;gBAElC,OAAO,GAAE,oBAAyB;IA2B9C,SAAS,CAAC,QAAQ,EAAE,qBAAqB,GAAG,MAAM,IAAI;IAOtD,kBAAkB,CAChB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC,GAAG,SAAS;IAStD,oBAAoB,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,KAAK,IAAI,CAAC,GAAG,SAAS;IAStE,iBAAiB,CACf,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS;IAW/C,WAAW,IAAI,kBAAkB;IAMjC,OAAO,CAAC,MAAM;IAiDd,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAUtD,YAAY,CAAC,SAAS,EAAE,MAAM;IAO9B,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,eAAe,GAAE,OAAO,EAAO;IA8B9D,aAAa,CAAC,SAAS,EAAE,MAAM;IAQ/B,cAAc,CAAC,SAAS,EAAE,MAAM;IAahC,OAAO,CAAC,mBAAmB;IAmD3B,OAAO,CAAC,cAAc;IAmBtB,aAAa,CACX,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,EAC7B,SAAS,UAAQ;IAwBb,UAAU,CACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE;QACP,GAAG,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KAClC;IAqVH,UAAU,CAAC,SAAS,EAAE,MAAM;IAgB5B;;OAEG;IACH,OAAO,CAAC,YAAY;IAsCpB;;;OAGG;IACH,OAAO,IAAI,IAAI;CAuBhB;AAGD,OAAO,EAAE,aAAa,EAAE,CAAC;AAIzB,eAAO,MAAM,aAAa,EAAE,cAAoC,CAAC"}
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/client/manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,qBAAqB,EACrB,kBAAkB,EACnB,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAI7D,MAAM,WAAW,cAAc;IAC7B,SAAS,CAAC,QAAQ,EAAE,qBAAqB,GAAG,MAAM,IAAI,CAAC;IACvD,WAAW,IAAI,kBAAkB,CAAC;IAClC,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,CAAC;IACvD,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAClE;;;;OAIG;IACH,kBAAkB,CAChB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC,GAAG,SAAS,GACnD,IAAI,CAAC;IACR;;OAEG;IACH,oBAAoB,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,KAAK,IAAI,CAAC,GAAG,SAAS,GACnE,IAAI,CAAC;IACR;;OAEG;IACH,iBAAiB,CACf,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,GAC5C,IAAI,CAAC;IACR;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACvC;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACxC;;;OAGG;IACH,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACtC,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC;IACtE,UAAU,CACR,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE;QACP,GAAG,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KAClC,GACA,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;OAGG;IACH,OAAO,IAAI,IAAI,CAAC;CACjB;AAED,MAAM,MAAM,oBAAoB,GAAG;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,cAAM,aAAc,YAAW,cAAc;IAC3C,OAAO,CAAC,KAAK,CAEX;IAEF,OAAO,CAAC,SAAS,CAAoC;IAGrD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,KAAK,CAAuB;IACpC,OAAO,CAAC,gBAAgB,CAA8C;IAEtE,OAAO,CAAC,MAAM;IAMd,OAAO,CAAC,MAAM;IAMd,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,KAAK,CAAU;IACvB,OAAO,CAAC,KAAK,CAAC,CAAS;IACvB,OAAO,CAAC,UAAU,CAA+C;IAEjE,OAAO,CAAC,iBAAiB,CAA6B;IAEtD,OAAO,CAAC,cAAc,CAAoD;IAE1E,OAAO,CAAC,gBAAgB,CAGpB;IAEJ,OAAO,CAAC,aAAa,CAA6C;IAElE,OAAO,CAAC,SAAS,CAA6B;gBAElC,OAAO,GAAE,oBAAyB;IAoB9C,SAAS,CAAC,QAAQ,EAAE,qBAAqB,GAAG,MAAM,IAAI;IAOtD,kBAAkB,CAChB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC,GAAG,SAAS;IAStD,oBAAoB,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,KAAK,IAAI,CAAC,GAAG,SAAS;IAStE,iBAAiB,CACf,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS;IAW/C,WAAW,IAAI,kBAAkB;IAMjC,OAAO,CAAC,MAAM;IAiDd,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAUtD,YAAY,CAAC,SAAS,EAAE,MAAM;IAO9B,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,eAAe,GAAE,OAAO,EAAO;IA8B9D,aAAa,CAAC,SAAS,EAAE,MAAM;IAqB/B,cAAc,CAAC,SAAS,EAAE,MAAM;IAahC,OAAO,CAAC,mBAAmB;IAmD3B,OAAO,CAAC,cAAc;IAmBtB,aAAa,CACX,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,EAC7B,SAAS,UAAQ;IAwBb,UAAU,CACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE;QACP,GAAG,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KAClC;IAqVH,UAAU,CAAC,SAAS,EAAE,MAAM;IAgB5B;;OAEG;IACH,OAAO,CAAC,YAAY;IAqCpB;;;OAGG;IACH,OAAO,IAAI,IAAI;CAgChB;AAGD,OAAO,EAAE,aAAa,EAAE,CAAC;AAIzB,eAAO,MAAM,aAAa,EAAE,cAAoC,CAAC"}

package/dist/client/manager.js

@@ -44,20 +44,13 @@ class StreamManager {
         // TTL が設定されている場合、一定間隔で古いセッションを掃除する
         const sweepEnabled = this.ttlMs && this.ttlMs > 0;
         if (sweepEnabled) {
-            // サーバーサイドでは setInterval を使わずに、必要時に手動で sweep を呼ぶ
+            // サーバーサイドでは sweepは不要
             if (typeof window !== "undefined") {
                 // ブラウザ環境でのみ setInterval を使用
                 this.sweepTimer = setInterval(() => {
                     this.performSweep();
                 }, this.ttlMs);
             }
-            // サーバーサイドでは初回実行のみ行う
-            else {
-                // 初回のみ実行してタイマーは設定しない
-                setTimeout(() => {
-                    this.performSweep();
-                }, 0);
-            }
         }
     }
     // 状態変更を監視するリスナーを登録する
@@ -189,9 +182,22 @@ class StreamManager {
     }
     retainSession(sessionId) {
         const prev = this.activeSubscribers.get(sessionId) ?? 0;
-        this.activeSubscribers.set(sessionId, prev + 1);
+        const next = prev + 1;
+        this.activeSubscribers.set(sessionId, next);
         if (this.debug) {
-            console.info(`[SM] retainSession: ${sessionId} -> ${prev + 1}`);
+            console.info(`[SM] retainSession: ${sessionId} -> ${next}`);
+        }
+        // 修正意図（日本語）:
+        // - 0→1 に購読者数が増えた直後は、これまで抑制されていたストリーミング更新を
+        //   即座に UI に反映させたいケースがある。
+        // - そこで、対象セッションが streaming/submitted の場合に限り、即時通知を一度だけ行う。
+        //   （注意: 現在の useChatSession 実装では retain が subscribe より先に呼ばれるため、
+        //    この通知は新規購読者より前に発生することがある。将来の呼び出し順や他の購読者に対しては有効。）
+        if (next === 1) {
+            const s = this.state.sessions.get(sessionId);
+            if (s && (s.status === "streaming" || s.status === "submitted")) {
+                this.notify(true);
+            }
         }
     }
     releaseSession(sessionId) {
@@ -666,6 +672,15 @@ class StreamManager {
             clearInterval(this.sweepTimer);
             this.sweepTimer = null;
         }
+        // 追加クリーンアップ（意図）:
+        // - 参照を明示的に破棄してリークを防止する。
+        // - Provider のライフサイクル終了時にのみ実行される想定。
+        this.listeners.clear();
+        this.activeSubscribers.clear();
+        this.latestOnFinish.clear();
+        this.latestOnMetadata.clear();
+        this.latestOnError.clear();
+        this.streamIds.clear();
     }
 }
 // Export the StreamManager class for custom initialization

package/dist/client/useChatSession.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChatSession.d.ts","sourceRoot":"","sources":["../../src/client/useChatSession.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAK9E,wBAAgB,cAAc,CAC5B,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE,qBAA0B,GAClC,oBAAoB,CA8GtB"}
+{"version":3,"file":"useChatSession.d.ts","sourceRoot":"","sources":["../../src/client/useChatSession.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAK9E,wBAAgB,cAAc,CAC5B,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE,qBAA0B,GAClC,oBAAoB,CA2ItB"}

package/dist/client/useChatSession.js

@@ -1,37 +1,54 @@
-import { useEffect, useSyncExternalStore, useCallback, useLayoutEffect, } from "react";
+import { useEffect, useSyncExternalStore, useCallback, useRef } from "react";
 import { useAIStreamManager } from "./context.js";
 import { generateId } from "../shared/generate-id.js";
 export function useChatSession(sessionId, options = {}) {
     const { api = "/api/chat", initialMessages = [], onMetadata, onError, onFinish, } = options;
     const manager = useAIStreamManager();
-    useLayoutEffect(() => {
-        // このフックの購読開始を通知（TTL GC から保護）
+    // セッションIDの変更検知用。初期化時に正しい initialMessages を渡すため、
+    // レンダリング中に参照を更新する（副作用ではないので安全）。
+    const lastSessionIdRef = useRef(sessionId);
+    const initialMessagesRef = useRef(initialMessages);
+    if (lastSessionIdRef.current !== sessionId) {
+        initialMessagesRef.current = initialMessages;
+        lastSessionIdRef.current = sessionId;
+    }
+    // useSyncExternalStoreでmanagerの状態をReactコンポーネントと同期
+    // - 第1引数: 状態変更の監視を開始/停止する関数（subscribe）
+    // - 第2引数: 現在の状態を取得する関数（getSnapshot）
+    // managerの状態が変更されると自動的に再レンダリングされる
+    //
+    // - initialMessages の参照がレンダリングごとに変わると、subscribe の依存関係が変化し、
+    //   購読の再登録（retain/release）が発生して一時的に subCount=0 となる時間が生まれる。
+    //   その間のストリーミング更新は manager 側で通知が抑制され、UI が更新されないことがある。
+    // - これを避けるため、initialMessages は useRef で固定化し、subscribe の依存配列から外す。
+    // - また、initSession の即時通知を確実に受け取れるよう、retain → subscribe → init の順に実行する。
+    const state = useSyncExternalStore(useCallback((callback) => {
+        // セッション初期化とサブスクリプション管理を同期的に実行
+        // - レース回避のため retain を先に行い subCount>0 を担保
+        // - initSession の即時通知を受け取るため、subscribe 登録後に init を実行
         manager.retainSession(sessionId);
-        manager.initSession(sessionId, initialMessages);
+        const unsubscribe = manager.subscribe(callback);
+        manager.initSession(sessionId, initialMessagesRef.current);
         return () => {
-            // 購読解除
+            // 先に release して購読者数を即時に正しく反映
             manager.releaseSession(sessionId);
+            unsubscribe();
             // アンマウント時にハンドラは消さない（完了時に呼ぶため残す）
         };
-    }, [sessionId, manager, initialMessages]);
+    }, [manager, sessionId]), useCallback(() => manager.getSnapshot(), [manager]));
     // ハンドラが変わるたびに最新を登録
     useEffect(() => {
-        if (onFinish)
-            manager.setOnFinishHandler(sessionId, onFinish);
+        // onFinish が undefined の場合は登録解除される
+        manager.setOnFinishHandler(sessionId, onFinish);
     }, [manager, sessionId, onFinish]);
     useEffect(() => {
-        if (onMetadata)
-            manager.setOnMetadataHandler(sessionId, onMetadata);
+        // onMetadata が undefined の場合は登録解除される
+        manager.setOnMetadataHandler(sessionId, onMetadata);
     }, [manager, sessionId, onMetadata]);
     useEffect(() => {
-        if (onError)
-            manager.setOnErrorHandler(sessionId, onError);
+        // onError が undefined の場合は登録解除される
+        manager.setOnErrorHandler(sessionId, onError);
     }, [manager, sessionId, onError]);
-    // useSyncExternalStoreでmanagerの状態をReactコンポーネントと同期
-    // - 第1引数: 状態変更の監視を開始/停止する関数（subscribe）
-    // - 第2引数: 現在の状態を取得する関数（getSnapshot）
-    // managerの状態が変更されると自動的に再レンダリングされる
-    const state = useSyncExternalStore(useCallback((callback) => manager.subscribe(callback), [manager]), useCallback(() => manager.getSnapshot(), [manager]));
     const session = state.sessions.get(sessionId) || {
         id: sessionId,
         messages: initialMessages,
@@ -62,6 +79,11 @@ export function useChatSession(sessionId, options = {}) {
         if (lastUserIndex === -1)
             return;
         const messagesUntilLastUser = session.messages.slice(0, lastUserIndex + 1);
+        // キャッシュなどを防ぐため、最後のmessageのidを変える
+        messagesUntilLastUser[messagesUntilLastUser.length - 1] = {
+            ...messagesUntilLastUser[lastUserIndex],
+            id: generateId({ prefix: "msg-" }),
+        };
         await manager.streamChat(sessionId, messagesUntilLastUser, {
             api,
             body: params?.body,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "simple-ai-sdk",
-  "version": "1.0.11",
+  "version": "1.0.14",
   "private": false,
   "description": "Simple AI SDK for Hono / React19+ / OpenAI",
   "type": "module",
@@ -53,7 +53,7 @@
     "typecheck": "tsc --noEmit",
     "lint": "eslint .",
     "prepublish": "pnpm typecheck && pnpm lint",
-    "publish-sdk": "pnpm build && pnpm publish --no-git-checks --access public",
+    "publish-sdk": "npm version patch --no-git-tag-version && pnpm build && pnpm publish --no-git-checks --access public",
     "test": "echo \"Error: no test specified\" && exit 1"
   }
 }