@tablestore/tablestore-agent-cli 0.3.1 → 0.4.0

package/dist/dashboard-runtime/dashboard/src/components/chat/chat-scope-entry-hydrator.tsx

@@ -0,0 +1,116 @@
+"use client";
+
+import { useEffect, useRef } from "react";
+import { useSearchParams } from "next/navigation";
+
+import { useChatConfig } from "@/components/chat/chat-config-context";
+import { useWiki } from "@/hooks/use-wikis";
+import {
+  decideChatScopeHydration,
+  parseChatScopeEntry,
+  type ChatScopeEntry,
+} from "@/lib/chat-scope-entry";
+import { resolveRetrievalMode } from "@/lib/memory-storage-mode";
+
+export interface ChatScopeEntryHydratorProps {
+  activeSessionId: string | null;
+  messageCount: number;
+  activeInstanceReady: boolean;
+  projectIdentity: string;
+}
+
+export function ChatScopeEntryHydrator(props: ChatScopeEntryHydratorProps) {
+  const searchParams = useSearchParams();
+  const entry = parseChatScopeEntry(
+    new URLSearchParams(searchParams.toString()),
+  );
+  const latestChatRef = useRef({
+    activeSessionId: props.activeSessionId,
+    messageCount: props.messageCount,
+  });
+  latestChatRef.current = {
+    activeSessionId: props.activeSessionId,
+    messageCount: props.messageCount,
+  };
+  const identity = JSON.stringify([
+    props.projectIdentity,
+    entry
+      ? [
+          entry.memoryStoreName,
+          entry.storageMode,
+          entry.scope.appId,
+          entry.scope.tenantId,
+          entry.scope.agentId,
+          entry.scope.runId,
+          entry.wiki,
+        ]
+      : null,
+  ]);
+  const handledRef = useRef({ identity, handled: false });
+  if (handledRef.current.identity !== identity) {
+    handledRef.current = { identity, handled: false };
+  }
+
+  if (!props.activeInstanceReady || !entry) return null;
+  return (
+    <ReadyChatScopeEntryHydrator
+      entry={entry}
+      identity={identity}
+      latestChatRef={latestChatRef}
+      handledRef={handledRef}
+    />
+  );
+}
+
+function ReadyChatScopeEntryHydrator({
+  entry,
+  identity,
+  latestChatRef,
+  handledRef,
+}: {
+  entry: ChatScopeEntry;
+  identity: string;
+  latestChatRef: {
+    current: { activeSessionId: string | null; messageCount: number };
+  };
+  handledRef: { current: { identity: string; handled: boolean } };
+}) {
+  const cfg = useChatConfig();
+  const query = useWiki(entry.memoryStoreName, entry.scope);
+
+  useEffect(() => {
+    if (query.status === "pending") return;
+    if (handledRef.current.identity !== identity) return;
+
+    const latestChat = latestChatRef.current;
+    const decision = decideChatScopeHydration({
+      entry,
+      activeSessionId: latestChat.activeSessionId,
+      messageCount: latestChat.messageCount,
+      wikiReady: query.data?.status === "ready",
+      handled: handledRef.current.handled,
+    });
+    handledRef.current.handled = true;
+    if (decision.action === "ignore") return;
+
+    cfg.setMemory({
+      memoryStoreName: entry.memoryStoreName,
+      storageMode: entry.storageMode,
+      scope: entry.scope,
+      autoSave: true,
+      topK: 5,
+      retrievalMode: resolveRetrievalMode(undefined, entry.storageMode),
+    });
+    cfg.setWikiEnabled(decision.enableWiki);
+  }, [
+    cfg,
+    entry,
+    handledRef,
+    identity,
+    latestChatRef,
+    query.data?.status,
+    query.status,
+  ]);
+
+  return null;
+}

package/dist/dashboard-runtime/dashboard/src/components/chat/memory-bind-section.tsx

@@ -27,11 +27,13 @@ import {
 } from "@/components/ui/select";
 import { useChatConfig } from "@/components/chat/chat-config-context";
 import { MemoryFilesDialog } from "@/components/chat/memory-files-dialog";
