@tablestore/tablestore-agent-cli 0.2.11 → 0.3.1

package/dist/dashboard-runtime/dashboard/src/components/memory/scope-combobox.tsx

@@ -1,6 +1,6 @@
 "use client";
 
-import { useState } from "react";
+import { useMemo, useState } from "react";
 import { Check, ChevronsUpDown } from "lucide-react";
 
 import { Button } from "@/components/ui/button";
@@ -18,8 +18,9 @@ import {
   PopoverTrigger,
 } from "@/components/ui/popover";
 import { useAllMemoryStoreScopes } from "@/hooks/use-memory-store-scopes";
+import { scopeLabel } from "@/lib/memory";
 import { cn } from "@/lib/utils";
-import { useI18n } from "@/i18n/i18n-context";
+import { useI18n, type MessageKey } from "@/i18n/i18n-context";
 import type { MemoryScope } from "@/types/memory";
 
 const WILDCARD: MemoryScope = {
@@ -29,16 +30,77 @@ const WILDCARD: MemoryScope = {
   runId: "*",
 };
 
-function scopeLabel(scope: MemoryScope): string {
-  return [scope.appId, scope.tenantId, scope.agentId, scope.runId]
-    .map((v) => v || "—")
-    .join(" / ");
+/** Stable identity so the memoized filter isn't invalidated every render. */
+const NO_SCOPES: MemoryScope[] = [];
+
+/** How many rows to render at once, and how many more each "show more" adds. */
+export const SCOPE_WINDOW_SIZE = 100;
+
+/** Distance from the bottom (px) at which scrolling grows the window. */
+const NEAR_BOTTOM_PX = 64;
+
+/**
+ * Case-insensitive substring match over the scope label, where every
+ * whitespace-separated term must hit somewhere — so `ots claude` narrows
+ * across two dimensions without caring about their order.
+ *
+ * We filter here rather than letting cmdk do it because cmdk's default is a
+ * fuzzy subsequence score, and run ids are uuids: almost any short query
+ * finds scattered hex characters in an unrelated tuple, so the "matches"
+ * drown out the real ones. Owning the filter also keeps DOM order equal to
+ * JSX order, which is what lets the window below stay correct.
+ */
+export function filterScopes(
+  scopes: readonly MemoryScope[],
+  query: string,
+): MemoryScope[] {
+  const terms = query.toLowerCase().split(/\s+/).filter(Boolean);
+  if (terms.length === 0) return [...scopes];
+  return scopes.filter((scope) => {
+    const label = scopeLabel(scope).toLowerCase();
+    return terms.every((term) => label.includes(term));
+  });
+}
+
+export function isNearListBottom(el: {
+  scrollTop: number;
+  clientHeight: number;
+  scrollHeight: number;
+}): boolean {
+  return el.scrollHeight - el.scrollTop - el.clientHeight <= NEAR_BOTTOM_PX;
+}
+
+/**
+ * Which message the status bar shows. Windowing and server-side truncation are
+ * independent and, on a big enough store, simultaneous — truncation in fact
+ * implies windowing — so they get a combined message rather than an either/or
+ * that would hide "· 已截断" in the one case it matters.
+ */
+export function scopeStatusKey(state: {
+  loading: boolean;
+  error: boolean;
+  hasMore: boolean;
+  truncated: boolean;
+}): MessageKey {
+  if (state.loading) return "memory.common.querying";
+  if (state.error) return "memory.scopeCombobox.loadFailed";
+  if (state.hasMore) {
+    return state.truncated
+      ? "memory.scopeCombobox.windowCountTruncated"
+      : "memory.scopeCombobox.windowCount";
+  }
+  return state.truncated
+    ? "memory.common.scopeCountTruncated"
+    : "memory.common.scopeCount";
 }
 
 /**
  * 通过 ListScopes 拉取某记忆库里真实存在的 scope 元组，供下拉选择。选中后回调
  * `onPick(完整 scope)`，由上层决定填哪些字段。对话配置的记忆 scope、Dream 任务
  * 的 scope 输入共用它，避免用户手动逐字段敲。
+ *
+ * scope 多起来时的三件事：列表本身要能滚（见下方 `modal` 注释）、一次只渲染
+ * 一屏窗口（滚到底或点「显示更多」再追加），以及搜索由本组件负责而不是 cmdk。
  */
 export function ScopeCombobox({
   memoryStoreName,
@@ -55,14 +117,44 @@ export function ScopeCombobox({
 }) {
   const { t } = useI18n();
   const [open, setOpen] = useState(false);
-  const query = useAllMemoryStoreScopes(memoryStoreName ?? "", WILDCARD);
-  const scopes = query.data?.items ?? [];
+  const [search, setSearch] = useState("");
+  const [windowSize, setWindowSize] = useState(SCOPE_WINDOW_SIZE);
+
+  // 只在弹层打开时才查。Dream 创建对话框最多同时挂 20 个 ScopeCombobox，
+  // 抽屉/对话框一打开就全量拉 scope 是白花的往返。
+  const query = useAllMemoryStoreScopes(memoryStoreName ?? "", WILDCARD, {
+    enabled: open,
+  });
+  const scopes = query.data?.items ?? NO_SCOPES;
   const truncated = query.data?.truncated ?? false;
   const resolvedPlaceholder =
     placeholder ?? t("memory.scopeCombobox.defaultPlaceholder");
 
+  const matches = useMemo(() => filterScopes(scopes, search), [scopes, search]);
+  const visible = matches.slice(0, windowSize);
+  const hasMore = visible.length < matches.length;
+
+  const showMore = () => setWindowSize((n) => n + SCOPE_WINDOW_SIZE);
+
+  const handleOpenChange = (next: boolean) => {
+    setOpen(next);
+    if (!next) {
+      setSearch("");
+      setWindowSize(SCOPE_WINDOW_SIZE);
+    }
+  };
+
   return (
-    <Popover open={open} onOpenChange={setOpen}>
+    // `modal` is load-bearing, not decoration. Both call sites live inside a
+    // Radix Dialog (the chat config Sheet / the Dream create dialog), whose
+    // overlay wraps everything in <RemoveScroll shards={[content]}> with no
+    // `noIsolation`. That cancels wheel events on any portalled node the
+    // dialog doesn't own — and PopoverContent portals to document.body, so
+    // the list silently refused to scroll and only the first ~10 rows were
+    // ever reachable. `modal` gives the popover its own scroll lock, which
+    // sits on top of react-remove-scroll's lockStack and hands scrolling
+    // back. See components/ui/__tests__/popover-in-dialog.test.ts.
+    <Popover open={open} onOpenChange={handleOpenChange} modal>
       <PopoverTrigger asChild>
         <Button
           type="button"
@@ -76,28 +168,40 @@ export function ScopeCombobox({
         </Button>
       </PopoverTrigger>
       <PopoverContent align="start" className="w-[420px] p-0">
-        <Command shouldFilter>
+        <Command shouldFilter={false}>
           <CommandInput
             placeholder={t("memory.common.searchScopePlaceholder")}
             className="text-xs"
+            value={search}
+            onValueChange={(v) => {
+              setSearch(v);
+              setWindowSize(SCOPE_WINDOW_SIZE);
+            }}
           />
-          <CommandList className="max-h-[320px]">
+          <CommandList
+            className="max-h-[min(52vh,420px)]"
+            onScroll={(e) => {
+              if (hasMore && isNearListBottom(e.currentTarget)) showMore();
+            }}
+          >
             <CommandEmpty>
               {query.isLoading
                 ? t("common.loading")
-                : t("memory.scopeCombobox.empty")}
+                : query.isError
+                  ? t("memory.scopeCombobox.loadFailed")
+                  : search
+                    ? t("memory.common.noMatchingScope")
+                    : t("memory.scopeCombobox.empty")}
             </CommandEmpty>
-            {scopes.length > 0 && (
+            {visible.length > 0 && (
+              // 标题只报命中数；服务端截断由状态条统一表达，避免「仅前 3 条」
+              // 这种把搜索命中数说成服务端上限的读法。
               <CommandGroup
-                heading={
-                  truncated
-                    ? t("memory.common.knownScopesTruncated", {
-                        count: scopes.length,
-                      })
-                    : t("memory.common.knownScopes", { count: scopes.length })
-                }
+                heading={t("memory.common.knownScopes", {
+                  count: matches.length,
+                })}
               >
-                {scopes.map((scope, idx) => {
+                {visible.map((scope, idx) => {
                   const label = scopeLabel(scope);
                   return (
                     <CommandItem
@@ -105,7 +209,7 @@ export function ScopeCombobox({
                       value={label}
                       onSelect={() => {
                         onPick(scope);
-                        setOpen(false);
+                        handleOpenChange(false);
                       }}
                       className="text-xs"
                     >
@@ -119,27 +223,50 @@ export function ScopeCombobox({
               </CommandGroup>
             )}
           </CommandList>
-          <div className="flex items-center justify-between border-t border-[hsl(var(--border-light))] px-2 py-1.5 text-[10px] text-muted-foreground">
-            <span>
-              {query.isLoading
-                ? t("memory.common.querying")
-                : truncated
-                  ? t("memory.common.scopeCountTruncated", {
-                      count: scopes.length,
-                    })
-                  : t("memory.common.scopeCount", { count: scopes.length })}
+          {/* 状态条留在 CommandList 之外：它不属于 [cmdk-list]，因此永远不会
+              被搜索过滤掉或被 cmdk 的排序搬走。 */}
+          <div className="flex items-center justify-between gap-2 border-t border-[hsl(var(--border-light))] px-2 py-1.5 text-[10px] text-muted-foreground">
+            <span className="truncate">
+              {t(
+                scopeStatusKey({
+                  loading: query.isLoading,
+                  error: query.isError,
+                  hasMore,
+                  truncated,
+                }),
+                // 各 key 只会用到其中一部分占位符，多余的会被 interpolate 忽略。
+                {
+                  shown: visible.length,
+                  total: matches.length,
+                  count: matches.length,
+                },
+              )}
+            </span>
+            <span className="flex shrink-0 items-center gap-1">
+              {hasMore ? (
+                <button
+                  type="button"
+                  onClick={(e) => {
+                    e.preventDefault();
+                    showMore();
+                  }}
+                  className="rounded px-1.5 py-0.5 font-medium text-[hsl(var(--primary))] hover:bg-[hsl(var(--accent))]"
+                >
+                  {t("memory.scopeCombobox.showMore")}
+                </button>
+              ) : null}
+              <button
+                type="button"
+                onClick={(e) => {
+                  e.preventDefault();
+                  query.refetch();
+                }}
+                disabled={query.isFetching}
+                className="rounded px-1.5 py-0.5 hover:bg-[hsl(var(--accent))] disabled:opacity-50"
+              >
+                {t("common.refresh")}
+              </button>
             </span>
-            <button
-              type="button"
-              onClick={(e) => {
-                e.preventDefault();
-                query.refetch();
-              }}
-              disabled={query.isFetching}
-              className="rounded px-1.5 py-0.5 hover:bg-[hsl(var(--accent))] disabled:opacity-50"
-            >
-              {t("common.refresh")}
-            </button>
           </div>
         </Command>
       </PopoverContent>

package/dist/dashboard-runtime/dashboard/src/components/qa/qa-message.tsx

@@ -34,7 +34,7 @@ import { Textarea } from "@/components/ui/textarea";
 import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
 import { ConfirmDialog } from "@/components/shared/confirm-dialog";
 import { useI18n, type TranslateFn } from "@/i18n/i18n-context";
-import { useCreateFileMemory } from "@/hooks/use-file-memory";
+import { useAddItem } from "@/hooks/use-file-memory";
 import { cn, extractFileName } from "@/lib/utils";
 import { IMAGE_FALLBACK_DATA_PREFIX } from "@/lib/constants";
 import type {
@@ -563,7 +563,7 @@ function AssistantBlock({
   const [touched, setTouched] = useState(false);
   const [confirmDeleteOpen, setConfirmDeleteOpen] = useState(false);
   const [confirmRegenOpen, setConfirmRegenOpen] = useState(false);
-  const createFile = useCreateFileMemory();
+  const addItem = useAddItem();
 
   const isStreaming = message.status === "streaming";
   const hasError = message.status === "error";
@@ -639,11 +639,12 @@ function AssistantBlock({
       return;
     }
     try {
-      await createFile.mutateAsync({
+      await addItem.mutateAsync({
         memoryStoreName: message.memoryFilesStoreName,
         scope: message.memoryFilesScope,
         path,
         content: message.content,
+        view: "full",
       });
       toast.success(t("qa.message.savedToFile", { path }));
     } catch (error) {

package/dist/dashboard-runtime/dashboard/src/components/shared/llm-model-selector.tsx

@@ -206,7 +206,11 @@ function LLMModelCombobox({
   };
 
   return (
-    <Popover open={open} onOpenChange={setOpen}>
+    // `modal` is load-bearing: this selector is rendered inside the chat config
+    // Sheet, and a Radix Dialog's scroll lock cancels wheel events on portalled
+    // content it doesn't own — without it the model list refuses to scroll.
+    // See components/ui/__tests__/popover-in-dialog.test.ts.
+    <Popover open={open} onOpenChange={setOpen} modal>
       <PopoverTrigger asChild>
         <Button
           type="button"

package/dist/dashboard-runtime/dashboard/src/components/ui/dialog.tsx

@@ -36,7 +36,7 @@ const DialogContent = React.forwardRef<
     <DialogPrimitive.Content
       ref={ref}
       className={cn(
-        "fixed left-[50%] top-[50%] z-50 grid w-full max-w-[480px] translate-x-[-50%] translate-y-[-50%] gap-4 rounded-lg border border-[hsl(var(--border))] p-6 shadow-md",
+        "fixed left-[50%] top-[50%] z-50 grid w-[calc(100%-2rem)] max-h-[calc(100dvh-2rem)] max-w-[480px] translate-x-[-50%] translate-y-[-50%] gap-4 overflow-y-auto rounded-lg border border-[hsl(var(--border))] p-6 shadow-md",
         "duration-200 ease-[cubic-bezier(0.34,1.56,0.64,1)]",
         "data-[state=open]:animate-in data-[state=closed]:animate-out",
         "data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0",

package/dist/dashboard-runtime/dashboard/src/contexts/subspace-context.tsx

@@ -10,11 +10,34 @@ import {
 } from "react";
 import { usePathname, useRouter, useSearchParams } from "next/navigation";
 import {
-  clearActiveSubspace,
-  getActiveSubspace,
-  setActiveSubspace,
+  getSubspaceScopeId,
   useSubspaceHistory,
 } from "@/hooks/use-subspace-history";
+import {
+  clearActiveSubspace,
+  readActiveSubspace,
+  writeActiveSubspace,
+} from "@/lib/subspace-history";
+import { useDataScopeKey } from "@/providers/runtime-context-provider";
+
+type ActiveSubspaceStorage = Pick<
+  Storage,
+  "getItem" | "setItem" | "removeItem"
+>;
+
+const unavailableSessionStorage: ActiveSubspaceStorage = {
+  getItem: () => null,
+  setItem: () => {},
+  removeItem: () => {},
+};
+
+export function getSessionStorage(): ActiveSubspaceStorage {
+  try {
+    return window.sessionStorage;
+  } catch {
+    return unavailableSessionStorage;
+  }
+}
 
 export interface SubspaceContextValue {
   subspace: string | null;
@@ -22,6 +45,7 @@ export interface SubspaceContextValue {
   clearSubspace: () => void;
   history: string[];
   addToHistory: (subspace: string) => void;
+  clearHistory: () => void;
 }
 
 export const SubspaceContext = createContext<SubspaceContextValue | null>(null);
@@ -52,12 +76,68 @@ interface SubspaceProviderProps {
   children: ReactNode;
 }
 
+export interface ActiveSubspaceReconcileState {
+  scopeId: string | null;
+  pendingRestore: string | null;
+}
+
+export type ActiveSubspaceReconcileResult = {
+  state: ActiveSubspaceReconcileState;
+  effect:
+    | { type: "restore"; subspace: string }
+    | { type: "persist"; subspace: string }
+    | { type: "clear" }
+    | { type: "none" };
+};
+
+export function reconcileActiveSubspace(
+  state: ActiveSubspaceReconcileState,
+  scopeId: string,
+  urlSubspace: string | null,
+  storedSubspace: string | null,
+): ActiveSubspaceReconcileResult {
+  if (state.scopeId !== scopeId) {
+    if (urlSubspace) {
+      return {
+        state: { scopeId, pendingRestore: null },
+        effect: { type: "persist", subspace: urlSubspace },
+      };
+    }
+    if (storedSubspace) {
+      return {
+        state: { scopeId, pendingRestore: storedSubspace },
+        effect: { type: "restore", subspace: storedSubspace },
+      };
+    }
+    return {
+      state: { scopeId, pendingRestore: null },
+      effect: { type: "clear" },
+    };
+  }
+
+  if (state.pendingRestore && !urlSubspace) {
+    return { state, effect: { type: "none" } };
+  }
+
+  if (urlSubspace) {
+    return {
+      state: { scopeId, pendingRestore: null },
+      effect: { type: "persist", subspace: urlSubspace },
+    };
+  }
+
+  return {
+    state: { scopeId, pendingRestore: null },
+    effect: { type: "clear" },
+  };
+}
+
 /**
  * Subspace state container backed by the URL as the single source of truth.
  * The `?subspace=` query parameter drives every read; writes only call
  * router.replace and let the URL change propagate. A per-tab sessionStorage
- * preference is replayed into the URL once on mount so revisiting the page
- * without a query string still lands on the previously chosen scope.
+ * preference is reconciled into the URL once per runtime/KB identity so
+ * revisiting without a query string lands on the previously chosen scope.
  */
 export function SubspaceProvider({
   knowledgeBaseName,
@@ -67,38 +147,65 @@ export function SubspaceProvider({
   const searchParamsKey = searchParams.toString();
   const pathname = usePathname();
   const router = useRouter();
-  const { history, addToHistory } = useSubspaceHistory(knowledgeBaseName);
+  const dataScopeKey = useDataScopeKey();
+  const { history, addToHistory, clearHistory } = useSubspaceHistory(
+    dataScopeKey,
+    knowledgeBaseName,
+  );
+  const activeScopeId = getSubspaceScopeId(dataScopeKey, knowledgeBaseName);
 
   // URL is the single source of truth — derive directly, no local state.
   const subspace = searchParams.get("subspace");
 
-  // One-shot mount-time sticky hydration: when arriving without a query
-  // param but a per-tab session preference exists, replay it into the URL.
-  // After this single redirect every read flows through URL again, so a
-  // later clearSubspace() cannot be undone by re-running this effect.
-  const hydratedRef = useRef(false);
+  // Reconcile once per runtime/KB identity. A pending restore suppresses the
+  // persistence branch until router.replace has replayed it into the URL.
+  const reconcileRef = useRef<ActiveSubspaceReconcileState>({
+    scopeId: null,
+    pendingRestore: null,
+  });
   useEffect(() => {
-    if (hydratedRef.current) return;
-    hydratedRef.current = true;
-    if (subspace) return;
-    const stored = getActiveSubspace(knowledgeBaseName);
-    if (!stored) return;
-    const params = new URLSearchParams(searchParamsKey);
-    params.set("subspace", stored);
-    router.replace(`${pathname}?${params.toString()}`);
-  }, [knowledgeBaseName, subspace, searchParamsKey, pathname, router]);
+    const storage = getSessionStorage();
+    const stored =
+      reconcileRef.current.scopeId !== activeScopeId && !subspace
+        ? readActiveSubspace(storage, dataScopeKey, knowledgeBaseName)
+        : null;
+    const result = reconcileActiveSubspace(
+      reconcileRef.current,
+      activeScopeId,
+      subspace,
+      stored,
+    );
+    reconcileRef.current = result.state;
 
-  // Side effects follow the URL: persist to sessionStorage and history
-  // whenever the active scope changes. Centralising the writes here means
-  // setSubspace / clearSubspace stay tiny and free of race conditions.
-  useEffect(() => {
-    if (!subspace) {
-      clearActiveSubspace(knowledgeBaseName);
+    if (result.effect.type === "restore") {
+      const params = new URLSearchParams(searchParamsKey);
+      params.set("subspace", result.effect.subspace);
+      router.replace(`${pathname}?${params.toString()}`);
       return;
     }
-    setActiveSubspace(knowledgeBaseName, subspace);
-    addToHistory(subspace);
-  }, [subspace, knowledgeBaseName, addToHistory]);
+    if (result.effect.type === "persist") {
+      writeActiveSubspace(
+        storage,
+        dataScopeKey,
+        knowledgeBaseName,
+        result.effect.subspace,
+      );
+      addToHistory(result.effect.subspace);
+      return;
+    }
+    if (result.effect.type === "clear") {
+      clearActiveSubspace(storage, dataScopeKey, knowledgeBaseName);
+    }
+  }, [
+    activeScopeId,
+    dataScopeKey,
+    knowledgeBaseName,
+    subspace,
+    searchParamsKey,
+    pathname,
+    router,
+    addToHistory,
+  ]);
 
   const setSubspace = useCallback(
     (next: string) => {
@@ -120,7 +227,14 @@ export function SubspaceProvider({
 
   return (
     <SubspaceContext.Provider
-      value={{ subspace, setSubspace, clearSubspace, history, addToHistory }}
+      value={{
+        subspace,
+        setSubspace,
+        clearSubspace,
+        history,
+        addToHistory,
+        clearHistory,
+      }}
     >
       {children}
     </SubspaceContext.Provider>

package/dist/dashboard-runtime/dashboard/src/hooks/use-approve-dream-plan.ts

@@ -0,0 +1,25 @@
+import { useMutation, useQueryClient } from "@tanstack/react-query";
+
+import { approveDreamPlan } from "@/services/memory";
+
+export function useApproveDreamPlan(memoryStoreName: string) {
+  const queryClient = useQueryClient();
+  return useMutation({
+    mutationFn: ({
+      dreamId,
+      approver,
+    }: {
+      dreamId: string;
+      approver: string;
+    }) => approveDreamPlan(memoryStoreName, dreamId, approver),
+    onSuccess: () => {
+      queryClient.invalidateQueries({
+        queryKey: ["memory"],
+        predicate: (query) =>
+          Array.isArray(query.queryKey) &&
+          query.queryKey.includes("dream-tasks") &&
+          query.queryKey.includes(memoryStoreName),
+      });
+    },
+  });
+}