@liveblocks/react 3.2.1 → 3.3.1

package/dist/_private.cjs

@@ -25,7 +25,7 @@
 
 
 
-var _chunkLJQILG65cjs = require('./chunk-LJQILG65.cjs');
+var _chunkV36VCKRBcjs = require('./chunk-V36VCKRB.cjs');
 
 // src/lib/use-layout-effect.ts
 var _react = require('react');
@@ -43,8 +43,8 @@ function normalizeMentionSuggestions(suggestions) {
 function useMentionSuggestions(roomId, search) {
   const [mentionSuggestions, setMentionSuggestions] = _react.useState.call(void 0, );
   const lastInvokedAt = _react.useRef.call(void 0, );
-  const resolveMentionSuggestions = _chunkLJQILG65cjs.useResolveMentionSuggestions.call(void 0, );
-  const mentionSuggestionsCache = _chunkLJQILG65cjs.useMentionSuggestionsCache.call(void 0, );
+  const resolveMentionSuggestions = _chunkV36VCKRBcjs.useResolveMentionSuggestions.call(void 0, );
+  const mentionSuggestionsCache = _chunkV36VCKRBcjs.useMentionSuggestionsCache.call(void 0, );
   _react.useEffect.call(void 0, () => {
     if (search === void 0 || !resolveMentionSuggestions) {
       return;
@@ -99,7 +99,7 @@ function useMentionSuggestions(roomId, search) {
 
 
 function useSyncSource() {
-  const client = _chunkLJQILG65cjs.useClient.call(void 0, );
+  const client = _chunkV36VCKRBcjs.useClient.call(void 0, );
   const createSyncSource = client[_core.kInternal].createSyncSource;
   const [syncSource, setSyncSource] = _react.useState.call(void 0, );
   _react.useEffect.call(void 0, () => {
@@ -138,5 +138,5 @@ function useSyncSource() {
 
 
 
-exports.getUmbrellaStoreForClient = _chunkLJQILG65cjs.getUmbrellaStoreForClient; exports.useAddRoomCommentReaction = _chunkLJQILG65cjs.useAddRoomCommentReaction; exports.useClientOrNull = _chunkLJQILG65cjs.useClientOrNull; exports.useCreateRoomComment = _chunkLJQILG65cjs.useCreateRoomComment; exports.useCreateRoomThread = _chunkLJQILG65cjs.useCreateRoomThread; exports.useCreateTextMention = _chunkLJQILG65cjs.useCreateTextMention; exports.useDeleteRoomComment = _chunkLJQILG65cjs.useDeleteRoomComment; exports.useDeleteRoomThread = _chunkLJQILG65cjs.useDeleteRoomThread; exports.useDeleteTextMention = _chunkLJQILG65cjs.useDeleteTextMention; exports.useEditRoomComment = _chunkLJQILG65cjs.useEditRoomComment; exports.useEditRoomThreadMetadata = _chunkLJQILG65cjs.useEditRoomThreadMetadata; exports.useLatest = _chunkLJQILG65cjs.useLatest; exports.useLayoutEffect = useLayoutEffect; exports.useMarkRoomThreadAsRead = _chunkLJQILG65cjs.useMarkRoomThreadAsRead; exports.useMarkRoomThreadAsResolved = _chunkLJQILG65cjs.useMarkRoomThreadAsResolved; exports.useMarkRoomThreadAsUnresolved = _chunkLJQILG65cjs.useMarkRoomThreadAsUnresolved; exports.useMentionSuggestions = useMentionSuggestions; exports.useMentionSuggestionsCache = _chunkLJQILG65cjs.useMentionSuggestionsCache; exports.useRemoveRoomCommentReaction = _chunkLJQILG65cjs.useRemoveRoomCommentReaction; exports.useReportTextEditor = _chunkLJQILG65cjs.useReportTextEditor; exports.useResolveMentionSuggestions = _chunkLJQILG65cjs.useResolveMentionSuggestions; exports.useRoomAttachmentUrl = _chunkLJQILG65cjs.useRoomAttachmentUrl; exports.useRoomPermissions = _chunkLJQILG65cjs.useRoomPermissions; exports.useRoomThreadSubscription = _chunkLJQILG65cjs.useRoomThreadSubscription; exports.useSignal = _chunkLJQILG65cjs.useSignal; exports.useSyncExternalStoreWithSelector = _chunkLJQILG65cjs.useSyncExternalStoreWithSelector; exports.useSyncSource = useSyncSource; exports.useYjsProvider = _chunkLJQILG65cjs.useYjsProvider;
+exports.getUmbrellaStoreForClient = _chunkV36VCKRBcjs.getUmbrellaStoreForClient; exports.useAddRoomCommentReaction = _chunkV36VCKRBcjs.useAddRoomCommentReaction; exports.useClientOrNull = _chunkV36VCKRBcjs.useClientOrNull; exports.useCreateRoomComment = _chunkV36VCKRBcjs.useCreateRoomComment; exports.useCreateRoomThread = _chunkV36VCKRBcjs.useCreateRoomThread; exports.useCreateTextMention = _chunkV36VCKRBcjs.useCreateTextMention; exports.useDeleteRoomComment = _chunkV36VCKRBcjs.useDeleteRoomComment; exports.useDeleteRoomThread = _chunkV36VCKRBcjs.useDeleteRoomThread; exports.useDeleteTextMention = _chunkV36VCKRBcjs.useDeleteTextMention; exports.useEditRoomComment = _chunkV36VCKRBcjs.useEditRoomComment; exports.useEditRoomThreadMetadata = _chunkV36VCKRBcjs.useEditRoomThreadMetadata; exports.useLatest = _chunkV36VCKRBcjs.useLatest; exports.useLayoutEffect = useLayoutEffect; exports.useMarkRoomThreadAsRead = _chunkV36VCKRBcjs.useMarkRoomThreadAsRead; exports.useMarkRoomThreadAsResolved = _chunkV36VCKRBcjs.useMarkRoomThreadAsResolved; exports.useMarkRoomThreadAsUnresolved = _chunkV36VCKRBcjs.useMarkRoomThreadAsUnresolved; exports.useMentionSuggestions = useMentionSuggestions; exports.useMentionSuggestionsCache = _chunkV36VCKRBcjs.useMentionSuggestionsCache; exports.useRemoveRoomCommentReaction = _chunkV36VCKRBcjs.useRemoveRoomCommentReaction; exports.useReportTextEditor = _chunkV36VCKRBcjs.useReportTextEditor; exports.useResolveMentionSuggestions = _chunkV36VCKRBcjs.useResolveMentionSuggestions; exports.useRoomAttachmentUrl = _chunkV36VCKRBcjs.useRoomAttachmentUrl; exports.useRoomPermissions = _chunkV36VCKRBcjs.useRoomPermissions; exports.useRoomThreadSubscription = _chunkV36VCKRBcjs.useRoomThreadSubscription; exports.useSignal = _chunkV36VCKRBcjs.useSignal; exports.useSyncExternalStoreWithSelector = _chunkV36VCKRBcjs.useSyncExternalStoreWithSelector; exports.useSyncSource = useSyncSource; exports.useYjsProvider = _chunkV36VCKRBcjs.useYjsProvider;
 //# sourceMappingURL=_private.cjs.map

package/dist/_private.d.cts

@@ -1,4 +1,4 @@
-export { g as getUmbrellaStoreForClient, a as useAddRoomCommentReaction, u as useClientOrNull, b as useCreateRoomComment, c as useCreateRoomThread, d as useCreateTextMention, e as useDeleteRoomComment, f as useDeleteRoomThread, h as useDeleteTextMention, i as useEditRoomComment, j as useEditRoomThreadMetadata, k as useMarkRoomThreadAsRead, l as useMarkRoomThreadAsResolved, m as useMarkRoomThreadAsUnresolved, n as useMentionSuggestionsCache, o as useRemoveRoomCommentReaction, p as useReportTextEditor, q as useResolveMentionSuggestions, r as useRoomAttachmentUrl, s as useRoomPermissions, t as useRoomThreadSubscription, v as useYjsProvider } from './room-DRYXmQT5.cjs';
+export { g as getUmbrellaStoreForClient, a as useAddRoomCommentReaction, u as useClientOrNull, b as useCreateRoomComment, c as useCreateRoomThread, d as useCreateTextMention, e as useDeleteRoomComment, f as useDeleteRoomThread, h as useDeleteTextMention, i as useEditRoomComment, j as useEditRoomThreadMetadata, k as useMarkRoomThreadAsRead, l as useMarkRoomThreadAsResolved, m as useMarkRoomThreadAsUnresolved, n as useMentionSuggestionsCache, o as useRemoveRoomCommentReaction, p as useReportTextEditor, q as useResolveMentionSuggestions, r as useRoomAttachmentUrl, s as useRoomPermissions, t as useRoomThreadSubscription, v as useYjsProvider } from './room-BE4TZf40.cjs';
 import { MutableRefObject, useEffect } from 'react';
 import { ISignal, SyncSource } from '@liveblocks/core';
 import '@liveblocks/client';

package/dist/_private.d.ts

@@ -1,4 +1,4 @@
-export { g as getUmbrellaStoreForClient, a as useAddRoomCommentReaction, u as useClientOrNull, b as useCreateRoomComment, c as useCreateRoomThread, d as useCreateTextMention, e as useDeleteRoomComment, f as useDeleteRoomThread, h as useDeleteTextMention, i as useEditRoomComment, j as useEditRoomThreadMetadata, k as useMarkRoomThreadAsRead, l as useMarkRoomThreadAsResolved, m as useMarkRoomThreadAsUnresolved, n as useMentionSuggestionsCache, o as useRemoveRoomCommentReaction, p as useReportTextEditor, q as useResolveMentionSuggestions, r as useRoomAttachmentUrl, s as useRoomPermissions, t as useRoomThreadSubscription, v as useYjsProvider } from './room-DRYXmQT5.js';
+export { g as getUmbrellaStoreForClient, a as useAddRoomCommentReaction, u as useClientOrNull, b as useCreateRoomComment, c as useCreateRoomThread, d as useCreateTextMention, e as useDeleteRoomComment, f as useDeleteRoomThread, h as useDeleteTextMention, i as useEditRoomComment, j as useEditRoomThreadMetadata, k as useMarkRoomThreadAsRead, l as useMarkRoomThreadAsResolved, m as useMarkRoomThreadAsUnresolved, n as useMentionSuggestionsCache, o as useRemoveRoomCommentReaction, p as useReportTextEditor, q as useResolveMentionSuggestions, r as useRoomAttachmentUrl, s as useRoomPermissions, t as useRoomThreadSubscription, v as useYjsProvider } from './room-BE4TZf40.js';
 import { MutableRefObject, useEffect } from 'react';
 import { ISignal, SyncSource } from '@liveblocks/core';
 import '@liveblocks/client';

package/dist/_private.js

@@ -25,7 +25,7 @@ import {
   useSignal,
   useSyncExternalStoreWithSelector,
   useYjsProvider
-} from "./chunk-YZRXMG3Z.js";
+} from "./chunk-UHR5FZDL.js";
 
 // src/lib/use-layout-effect.ts
 import { useEffect, useLayoutEffect as useOriginalLayoutEffect } from "react";

package/dist/{chunk-G4GBR6VV.js → chunk-45ZT2X3R.js}

@@ -1,6 +1,6 @@
 // src/version.ts
 var PKG_NAME = "@liveblocks/react";
-var PKG_VERSION = "3.2.1";
+var PKG_VERSION = "3.3.1";
 var PKG_FORMAT = "esm";
 
 // src/ClientSideSuspense.tsx
@@ -20,4 +20,4 @@ export {
   PKG_FORMAT,
   ClientSideSuspense
 };
-//# sourceMappingURL=chunk-G4GBR6VV.js.map
+//# sourceMappingURL=chunk-45ZT2X3R.js.map

package/dist/{chunk-VARKVHO4.cjs → chunk-S7XMPZC4.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/version.ts
 var PKG_NAME = "@liveblocks/react";
-var PKG_VERSION = "3.2.1";
+var PKG_VERSION = "3.3.1";
 var PKG_FORMAT = "cjs";
 
 // src/ClientSideSuspense.tsx
@@ -20,4 +20,4 @@ function ClientSideSuspense(props) {
 
 
 exports.PKG_NAME = PKG_NAME; exports.PKG_VERSION = PKG_VERSION; exports.PKG_FORMAT = PKG_FORMAT; exports.ClientSideSuspense = ClientSideSuspense;
-//# sourceMappingURL=chunk-VARKVHO4.cjs.map
+//# sourceMappingURL=chunk-S7XMPZC4.cjs.map

package/dist/{chunk-VARKVHO4.cjs.map → chunk-S7XMPZC4.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["/home/runner/work/liveblocks/liveblocks/packages/liveblocks-react/dist/chunk-VARKVHO4.cjs","../src/version.ts","../src/ClientSideSuspense.tsx"],"names":[],"mappings":"AAAA;ACGO,IAAM,SAAA,EAAW,mBAAA;AACjB,IAAM,YAAA,EAAiD,OAAA;AACvD,IAAM,WAAA,EAAgD,KAAA;ADD7D;AACA;AEJA,8BAA8C;AAkC1C,+CAAA;AAVG,SAAS,kBAAA,CAAmB,KAAA,EAAc;AAC/C,EAAA,MAAM,CAAC,OAAA,EAAS,UAAU,EAAA,EAAI,6BAAA,KAAc,CAAA;AAE5C,EAAA,8BAAA,CAAU,EAAA,GAAM;AAGd,IAAA,UAAA,CAAW,IAAI,CAAA;AAAA,EACjB,CAAA,EAAG,CAAC,CAAC,CAAA;AAEL,EAAA,uBACE,6BAAA,eAAC,EAAA,EAAS,QAAA,EAAU,KAAA,CAAM,QAAA,EACvB,QAAA,EAAA,QAAA,EACG,OAAO,KAAA,CAAM,SAAA,IAAa,WAAA,EACxB,KAAA,CAAM,QAAA,CAAS,EAAA,EACf,KAAA,CAAM,SAAA,EACR,KAAA,CAAM,SAAA,CACZ,CAAA;AAEJ;AF5BA;AACA;AACE;AACA;AACA;AACA;AACF,iJAAC","file":"/home/runner/work/liveblocks/liveblocks/packages/liveblocks-react/dist/chunk-VARKVHO4.cjs","sourcesContent":[null,"declare const __VERSION__: string;\ndeclare const TSUP_FORMAT: string;\n\nexport const PKG_NAME = \"@liveblocks/react\";\nexport const PKG_VERSION = typeof __VERSION__ === \"string\" && __VERSION__;\nexport const PKG_FORMAT = typeof TSUP_FORMAT === \"string\" && TSUP_FORMAT;\n","import type { ReactNode } from \"react\";\nimport { Suspense, useEffect, useState } from \"react\";\n\ntype Props = {\n  fallback: ReactNode;\n  children: (() => ReactNode | undefined) | ReactNode | undefined;\n};\n\n/**\n * Almost like a normal <Suspense> component, except that for server-side\n * renders, the fallback will be used.\n *\n * The child props will have to be provided in a function, i.e. change:\n *\n *   <Suspense fallback={<Loading />}>\n *     <MyRealComponent a={1} />\n *   </Suspense>\n *\n * To:\n *\n *   <ClientSideSuspense fallback={<Loading />}>\n *     <MyRealComponent a={1} />\n *   </ClientSideSuspense>\n *\n */\nexport function ClientSideSuspense(props: Props) {\n  const [mounted, setMounted] = useState(false);\n\n  useEffect(() => {\n    // Effects are never executed on the server side. The point of this is to\n    // delay the flipping of this boolean until after hydration has happened.\n    setMounted(true);\n  }, []);\n\n  return (\n    <Suspense fallback={props.fallback}>\n      {mounted\n        ? typeof props.children === \"function\"\n          ? props.children()\n          : props.children\n        : props.fallback}\n    </Suspense>\n  );\n}\n"]}
+{"version":3,"sources":["/home/runner/work/liveblocks/liveblocks/packages/liveblocks-react/dist/chunk-S7XMPZC4.cjs","../src/version.ts","../src/ClientSideSuspense.tsx"],"names":[],"mappings":"AAAA;ACGO,IAAM,SAAA,EAAW,mBAAA;AACjB,IAAM,YAAA,EAAiD,OAAA;AACvD,IAAM,WAAA,EAAgD,KAAA;ADD7D;AACA;AEJA,8BAA8C;AAkC1C,+CAAA;AAVG,SAAS,kBAAA,CAAmB,KAAA,EAAc;AAC/C,EAAA,MAAM,CAAC,OAAA,EAAS,UAAU,EAAA,EAAI,6BAAA,KAAc,CAAA;AAE5C,EAAA,8BAAA,CAAU,EAAA,GAAM;AAGd,IAAA,UAAA,CAAW,IAAI,CAAA;AAAA,EACjB,CAAA,EAAG,CAAC,CAAC,CAAA;AAEL,EAAA,uBACE,6BAAA,eAAC,EAAA,EAAS,QAAA,EAAU,KAAA,CAAM,QAAA,EACvB,QAAA,EAAA,QAAA,EACG,OAAO,KAAA,CAAM,SAAA,IAAa,WAAA,EACxB,KAAA,CAAM,QAAA,CAAS,EAAA,EACf,KAAA,CAAM,SAAA,EACR,KAAA,CAAM,SAAA,CACZ,CAAA;AAEJ;AF5BA;AACA;AACE;AACA;AACA;AACA;AACF,iJAAC","file":"/home/runner/work/liveblocks/liveblocks/packages/liveblocks-react/dist/chunk-S7XMPZC4.cjs","sourcesContent":[null,"declare const __VERSION__: string;\ndeclare const TSUP_FORMAT: string;\n\nexport const PKG_NAME = \"@liveblocks/react\";\nexport const PKG_VERSION = typeof __VERSION__ === \"string\" && __VERSION__;\nexport const PKG_FORMAT = typeof TSUP_FORMAT === \"string\" && TSUP_FORMAT;\n","import type { ReactNode } from \"react\";\nimport { Suspense, useEffect, useState } from \"react\";\n\ntype Props = {\n  fallback: ReactNode;\n  children: (() => ReactNode | undefined) | ReactNode | undefined;\n};\n\n/**\n * Almost like a normal <Suspense> component, except that for server-side\n * renders, the fallback will be used.\n *\n * The child props will have to be provided in a function, i.e. change:\n *\n *   <Suspense fallback={<Loading />}>\n *     <MyRealComponent a={1} />\n *   </Suspense>\n *\n * To:\n *\n *   <ClientSideSuspense fallback={<Loading />}>\n *     <MyRealComponent a={1} />\n *   </ClientSideSuspense>\n *\n */\nexport function ClientSideSuspense(props: Props) {\n  const [mounted, setMounted] = useState(false);\n\n  useEffect(() => {\n    // Effects are never executed on the server side. The point of this is to\n    // delay the flipping of this boolean until after hydration has happened.\n    setMounted(true);\n  }, []);\n\n  return (\n    <Suspense fallback={props.fallback}>\n      {mounted\n        ? typeof props.children === \"function\"\n          ? props.children()\n          : props.children\n        : props.fallback}\n    </Suspense>\n  );\n}\n"]}

package/dist/{chunk-YZRXMG3Z.js → chunk-UHR5FZDL.js}

@@ -258,31 +258,30 @@ function useInitialUnlessFunction(latestValue) {
 }
 
 // src/lib/use-polyfill.ts
-var use = (
-  // React.use ||
-  (promise) => {
-    if (promise.status === "pending") {
-      throw promise;
-    } else if (promise.status === "fulfilled") {
-      return promise.value;
-    } else if (promise.status === "rejected") {
-      throw promise.reason;
-    } else {
-      promise.status = "pending";
-      promise.then(
-        (v) => {
-          promise.status = "fulfilled";
-          promise.value = v;
-        },
-        (e) => {
-          promise.status = "rejected";
-          promise.reason = e;
-        }
-      );
-      throw promise;
-    }
+import * as React from "react";
+var reactUse = React[" use ".trim().toString()];
+var use = reactUse ?? ((promise) => {
+  if (promise.status === "pending") {
+    throw promise;
+  } else if (promise.status === "fulfilled") {
+    return promise.value;
+  } else if (promise.status === "rejected") {
+    throw promise.reason;
+  } else {
+    promise.status = "pending";
+    promise.then(
+      (v) => {
+        promise.status = "fulfilled";
+        promise.value = v;
+      },
+      (e) => {
+        promise.status = "rejected";
+        promise.reason = e;
+      }
+    );
+    throw promise;
   }
-);
+});
 
 // src/umbrella-store.ts
 import {
@@ -488,6 +487,9 @@ function makeRoomThreadsQueryKey(roomId, query) {
 function makeUserThreadsQueryKey(query) {
   return stableStringify(query ?? {});
 }
+function makeAiChatsQueryKey(query) {
+  return stableStringify(query ?? {});
+}
 function usify(promise) {
   if ("status" in promise) {
     return promise;
@@ -925,8 +927,6 @@ var UmbrellaStore = class {
   #roomVersionsLastRequestedAtByRoom = /* @__PURE__ */ new Map();
   // Notification Settings
   #notificationSettings;
-  // Copilot chats
-  #aiChats;
   constructor(client) {
     this.#client = client[kInternal2].as();
     this.optimisticUpdates = createStore_forOptimistic(this.#client);
@@ -956,12 +956,6 @@ var UmbrellaStore = class {
     this.#notificationSettings = new SinglePageResource(
       notificationSettingsFetcher
     );
-    this.#aiChats = new PaginatedResource(async (cursor) => {
-      const result = await this.#client[kInternal2].ai.getChats({
-        cursor
-      });
-      return result.nextCursor;
-    });
     this.threads = new ThreadDB();
     this.subscriptions = createStore_forSubscriptions(
       this.optimisticUpdates.signal,
@@ -1168,23 +1162,34 @@ var UmbrellaStore = class {
       }, shallow),
       waitUntilLoaded: this.#notificationSettings.waitUntilLoaded
     };
-    const aiChats = {
-      signal: DerivedSignal.from(() => {
-        const result = this.#aiChats.get();
-        if (result.isLoading || result.error) {
-          return result;
-        }
-        return {
-          isLoading: false,
-          chats: this.#client[kInternal2].ai.signals.chats\u03A3.get(),
-          hasFetchedAll: result.data.hasFetchedAll,
-          isFetchingMore: result.data.isFetchingMore,
-          fetchMore: result.data.fetchMore,
-          fetchMoreError: result.data.fetchMoreError
-        };
-      }, shallow),
-      waitUntilLoaded: this.#aiChats.waitUntilLoaded
-    };
+    const aiChats = new DefaultMap(
+      (queryKey) => {
+        const query = JSON.parse(queryKey);
+        const resource = new PaginatedResource(async (cursor) => {
+          const result = await this.#client[kInternal2].ai.getChats({
+            cursor,
+            query
+          });
+          return result.nextCursor;
+        });
+        const signal = DerivedSignal.from(() => {
+          const result = resource.get();
+          if (result.isLoading || result.error) {
+            return result;
+          }
+          const chats = this.#client[kInternal2].ai.queryChats(query);
+          return {
+            isLoading: false,
+            chats,
+            hasFetchedAll: result.data.hasFetchedAll,
+            isFetchingMore: result.data.isFetchingMore,
+            fetchMore: result.data.fetchMore,
+            fetchMoreError: result.data.fetchMoreError
+          };
+        }, shallow);
+        return { signal, waitUntilLoaded: resource.waitUntilLoaded };
+      }
+    );
     const messagesByChatId = new DefaultMap((chatId) => {
       const resource\u03A3 = new SinglePageResource(async () => {
         await this.#client[kInternal2].ai.getMessageTree(chatId);
@@ -2526,12 +2531,13 @@ function useRoomInfoSuspense_withClient(client, roomId) {
     error: void 0
   };
 }
-function useAiChats() {
+function useAiChats(options) {
   const client = useClient();
   const store = getUmbrellaStoreForClient(client);
+  const queryKey = makeAiChatsQueryKey(options?.query);
   useEnsureAiConnection(client);
   useEffect4(
-    () => void store.outputs.aiChats.waitUntilLoaded()
+    () => void store.outputs.aiChats.getOrCreate(queryKey).waitUntilLoaded()
     // NOTE: Deliberately *not* using a dependency array here!
     //
     // It is important to call waitUntil on *every* render.
@@ -2541,15 +2547,16 @@ function useAiChats() {
     // 3. If ever the promise would fail, then after 5 seconds it would reset, and on the very
     //    *next* render after that, a *new* fetch/promise will get created.
   );
-  return useSignal(store.outputs.aiChats.signal, identity2, shallow3);
+  return useSignal(store.outputs.aiChats.getOrCreate(queryKey).signal, identity2, shallow3);
 }
-function useAiChatsSuspense() {
+function useAiChatsSuspense(options) {
   ensureNotServerSide();
   const client = useClient();
   const store = getUmbrellaStoreForClient(client);
   useEnsureAiConnection(client);
-  use(store.outputs.aiChats.waitUntilLoaded());
-  const result = useAiChats();
+  const queryKey = makeAiChatsQueryKey(options?.query);
+  use(store.outputs.aiChats.getOrCreate(queryKey).waitUntilLoaded());
+  const result = useAiChats(options);
   assert(!result.error, "Did not expect error");
   assert(!result.isLoading, "Did not expect loading");
   return result;
@@ -2618,6 +2625,9 @@ function useCreateAiChat() {
   const client = useClient();
   return useCallback2(
     (options) => {
+      if (typeof options === "string") {
+        options = { id: options };
+      }
       client[kInternal3].ai.getOrCreateChat(options.id, {
         title: options.title,
         metadata: options.metadata
@@ -2645,35 +2655,54 @@ function useDeleteAiChat() {
 }
 function useSendAiMessage(chatId, options) {
   const client = useClient();
-  const copilotId = options?.copilotId;
   return useCallback2(
     (message) => {
-      const messages = client[kInternal3].ai.signals.getChatMessagesForBranch\u03A3(chatId).get();
+      const {
+        text: messageText,
+        chatId: messageOptionsChatId,
+        ...messageOptions
+      } = typeof message === "string" ? { text: message } : message;
+      const resolvedChatId = messageOptionsChatId ?? chatId ?? // The `useSendAiMessage` overloads prevent this scenario from happening
+      // at the type level, and this error prevents it from happening at runtime.
+      raise2(
+        "chatId must be provided to either `useSendAiMessage` or its returned function."
+      );
+      const messages = client[kInternal3].ai.signals.getChatMessagesForBranch\u03A3(resolvedChatId).get();
       const lastMessageId = messages[messages.length - 1]?.id ?? null;
-      const content = [{ type: "text", text: message }];
+      const content = [{ type: "text", text: messageText }];
       const newMessageId = client[kInternal3].ai[kInternal3].context.messagesStore.createOptimistically(
-        chatId,
+        resolvedChatId,
         "user",
         lastMessageId,
         content
       );
+      const newMessage = client[kInternal3].ai[kInternal3].context.messagesStore.getMessageById(newMessageId);
       const targetMessageId = client[kInternal3].ai[kInternal3].context.messagesStore.createOptimistically(
-        chatId,
+        resolvedChatId,
         "assistant",
         newMessageId
       );
       void client[kInternal3].ai.askUserMessageInChat(
-        chatId,
+        resolvedChatId,
         { id: newMessageId, parentMessageId: lastMessageId, content },
         targetMessageId,
         {
-          stream: options?.stream,
-          copilotId,
-          timeout: options?.timeout
+          stream: messageOptions.stream ?? options?.stream,
+          copilotId: messageOptions.copilotId ?? options?.copilotId,
+          timeout: messageOptions.timeout ?? options?.timeout,
+          knowledge: messageOptions.knowledge ?? options?.knowledge
         }
       );
+      return newMessage;
     },
-    [client, chatId, copilotId, options?.stream, options?.timeout]
+    [
+      client,
+      chatId,
+      options?.copilotId,
+      options?.stream,
+      options?.timeout,
+      options?.knowledge
+    ]
   );
 }
 function createSharedContext(client) {
@@ -4639,4 +4668,4 @@ export {
   _useStorageRoot,
   _useUpdateMyPresence
 };
-//# sourceMappingURL=chunk-YZRXMG3Z.js.map
+//# sourceMappingURL=chunk-UHR5FZDL.js.map