+import { WikiBindSection } from "@/components/chat/wiki-bind-section";
 import { ScopeCombobox } from "@/components/memory/scope-combobox";
 import { useI18n } from "@/i18n/i18n-context";
 import { cn } from "@/lib/utils";
 import {
   normalizeStorageMode,
+  parseKnownStorageMode,
   resolveRetrievalMode,
   storageModeCapabilities,
   type StorageMode,
@@ -51,7 +53,7 @@ import {
 export function MemoryBindSection() {
   const { t } = useI18n();
   const cfg = useChatConfig();
-  const { data: storesPage, isLoading } = useMemoryStores();
+  const { data: storesPage, isLoading, isSuccess } = useMemoryStores();
   const stores = storesPage?.items ?? [];
   const binding = cfg.memory;
   const topKId = useId();
@@ -59,7 +61,13 @@ export function MemoryBindSection() {
   const boundStore = binding
     ? stores.find((s) => s.memoryStoreName === binding.memoryStoreName)
     : undefined;
-  const storageMode = normalizeStorageMode(boundStore?.storageMode);
+  const boundStoreFound = !!boundStore;
+  const authoritativeStorageMode = parseKnownStorageMode(
+    boundStore?.storageMode,
+  );
+  const storageMode = normalizeStorageMode(
+    boundStore?.storageMode ?? binding?.storageMode,
+  );
   const capabilities = storageModeCapabilities(storageMode);
   const autoSaveAllowed =
     capabilities.structuredAutoSave || capabilities.writableFiles;
@@ -69,11 +77,14 @@ export function MemoryBindSection() {
   // 防止加载态误判。
   const currentMode = binding?.retrievalMode;
   useEffect(() => {
-    if (!binding || isLoading) return;
+    if (!binding || !isSuccess || !boundStore) return;
     const corrected = resolveRetrievalMode(currentMode, storageMode);
+    const nextStorageMode = authoritativeStorageMode ?? undefined;
     const patch = {
       ...(currentMode !== corrected ? { retrievalMode: corrected } : {}),
-      ...(binding.storageMode !== storageMode ? { storageMode } : {}),
+      ...(binding.storageMode !== nextStorageMode
+        ? { storageMode: nextStorageMode }
+        : {}),
     };
     if (Object.keys(patch).length > 0) {
       cfg.updateMemory(patch);
@@ -84,8 +95,10 @@ export function MemoryBindSection() {
     binding?.memoryStoreName,
     binding?.storageMode,
     currentMode,
+    authoritativeStorageMode,
     storageMode,
-    isLoading,
+    isSuccess,
+    boundStoreFound,
   ]);
 
   return (
@@ -131,15 +144,20 @@ export function MemoryBindSection() {
         <MemoryBindEmptyState
           stores={stores}
           isLoading={isLoading}
-          onBind={() =>
+          onBind={() => {
+            const authoritativeStorageMode = parseKnownStorageMode(
+              stores[0]?.storageMode,
+            );
             cfg.setMemory({
               memoryStoreName: stores[0]?.memoryStoreName ?? "",
-              storageMode: normalizeStorageMode(stores[0]?.storageMode),
+              ...(authoritativeStorageMode !== null
+                ? { storageMode: authoritativeStorageMode }
+                : {}),
               scope: { appId: "", tenantId: "", agentId: "", runId: "" },
               autoSave: true,
               topK: 5,
-            })
-          }
+            });
+          }}
         />
       ) : (
         <>
@@ -147,16 +165,17 @@ export function MemoryBindSection() {
             value={binding.memoryStoreName}
             stores={stores}
             onChange={(v) => {
-              const nextMode = normalizeStorageMode(
-                stores.find((store) => store.memoryStoreName === v)
-                  ?.storageMode,
-              );
+              const rawMode = stores.find(
+                (store) => store.memoryStoreName === v,
+              )?.storageMode;
+              const authoritativeStorageMode = parseKnownStorageMode(rawMode);
+              const runtimeStorageMode = normalizeStorageMode(rawMode);
               cfg.updateMemory({
                 memoryStoreName: v,
-                storageMode: nextMode,
+                storageMode: authoritativeStorageMode ?? undefined,
                 retrievalMode: resolveRetrievalMode(
                   binding.retrievalMode,
-                  nextMode,
+                  runtimeStorageMode,
                 ),
               });
             }}
@@ -184,6 +203,7 @@ export function MemoryBindSection() {
             onScopeChange={(patch) => cfg.updateMemory({ scope: patch })}
             onAutoSaveChange={(v) => cfg.updateMemory({ autoSave: v })}
           />
+          <WikiBindSection />
         </>
       )}
     </div>
@@ -448,6 +468,7 @@ function MemoryBindingForm(props: {
     appId?: string;
     tenantId?: string;
     agentId?: string;
+    runId?: string;
   }) => void;
   onAutoSaveChange: (v: boolean) => void;
 }) {
@@ -477,6 +498,7 @@ function MemoryBindingForm(props: {
                 appId: s.appId ?? "",
                 tenantId: s.tenantId ?? "",
                 agentId: s.agentId ?? "",
+                runId: s.runId ?? "",
               })
             }
           />

package/dist/dashboard-runtime/dashboard/src/components/chat/wiki-bind-section.tsx

@@ -0,0 +1,153 @@
+"use client";
+
+import { useEffect, useState, type ReactNode } from "react";
+import { BookOpen } from "lucide-react";
+
+import {
+  DEFAULT_WIKI_TOP_K,
+  isWikiEligibleMemoryBinding,
+  type ChatMemoryWriteScope,
+  useChatConfig,
+} from "@/components/chat/chat-config-context";
+import { Input } from "@/components/ui/input";
+import { Label } from "@/components/ui/label";
+import { Switch } from "@/components/ui/switch";
+import { useWiki } from "@/hooks/use-wikis";
+import { useI18n } from "@/i18n/i18n-context";
+import type { Wiki } from "@/types/wiki";
+
+function isExactScope(
+  scope: ChatMemoryWriteScope | undefined,
+): scope is ChatMemoryWriteScope {
+  if (!scope) return false;
+  return [scope.appId, scope.tenantId, scope.agentId, scope.runId].every(
+    (value) => value.trim().length > 0 && value.trim() !== "*",
+  );
+}
+
+function isReadyWiki(wiki: Wiki | undefined): wiki is Wiki {
+  return wiki?.status === "ready";
+}
+
+export function clampWikiTopK(value: string, fallback: number): number {
+  const parsed = Number(value);
+  if (!Number.isFinite(parsed) || value.trim().length === 0) return fallback;
+  return Math.min(10, Math.max(1, Math.trunc(parsed)));
+}
+
+export function WikiBindSection() {
+  const { t } = useI18n();
+  const cfg = useChatConfig();
+  const binding = cfg.memory;
+  const knownStorageMode =
+    binding?.storageMode === "ots" ||
+    binding?.storageMode === "file+ots" ||
+    binding?.storageMode === "filemem";
+  const exactScope = isExactScope(binding?.scope) ? binding.scope : undefined;
+  const eligibleScope = isWikiEligibleMemoryBinding(binding)
+    ? exactScope
+    : undefined;
+  const query = useWiki(
+    knownStorageMode ? binding?.memoryStoreName : undefined,
+    eligibleScope,
+  );
+  const ready = Boolean(eligibleScope) && isReadyWiki(query.data);
+  const wikiEnabled = cfg.wikiEnabled && Boolean(eligibleScope);
+  const [draftTopK, setDraftTopK] = useState(String(cfg.wikiTopK));
+
+  useEffect(() => setDraftTopK(String(cfg.wikiTopK)), [cfg.wikiTopK]);
+
+  function commitTopK() {
+    const topK = clampWikiTopK(draftTopK, cfg.wikiTopK ?? DEFAULT_WIKI_TOP_K);
+    setDraftTopK(String(topK));
+    cfg.setWikiTopK(topK);
+  }
+
+  const unavailableHint = !binding
+    ? t("chat.wikiBind.requiresMemory")
+    : !knownStorageMode
+      ? t("chat.wikiBind.requiresKnownStorageMode")
+      : !exactScope
+        ? t("chat.wikiBind.requiresExactScope")
+        : query.isLoading
+          ? t("common.loading")
+          : query.error
+            ? t("chat.wikiBind.loadFailed")
+            : !query.data
+              ? t("chat.wikiBind.notFound")
+              : !ready
+                ? t("chat.wikiBind.notReady")
+                : undefined;
+
+  return (
+    <div className="space-y-3 border-t border-[hsl(var(--border-light))] pt-3">
+      <div>
+        <Label className="text-xs font-medium uppercase tracking-wider text-muted-foreground">
+          {t("chat.wikiBind.title")}
+        </Label>
+        <p className="mt-1 text-[11px] leading-relaxed text-muted-foreground">
+          {t("chat.wikiBind.description")}
+        </p>
+      </div>
+
+      {eligibleScope && query.data ? (
+        <div className="flex items-center gap-3 rounded-md border border-[hsl(var(--border))] bg-[hsl(var(--muted)/0.3)] px-3 py-2">
+          <BookOpen className="h-3.5 w-3.5 shrink-0 text-[hsl(var(--primary))]" />
+          <div className="min-w-0 flex-1">
+            <p className="truncate text-sm font-medium">{query.data.name}</p>
+            <p className="text-[11px] text-muted-foreground">
+              {ready ? t("chat.wikiBind.ready") : t("chat.wikiBind.notReady")}
+            </p>
+          </div>
+          <Switch
+            checked={wikiEnabled}
+            disabled={!wikiEnabled && !ready}
+            onCheckedChange={(checked) => cfg.setWikiEnabled(checked && ready)}
+            aria-label={t("chat.wikiBind.toggle")}
+          />
+        </div>
+      ) : (
+        <Switch
+          checked={false}
+          disabled
+          onCheckedChange={() => undefined}
+          aria-label={t("chat.wikiBind.toggle")}
+        />
+      )}
+
+      {wikiEnabled ? (
+        <details className="rounded-md border border-[hsl(var(--border))] px-3 py-2">
+          <summary className="cursor-pointer text-xs font-medium text-muted-foreground">
+            {t("chat.wikiBind.advanced")}
+          </summary>
+          <div className="mt-2 flex items-center justify-end gap-2">
+            <Label className="text-xs text-muted-foreground">
+              {t("chat.wikiBind.topK")}
+            </Label>
+            <Input
+              type="number"
+              min={1}
+              max={10}
+              step={1}
+              className="h-8 w-16"
+              value={draftTopK}
+              onChange={(event) => setDraftTopK(event.target.value)}
+              onBlur={commitTopK}
+              aria-label={t("chat.wikiBind.topK")}
+            />
+          </div>
+        </details>
+      ) : null}
+
+      {unavailableHint ? <WikiHint>{unavailableHint}</WikiHint> : null}
+    </div>
+  );
+}
+
+function WikiHint({ children }: { children: ReactNode }) {
+  return (
+    <p className="rounded-md border border-dashed border-[hsl(var(--border))] px-3 py-2 text-xs text-muted-foreground">
+      {children}
+    </p>
+  );
+}

package/dist/dashboard-runtime/dashboard/src/components/memory/memory-dream-actions-table.tsx

@@ -1,6 +1,7 @@
 "use client";
 
 import { useMemo, useState } from "react";
+import Link from "next/link";
 import { toast } from "sonner";
 import { Sparkles } from "lucide-react";
 
@@ -56,6 +57,7 @@ interface DreamActionsTableProps {
   memoryStoreName: string;
   dreamId: string;
   taskType?: string;
+  artifactWikiHref?: (action: DreamActionRecord) => string | undefined;
 }
 
 const STATUS_CLASS: Record<DreamActionStatus, string> = {
@@ -273,6 +275,7 @@ export function MemoryDreamActionsTable({
   memoryStoreName,
   dreamId,
   taskType,
+  artifactWikiHref,
 }: DreamActionsTableProps) {
   const { t } = useI18n();
   const STATUS_OPTIONS = useMemo<
@@ -530,6 +533,7 @@ export function MemoryDreamActionsTable({
               <TableBody>
                 {actions.map((action) => {
                   const canApply = isApplyable(action);
+                  const wikiHref = artifactWikiHref?.(action);
                   return (
                     <TableRow
                       key={action.actionId}
@@ -581,6 +585,15 @@ export function MemoryDreamActionsTable({
                           memoryStoreName={memoryStoreName}
                           t={t}
                         />
+                        {wikiHref ? (
+                          <Link
+                            href={wikiHref}
+                            className="mt-1 inline-block text-xs text-primary hover:underline"
+                            onClick={(event) => event.stopPropagation()}
+                          >
+                            {t("memory.dream.actions.openInWiki")}
+                          </Link>
+                        ) : null}
                       </TableCell>
                       {!isExtractTask && (
                         <TableCell className="text-right font-mono text-xs">