@btst/stack 3.0.0-rc.1 → 3.0.0-rc.3

package/dist/packages/stack/src/plugins/ai-chat/client/components/tool-call-display.mjs

@@ -50,7 +50,7 @@ function ToolCallDisplay({
   errorText,
   isLoading
 }) {
-  const { localization } = usePluginOverrides("ai-chat", {});
+  const { localization } = usePluginOverrides("aiChat", {});
   const tr = useAiChatTranslation(localization);
   const displayName = formatToolName(toolName);
   const statusLabel = isLoading || state === "input-streaming" ? tr("TOOL_STATUS_RUNNING", "aiChat.tools.status.running", "Running...") : state === "input-available" ? tr(

package/dist/packages/stack/src/plugins/ai-chat/client/hooks/ai-chat-resource.cjs

@@ -5,7 +5,7 @@ const hooks = require('@btst/stack/plugins/client/hooks');
 const plugins_aiChat_queryKeys = require('../../../../../../../plugins/ai-chat/query-keys.cjs');
 
 const aiChat = hooks.createResource({
-  plugin: "ai-chat",
+  plugin: "aiChat",
   resources: plugins_aiChat_queryKeys.aiChatResources
 });
 

package/dist/packages/stack/src/plugins/ai-chat/client/hooks/ai-chat-resource.mjs

@@ -3,7 +3,7 @@ import { createResource } from '@btst/stack/plugins/client/hooks';
 import { aiChatResources } from '../../../../../../../plugins/ai-chat/query-keys.mjs';
 
 const aiChat = createResource({
-  plugin: "ai-chat",
+  plugin: "aiChat",
   resources: aiChatResources
 });
 

package/dist/packages/stack/src/plugins/ai-chat/client/hooks/chat-hooks.cjs

@@ -1,58 +1,210 @@
 "use client";
 'use strict';
 
+const reactQuery = require('@tanstack/react-query');
+const React = require('react');
+const client = require('@btst/stack/plugins/client');
 const context = require('@btst/stack/context');
+const plugins_aiChat_queryKeys = require('../../../../../../../plugins/ai-chat/query-keys.cjs');
 const aiChatResource = require('./ai-chat-resource.cjs');
 
+function useAiChatIdentityState() {
+  const { identity, isPending, error, refetch } = context.useIdentity();
+  const sourceGeneration = context.useIdentitySourceGeneration();
+  return {
+    partition: isPending ? `pending:${sourceGeneration}` : error ? `error:${sourceGeneration}` : identity ? identity : "anonymous",
+    isPending,
+    ...error ? { error } : {},
+    refetchIdentity: refetch
+  };
+}
+function useAiChatIdentityPartition() {
+  return useAiChatIdentityState().partition;
+}
+function isUnresolvedIdentityPartition(partition) {
+  return typeof partition === "string" && (partition.startsWith("pending:") || partition.startsWith("error:"));
+}
+function samePartition(left, right) {
+  return reactQuery.hashKey([plugins_aiChat_queryKeys.aiChatIdentityKey(left)]) === reactQuery.hashKey([plugins_aiChat_queryKeys.aiChatIdentityKey(right)]);
+}
+function queryBelongsToPartition(queryKey, partition) {
+  const marker = queryKey[3];
+  if (partition === void 0) return marker === void 0;
+  return Boolean(
+    marker && reactQuery.hashKey([marker.identity]) === reactQuery.hashKey([plugins_aiChat_queryKeys.aiChatIdentityKey(partition)])
+  );
+}
+function useCurrentHistoryRefresh() {
+  const { queryClient: stackQueryClient } = context.useStack();
+  const queryClient = reactQuery.useQueryClient(stackQueryClient);
+  const { partition: identityPartition } = useAiChatIdentityState();
+  const latestPartition = React.useRef(identityPartition);
+  const mounted = React.useRef(true);
+  React.useLayoutEffect(() => {
+    latestPartition.current = identityPartition;
+  }, [identityPartition]);
+  React.useLayoutEffect(() => {
+    mounted.current = true;
+    return () => {
+      mounted.current = false;
+    };
+  }, []);
+  const refreshAfterSuccess = async (startedAs, queryKey = ["conversations"]) => {
+    const current = latestPartition.current;
+    if (!mounted.current || !samePartition(startedAs, current)) {
+      queryClient.removeQueries({
+        queryKey: ["conversations"],
+        predicate: ({ queryKey: queryKey2 }) => queryBelongsToPartition(queryKey2, startedAs)
+      });
+      return;
+    }
+    await queryClient.invalidateQueries({
+      queryKey,
+      predicate: ({ queryKey: queryKey2 }) => queryBelongsToPartition(queryKey2, current),
+      refetchType: "all"
+    });
+  };
+  const removeConversationDetail = (id, partition) => {
+    queryClient.removeQueries({
+      queryKey: client.buildQueryKey(
+        "conversations",
+        "detail",
+        plugins_aiChat_queryKeys.aiChatResources.conversations.queries.detail,
+        [id, partition]
+      ),
+      exact: true
+    });
+  };
+  return {
+    currentPartition: () => identityPartition,
+    refreshAfterSuccess,
+    removeConversationDetail
+  };
+}
+function withCurrentHistoryRefresh(mutation, refresh, queryKey, afterSuccess) {
+  const mutateAsync = async (variables, options) => {
+    const startedAs = refresh.currentPartition();
+    const result = await mutation.mutateAsync(variables, options);
+    try {
+      await refresh.refreshAfterSuccess(startedAs, queryKey);
+    } finally {
+      afterSuccess?.(variables, startedAs);
+    }
+    return result;
+  };
+  const mutate = (variables, options) => {
+    void mutateAsync(variables, options).catch(() => {
+    });
+  };
+  return { ...mutation, mutate, mutateAsync };
+}
 function useConversations(options = {}) {
-  const query = aiChatResource.aiChat.conversations.list.use([], {
-    enabled: options.enabled ?? true
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const query = aiChatResource.aiChat.conversations.list.use([identityPartition], {
+    enabled: (options.enabled ?? true) && !isUnresolvedIdentityPartition(identityPartition)
   });
   return {
     conversations: query.data ?? [],
-    isLoading: query.isLoading,
-    error: query.error,
-    refetch: query.refetch
+    isLoading: isIdentityPending || query.isLoading,
+    error: identityError ?? query.error,
+    refetch: () => {
+      if (isUnresolvedIdentityPartition(identityPartition)) {
+        void refetchIdentity();
+        return;
+      }
+      void query.refetch();
+    }
   };
 }
 function useSuspenseConversations() {
-  const query = aiChatResource.aiChat.conversations.list.useSuspense([]);
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const identityResolution = context.useIdentityResolutionPromise();
+  const query = aiChatResource.aiChat.conversations.list.useSuspense([identityPartition]);
+  if (identityError) throw identityError;
+  if (isIdentityPending) throw identityResolution ?? Promise.resolve();
   return {
     conversations: query.data ?? [],
-    refetch: query.refetch
+    refetch: isUnresolvedIdentityPartition(identityPartition) ? refetchIdentity : query.refetch
   };
 }
 function useConversation(id, options = {}) {
-  const query = aiChatResource.aiChat.conversations.detail.use([id ?? ""], {
-    enabled: (options.enabled ?? true) && !!id
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const query = aiChatResource.aiChat.conversations.detail.use([id ?? "", identityPartition], {
+    enabled: (options.enabled ?? true) && !!id && !isUnresolvedIdentityPartition(identityPartition)
   });
   return {
     conversation: query.data ?? null,
-    isLoading: query.isLoading,
-    error: query.error,
-    refetch: query.refetch
+    isLoading: isIdentityPending || query.isLoading,
+    error: identityError ?? query.error,
+    refetch: () => {
+      if (isUnresolvedIdentityPartition(identityPartition)) {
+        void refetchIdentity();
+        return;
+      }
+      void query.refetch();
+    }
   };
 }
 function useSuspenseConversation(id) {
-  const query = aiChatResource.aiChat.conversations.detail.useSuspense([id]);
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const identityResolution = context.useIdentityResolutionPromise();
+  const query = aiChatResource.aiChat.conversations.detail.useSuspense([
+    id,
+    identityPartition
+  ]);
+  if (identityError) throw identityError;
+  if (isIdentityPending) throw identityResolution ?? Promise.resolve();
   return {
     conversation: query.data ?? null,
-    refetch: query.refetch
+    refetch: isUnresolvedIdentityPartition(identityPartition) ? refetchIdentity : query.refetch
   };
 }
 function useCreateConversation() {
-  return aiChatResource.aiChat.conversations.create.use();
+  return withCurrentHistoryRefresh(
+    aiChatResource.aiChat.conversations.create.use(),
+    useCurrentHistoryRefresh()
+  );
 }
 function useRenameConversation() {
-  return aiChatResource.aiChat.conversations.rename.use();
+  return withCurrentHistoryRefresh(
+    aiChatResource.aiChat.conversations.rename.use(),
+    useCurrentHistoryRefresh()
+  );
 }
 function useDeleteConversation() {
-  return aiChatResource.aiChat.conversations.delete.use();
+  const historyRefresh = useCurrentHistoryRefresh();
+  return withCurrentHistoryRefresh(
+    aiChatResource.aiChat.conversations.delete.use(),
+    historyRefresh,
+    ["conversations", "list"],
+    ({ id }, startedAs) => historyRefresh.removeConversationDetail(id, startedAs)
+  );
 }
 function useRenameConversationForm(options) {
   const t = context.useTranslate();
-  const { localization } = context.usePluginOverrides("ai-chat");
-  return aiChatResource.aiChat.conversations.useForm({
+  const { localization } = context.usePluginOverrides("aiChat");
+  const historyRefresh = useCurrentHistoryRefresh();
+  const form = aiChatResource.aiChat.conversations.useForm({
     action: "edit",
     updateMutation: "rename",
     record: options.conversation,
@@ -69,8 +221,16 @@ function useRenameConversationForm(options) {
     errorMessage: () => localization?.CONVERSATION_RENAME_FAILURE ?? t("aiChat.toasts.renameFailure", "Failed to rename conversation"),
     onSuccess: options.onSuccess
   });
+  const submit = async (values) => {
+    const startedAs = historyRefresh.currentPartition();
+    const result = await form.submit(values);
+    await historyRefresh.refreshAfterSuccess(startedAs);
+    return result;
+  };
+  return { ...form, submit };
 }
 
+exports.useAiChatIdentityPartition = useAiChatIdentityPartition;
 exports.useConversation = useConversation;
 exports.useConversations = useConversations;
 exports.useCreateConversation = useCreateConversation;

package/dist/packages/stack/src/plugins/ai-chat/client/hooks/chat-hooks.mjs

@@ -1,56 +1,208 @@
 "use client";
-import { useTranslate, usePluginOverrides } from '@btst/stack/context';
+import { useQueryClient, hashKey } from '@tanstack/react-query';
+import { useRef, useLayoutEffect } from 'react';
+import { buildQueryKey } from '@btst/stack/plugins/client';
+import { useIdentityResolutionPromise, useTranslate, usePluginOverrides, useIdentity, useIdentitySourceGeneration, useStack } from '@btst/stack/context';
+import { aiChatResources, aiChatIdentityKey } from '../../../../../../../plugins/ai-chat/query-keys.mjs';
 import { aiChat } from './ai-chat-resource.mjs';
 
+function useAiChatIdentityState() {
+  const { identity, isPending, error, refetch } = useIdentity();
+  const sourceGeneration = useIdentitySourceGeneration();
+  return {
+    partition: isPending ? `pending:${sourceGeneration}` : error ? `error:${sourceGeneration}` : identity ? identity : "anonymous",
+    isPending,
+    ...error ? { error } : {},
+    refetchIdentity: refetch
+  };
+}
+function useAiChatIdentityPartition() {
+  return useAiChatIdentityState().partition;
+}
+function isUnresolvedIdentityPartition(partition) {
+  return typeof partition === "string" && (partition.startsWith("pending:") || partition.startsWith("error:"));
+}
+function samePartition(left, right) {
+  return hashKey([aiChatIdentityKey(left)]) === hashKey([aiChatIdentityKey(right)]);
+}
+function queryBelongsToPartition(queryKey, partition) {
+  const marker = queryKey[3];
+  if (partition === void 0) return marker === void 0;
+  return Boolean(
+    marker && hashKey([marker.identity]) === hashKey([aiChatIdentityKey(partition)])
+  );
+}
+function useCurrentHistoryRefresh() {
+  const { queryClient: stackQueryClient } = useStack();
+  const queryClient = useQueryClient(stackQueryClient);
+  const { partition: identityPartition } = useAiChatIdentityState();
+  const latestPartition = useRef(identityPartition);
+  const mounted = useRef(true);
+  useLayoutEffect(() => {
+    latestPartition.current = identityPartition;
+  }, [identityPartition]);
+  useLayoutEffect(() => {
+    mounted.current = true;
+    return () => {
+      mounted.current = false;
+    };
+  }, []);
+  const refreshAfterSuccess = async (startedAs, queryKey = ["conversations"]) => {
+    const current = latestPartition.current;
+    if (!mounted.current || !samePartition(startedAs, current)) {
+      queryClient.removeQueries({
+        queryKey: ["conversations"],
+        predicate: ({ queryKey: queryKey2 }) => queryBelongsToPartition(queryKey2, startedAs)
+      });
+      return;
+    }
+    await queryClient.invalidateQueries({
+      queryKey,
+      predicate: ({ queryKey: queryKey2 }) => queryBelongsToPartition(queryKey2, current),
+      refetchType: "all"
+    });
+  };
+  const removeConversationDetail = (id, partition) => {
+    queryClient.removeQueries({
+      queryKey: buildQueryKey(
+        "conversations",
+        "detail",
+        aiChatResources.conversations.queries.detail,
+        [id, partition]
+      ),
+      exact: true
+    });
+  };
+  return {
+    currentPartition: () => identityPartition,
+    refreshAfterSuccess,
+    removeConversationDetail
+  };
+}
+function withCurrentHistoryRefresh(mutation, refresh, queryKey, afterSuccess) {
+  const mutateAsync = async (variables, options) => {
+    const startedAs = refresh.currentPartition();
+    const result = await mutation.mutateAsync(variables, options);
+    try {
+      await refresh.refreshAfterSuccess(startedAs, queryKey);
+    } finally {
+      afterSuccess?.(variables, startedAs);
+    }
+    return result;
+  };
+  const mutate = (variables, options) => {
+    void mutateAsync(variables, options).catch(() => {
+    });
+  };
+  return { ...mutation, mutate, mutateAsync };
+}
 function useConversations(options = {}) {
-  const query = aiChat.conversations.list.use([], {
-    enabled: options.enabled ?? true
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const query = aiChat.conversations.list.use([identityPartition], {
+    enabled: (options.enabled ?? true) && !isUnresolvedIdentityPartition(identityPartition)
   });
   return {
     conversations: query.data ?? [],
-    isLoading: query.isLoading,
-    error: query.error,
-    refetch: query.refetch
+    isLoading: isIdentityPending || query.isLoading,
+    error: identityError ?? query.error,
+    refetch: () => {
+      if (isUnresolvedIdentityPartition(identityPartition)) {
+        void refetchIdentity();
+        return;
+      }
+      void query.refetch();
+    }
   };
 }
 function useSuspenseConversations() {
-  const query = aiChat.conversations.list.useSuspense([]);
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const identityResolution = useIdentityResolutionPromise();
+  const query = aiChat.conversations.list.useSuspense([identityPartition]);
+  if (identityError) throw identityError;
+  if (isIdentityPending) throw identityResolution ?? Promise.resolve();
   return {
     conversations: query.data ?? [],
-    refetch: query.refetch
+    refetch: isUnresolvedIdentityPartition(identityPartition) ? refetchIdentity : query.refetch
   };
 }
 function useConversation(id, options = {}) {
-  const query = aiChat.conversations.detail.use([id ?? ""], {
-    enabled: (options.enabled ?? true) && !!id
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const query = aiChat.conversations.detail.use([id ?? "", identityPartition], {
+    enabled: (options.enabled ?? true) && !!id && !isUnresolvedIdentityPartition(identityPartition)
   });
   return {
     conversation: query.data ?? null,
-    isLoading: query.isLoading,
-    error: query.error,
-    refetch: query.refetch
+    isLoading: isIdentityPending || query.isLoading,
+    error: identityError ?? query.error,
+    refetch: () => {
+      if (isUnresolvedIdentityPartition(identityPartition)) {
+        void refetchIdentity();
+        return;
+      }
+      void query.refetch();
+    }
   };
 }
 function useSuspenseConversation(id) {
-  const query = aiChat.conversations.detail.useSuspense([id]);
+  const {
+    partition: identityPartition,
+    isPending: isIdentityPending,
+    error: identityError,
+    refetchIdentity
+  } = useAiChatIdentityState();
+  const identityResolution = useIdentityResolutionPromise();
+  const query = aiChat.conversations.detail.useSuspense([
+    id,
+    identityPartition
+  ]);
+  if (identityError) throw identityError;
+  if (isIdentityPending) throw identityResolution ?? Promise.resolve();
   return {
     conversation: query.data ?? null,
-    refetch: query.refetch
+    refetch: isUnresolvedIdentityPartition(identityPartition) ? refetchIdentity : query.refetch
   };
 }
 function useCreateConversation() {
-  return aiChat.conversations.create.use();
+  return withCurrentHistoryRefresh(
+    aiChat.conversations.create.use(),
+    useCurrentHistoryRefresh()
+  );
 }
 function useRenameConversation() {
-  return aiChat.conversations.rename.use();
+  return withCurrentHistoryRefresh(
+    aiChat.conversations.rename.use(),
+    useCurrentHistoryRefresh()
+  );
 }
 function useDeleteConversation() {
-  return aiChat.conversations.delete.use();
+  const historyRefresh = useCurrentHistoryRefresh();
+  return withCurrentHistoryRefresh(
+    aiChat.conversations.delete.use(),
+    historyRefresh,
+    ["conversations", "list"],
+    ({ id }, startedAs) => historyRefresh.removeConversationDetail(id, startedAs)
+  );
 }
 function useRenameConversationForm(options) {
   const t = useTranslate();
-  const { localization } = usePluginOverrides("ai-chat");
-  return aiChat.conversations.useForm({
+  const { localization } = usePluginOverrides("aiChat");
+  const historyRefresh = useCurrentHistoryRefresh();
+  const form = aiChat.conversations.useForm({
     action: "edit",
     updateMutation: "rename",
     record: options.conversation,
@@ -67,6 +219,13 @@ function useRenameConversationForm(options) {
     errorMessage: () => localization?.CONVERSATION_RENAME_FAILURE ?? t("aiChat.toasts.renameFailure", "Failed to rename conversation"),
     onSuccess: options.onSuccess
   });
+  const submit = async (values) => {
+    const startedAs = historyRefresh.currentPartition();
+    const result = await form.submit(values);
+    await historyRefresh.refreshAfterSuccess(startedAs);
+    return result;
+  };
+  return { ...form, submit };
 }
 
-export { useConversation, useConversations, useCreateConversation, useDeleteConversation, useRenameConversation, useRenameConversationForm, useSuspenseConversation, useSuspenseConversations };
+export { useAiChatIdentityPartition, useConversation, useConversations, useCreateConversation, useDeleteConversation, useRenameConversation, useRenameConversationForm, useSuspenseConversation, useSuspenseConversations };

package/dist/packages/stack/src/plugins/ai-chat/client/navigation.cjs

@@ -0,0 +1,11 @@
+'use strict';
+
+function navigateAiChatCrossOrigin(href, options = {}) {
+  if (options.replace) {
+    window.location.replace(href);
+    return;
+  }
+  window.location.assign(href);
+}
+
+exports.navigateAiChatCrossOrigin = navigateAiChatCrossOrigin;

package/dist/packages/stack/src/plugins/ai-chat/client/navigation.mjs

@@ -0,0 +1,9 @@
+function navigateAiChatCrossOrigin(href, options = {}) {
+  if (options.replace) {
+    window.location.replace(href);
+    return;
+  }
+  window.location.assign(href);
+}
+
+export { navigateAiChatCrossOrigin };

package/dist/packages/stack/src/plugins/ai-chat/client/overrides.cjs

@@ -1,5 +1,25 @@
 'use strict';
 
+const client = require('@btst/stack/client');
+
+function resolveAiChatMode(providerConfig) {
+  return providerConfig?.mode === "public" ? "public" : "authenticated";
+}
+function resolveAiChatSitePath(basePath, ...segments) {
+  return client.normalizePath([basePath, ...segments].join("/"));
+}
+function resolveAiChatApiUrl(baseURL, basePath) {
+  return `${baseURL ?? ""}${client.normalizePath([basePath ?? "", "chat"].join("/"))}`;
+}
+function resolveAiChatSiteLocation(site, currentOrigin, ...segments) {
+  const path = resolveAiChatSitePath(site.basePath, ...segments);
+  const href = site.baseURL ? `${site.baseURL}${path}` : path;
+  return {
+    path,
+    href,
+    crossOrigin: Boolean(site.baseURL && currentOrigin) && new URL(href).origin !== currentOrigin
+  };
+}
 const DEFAULT_ALLOWED_FILE_TYPES = ["image"];
 const FILE_TYPE_MIME_MAP = {
   image: ["image/*"],
@@ -12,3 +32,7 @@ const FILE_TYPE_MIME_MAP = {
 
 exports.DEFAULT_ALLOWED_FILE_TYPES = DEFAULT_ALLOWED_FILE_TYPES;
 exports.FILE_TYPE_MIME_MAP = FILE_TYPE_MIME_MAP;
+exports.resolveAiChatApiUrl = resolveAiChatApiUrl;
+exports.resolveAiChatMode = resolveAiChatMode;
+exports.resolveAiChatSiteLocation = resolveAiChatSiteLocation;
+exports.resolveAiChatSitePath = resolveAiChatSitePath;

package/dist/packages/stack/src/plugins/ai-chat/client/overrides.mjs

@@ -1,3 +1,23 @@
+import { normalizePath } from '@btst/stack/client';
+
+function resolveAiChatMode(providerConfig) {
+  return providerConfig?.mode === "public" ? "public" : "authenticated";
+}
+function resolveAiChatSitePath(basePath, ...segments) {
+  return normalizePath([basePath, ...segments].join("/"));
+}
+function resolveAiChatApiUrl(baseURL, basePath) {
+  return `${baseURL ?? ""}${normalizePath([basePath ?? "", "chat"].join("/"))}`;
+}
+function resolveAiChatSiteLocation(site, currentOrigin, ...segments) {
+  const path = resolveAiChatSitePath(site.basePath, ...segments);
+  const href = site.baseURL ? `${site.baseURL}${path}` : path;
+  return {
+    path,
+    href,
+    crossOrigin: Boolean(site.baseURL && currentOrigin) && new URL(href).origin !== currentOrigin
+  };
+}
 const DEFAULT_ALLOWED_FILE_TYPES = ["image"];
 const FILE_TYPE_MIME_MAP = {
   image: ["image/*"],
@@ -8,4 +28,4 @@ const FILE_TYPE_MIME_MAP = {
   json: ["application/json"]
 };
 
-export { DEFAULT_ALLOWED_FILE_TYPES, FILE_TYPE_MIME_MAP };
+export { DEFAULT_ALLOWED_FILE_TYPES, FILE_TYPE_MIME_MAP, resolveAiChatApiUrl, resolveAiChatMode, resolveAiChatSiteLocation, resolveAiChatSitePath };