@tablestore/tablestore-agent-cli 0.2.5 → 0.2.7

package/dist/dashboard-runtime/dashboard/src/components/memory/memory-dream-create-dialog.tsx

@@ -34,7 +34,7 @@ import {
 } from "@/components/ui/collapsible";
 import { Form, FormControl, FormField, FormItem, FormLabel, FormMessage } from "@/components/ui/form";
 import { useCreateDreamTask } from "@/hooks/use-create-dream-task";
-import type { MemoryScope } from "@/types/memory";
+import type { DreamTaskType, MemoryScope } from "@/types/memory";
 
 interface DreamCreateDialogProps {
   memoryStoreName: string;
@@ -56,6 +56,7 @@ const scopeSchema = z
   });
 
 const formSchema = z.object({
+  taskType: z.enum(["memory", "skill", "profile"]),
   scopes: z.array(scopeSchema).min(1, "至少一个 scope"),
   applyMode: z.enum(["proposal", "safe_auto"]),
   scopeOutputMode: z.enum(["preserve_scope", "promote_scope"]).optional(),
@@ -99,6 +100,7 @@ export function MemoryDreamCreateDialog({
   const form = useForm<FormValues>({
     resolver: zodResolver(formSchema),
     defaultValues: {
+      taskType: "memory",
       scopes: [
         {
           // Filter out wildcard "*" placeholder values from the URL ScopeBar
@@ -130,6 +132,9 @@ export function MemoryDreamCreateDialog({
     name: "scopes",
   });
 
+  const taskType = form.watch("taskType");
+  const isMemoryType = taskType === "memory";
+
   const [advancedOpen, setAdvancedOpen] = useState(false);
   const [thresholdsOpen, setThresholdsOpen] = useState(false);
 
@@ -142,6 +147,8 @@ export function MemoryDreamCreateDialog({
     if (tUpdate !== undefined) thresholds.update = tUpdate;
     if (tMerge !== undefined) thresholds.merge = tMerge;
 
+    const isMemory = values.taskType === "memory";
+
     try {
       const result = await createMutation.mutateAsync({
         scopes: values.scopes.map((s) => {
@@ -151,8 +158,9 @@ export function MemoryDreamCreateDialog({
           if (s.runId) out.runId = s.runId;
           return out;
         }),
-        applyMode: values.applyMode,
-        ...(values.scopeOutputMode && { scopeOutputMode: values.scopeOutputMode }),
+        taskType: values.taskType,
+        ...(isMemory && { applyMode: values.applyMode }),
+        ...(isMemory && values.scopeOutputMode && { scopeOutputMode: values.scopeOutputMode }),
         ...(values.minTimestamp && { minTimestamp: values.minTimestamp }),
         ...(values.maxTimestamp && { maxTimestamp: values.maxTimestamp }),
         ...(parseOptionalInt(values.maxSessions) !== undefined && {
@@ -166,12 +174,17 @@ export function MemoryDreamCreateDialog({
         }),
         ...(values.instructions && { instructions: values.instructions }),
         ...(values.clientToken && { clientToken: values.clientToken }),
-        ...(Object.keys(thresholds).length > 0 && {
+        ...(isMemory && Object.keys(thresholds).length > 0 && {
           confidenceThresholds: thresholds,
         }),
       });
 
-      toast.success("记忆整理任务已创建");
+      const TOAST_MSG: Record<DreamTaskType, string> = {
+        memory: "记忆整理任务已创建",
+        skill: "技能抽取任务已创建",
+        profile: "用户画像任务已创建",
+      };
+      toast.success(TOAST_MSG[values.taskType]);
       onOpenChange(false);
       form.reset();
       router.push(
@@ -187,15 +200,38 @@ export function MemoryDreamCreateDialog({
     <Dialog open={open} onOpenChange={onOpenChange}>
       <DialogContent className="max-w-2xl">
         <DialogHeader>
-          <DialogTitle>新建记忆整理任务</DialogTitle>
+          <DialogTitle>新建 Dream 任务</DialogTitle>
           <DialogDescription>
-            Dream 会读取所选 scopes 下的消息与已有记忆，让 LLM 识别模式后生成动作建议（或自动应用）。
-            建议先用「建议模式」生成提案、人工审核后再应用。
+            Dream 会读取所选 scopes 下的消息与已有记忆，让 LLM 进行分析处理。
+            记忆整理模式建议先用「建议模式」生成提案、人工审核后再应用。
           </DialogDescription>
         </DialogHeader>
 
         <Form {...form}>
           <form onSubmit={form.handleSubmit(onSubmit)} className="space-y-5">
+            <FormField
+              control={form.control}
+              name="taskType"
+              render={({ field }) => (
+                <FormItem>
+                  <FormLabel>任务类型</FormLabel>
+                  <Select onValueChange={field.onChange} defaultValue={field.value}>
+                    <FormControl>
+                      <SelectTrigger>
+                        <SelectValue />
+                      </SelectTrigger>
+                    </FormControl>
+                    <SelectContent>
+                      <SelectItem value="memory">记忆整理 — 整合跨会话记忆</SelectItem>
+                      <SelectItem value="skill">技能抽取 — 从对话中提取可复用技能</SelectItem>
+                      <SelectItem value="profile">用户画像 — 整理用户画像</SelectItem>
+                    </SelectContent>
+                  </Select>
+                  <FormMessage />
+                </FormItem>
+              )}
+            />
+
             <section className="space-y-3">
               <div className="flex items-center justify-between">
                 <Label className="text-sm font-medium">Scope 范围</Label>
@@ -240,87 +276,98 @@ export function MemoryDreamCreateDialog({
               </div>
             </section>
 
-            <section className="grid gap-3 sm:grid-cols-2">
-              <FormField
-                control={form.control}
-                name="applyMode"
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>应用模式</FormLabel>
-                    <Select onValueChange={field.onChange} defaultValue={field.value}>
-                      <FormControl>
-                        <SelectTrigger>
-                          <SelectValue />
-                        </SelectTrigger>
-                      </FormControl>
-                      <SelectContent>
-                        <SelectItem value="proposal">
-                          建议模式（生成后人工 apply，推荐）
-                        </SelectItem>
-                        <SelectItem value="safe_auto">
-                          安全自动（仅 ADD/UPDATE 自动；DELETE/跨 scope 仍需人工）
-                        </SelectItem>
-                      </SelectContent>
-                    </Select>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-              <FormField
-                control={form.control}
-                name="scopeOutputMode"
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Scope 输出模式</FormLabel>
-                    <Select
-                      onValueChange={field.onChange}
-                      value={field.value ?? ""}
-                    >
-                      <FormControl>
-                        <SelectTrigger>
-                          <SelectValue placeholder="使用默认（preserve_scope）" />
-                        </SelectTrigger>
-                      </FormControl>
-                      <SelectContent>
-                        <SelectItem value="preserve_scope">保留原 scope</SelectItem>
-                        <SelectItem value="promote_scope">提升到父级 scope</SelectItem>
-                      </SelectContent>
-                    </Select>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </section>
+            {isMemoryType && (
+              <section className="grid gap-3 sm:grid-cols-2">
+                <FormField
+                  control={form.control}
+                  name="applyMode"
+                  render={({ field }) => (
+                    <FormItem>
+                      <FormLabel>应用模式</FormLabel>
+                      <Select onValueChange={field.onChange} defaultValue={field.value}>
+                        <FormControl>
+                          <SelectTrigger>
+                            <SelectValue />
+                          </SelectTrigger>
+                        </FormControl>
+                        <SelectContent>
+                          <SelectItem value="proposal">
+                            建议模式（生成后人工 apply，推荐）
+                          </SelectItem>
+                          <SelectItem value="safe_auto">
+                            安全自动（仅 ADD/UPDATE 自动；DELETE/跨 scope 仍需人工）
+                          </SelectItem>
+                        </SelectContent>
+                      </Select>
+                      <FormMessage />
+                    </FormItem>
+                  )}
+                />
+                <FormField
+                  control={form.control}
+                  name="scopeOutputMode"
+                  render={({ field }) => (
+                    <FormItem>
+                      <FormLabel>Scope 输出模式</FormLabel>
+                      <Select
+                        onValueChange={field.onChange}
+                        value={field.value ?? ""}
+                      >
+                        <FormControl>
+                          <SelectTrigger>
+                            <SelectValue placeholder="使用默认（preserve_scope）" />
+                          </SelectTrigger>
+                        </FormControl>
+                        <SelectContent>
+                          <SelectItem value="preserve_scope">保留原 scope</SelectItem>
+                          <SelectItem value="promote_scope">提升到父级 scope</SelectItem>
+                        </SelectContent>
+                      </Select>
+                      <FormMessage />
+                    </FormItem>
+                  )}
+                />
+              </section>
+            )}
 
             <FormField
               control={form.control}
               name="instructions"
-              render={({ field }) => (
-                <FormItem>
-                  <FormLabel>指令（可选）</FormLabel>
-                  <FormControl>
-                    <Textarea
-                      rows={2}
-                      placeholder="给 dream 规划器的额外指引"
-                      {...field}
-                    />
-                  </FormControl>
-                </FormItem>
-              )}
+              render={({ field }) => {
+                const placeholder: Record<DreamTaskType, string> = {
+                  memory: "给 dream 规划器的额外指引",
+                  skill: "指导技能抽取的方向，例如重点关注的领域",
+                  profile: "用户画像整理的额外指引，例如需要关注的维度",
+                };
+                return (
+                  <FormItem>
+                    <FormLabel>指令（可选）</FormLabel>
+                    <FormControl>
+                      <Textarea
+                        rows={2}
+                        placeholder={placeholder[taskType]}
+                        {...field}
+                      />
+                    </FormControl>
+                  </FormItem>
+                );
+              }}
             />
 
-            <Collapsible open={thresholdsOpen} onOpenChange={setThresholdsOpen}>
-              <CollapsibleTrigger asChild>
-                <Button type="button" variant="ghost" size="sm" className="px-0">
-                  {thresholdsOpen ? "▾" : "▸"} 置信度阈值
-                </Button>
-              </CollapsibleTrigger>
-              <CollapsibleContent className="grid gap-3 pt-2 sm:grid-cols-3">
-                <NumberField name="thresholdAdd" label="add" form={form} placeholder="0.7" />
-                <NumberField name="thresholdUpdate" label="update" form={form} placeholder="0.6" />
-                <NumberField name="thresholdMerge" label="merge" form={form} placeholder="0.6" />
-              </CollapsibleContent>
-            </Collapsible>
+            {isMemoryType && (
+              <Collapsible open={thresholdsOpen} onOpenChange={setThresholdsOpen}>
+                <CollapsibleTrigger asChild>
+                  <Button type="button" variant="ghost" size="sm" className="px-0">
+                    {thresholdsOpen ? "▾" : "▸"} 置信度阈值
+                  </Button>
+                </CollapsibleTrigger>
+                <CollapsibleContent className="grid gap-3 pt-2 sm:grid-cols-3">
+                  <NumberField name="thresholdAdd" label="add" form={form} placeholder="0.7" />
+                  <NumberField name="thresholdUpdate" label="update" form={form} placeholder="0.6" />
+                  <NumberField name="thresholdMerge" label="merge" form={form} placeholder="0.6" />
+                </CollapsibleContent>
+              </Collapsible>
+            )}
 
             <Collapsible open={advancedOpen} onOpenChange={setAdvancedOpen}>
               <CollapsibleTrigger asChild>

package/dist/dashboard-runtime/dashboard/src/components/memory/memory-search-form.tsx

@@ -14,6 +14,7 @@ export interface MemorySearchValues {
   scope: MemoryScope;
   topK: number;
   enableRerank: boolean;
+  minSimilarity: number;
   metadata?: Record<string, string>;
 }
 
@@ -27,6 +28,7 @@ export function MemorySearchForm({ scope, isLoading, onSubmit }: MemorySearchFor
   const [query, setQuery] = useState("");
   const [topK, setTopK] = useState(10);
   const [enableRerank, setEnableRerank] = useState(true);
+  const [minSimilarity, setMinSimilarity] = useState(0);
   const [metadataJson, setMetadataJson] = useState("");
   const [error, setError] = useState<string | null>(null);
 
@@ -47,6 +49,10 @@ export function MemorySearchForm({ scope, isLoading, onSubmit }: MemorySearchFor
       setError("Top K 必须在 0 到 50 之间");
       return;
     }
+    if (minSimilarity < 0 || minSimilarity > 1) {
+      setError("相似度阈值必须在 0 到 1 之间");
+      return;
+    }
 
     try {
       await onSubmit({
@@ -54,6 +60,7 @@ export function MemorySearchForm({ scope, isLoading, onSubmit }: MemorySearchFor
         scope,
         topK,
         enableRerank,
+        minSimilarity,
         metadata: parseJsonObject(metadataJson),
       });
     } catch (err) {
@@ -100,6 +107,21 @@ export function MemorySearchForm({ scope, isLoading, onSubmit }: MemorySearchFor
         </div>
       </div>
 
+      <div className="space-y-2">
+        <Label>相似度阈值</Label>
+        <Input
+          type="number"
+          min={0}
+          max={1}
+          step={0.05}
+          value={minSimilarity}
+          onChange={(event) => setMinSimilarity(Number(event.target.value))}
+        />
+        <p className="text-xs text-muted-foreground">
+          按归一化余弦相似度（0–1）过滤结果，0 表示不过滤。仅返回相似度不低于该值的命中。
+        </p>
+      </div>
+
       <div className="space-y-2">
         <Label>元数据过滤（JSON）</Label>
         <Textarea

package/dist/dashboard-runtime/dashboard/src/components/memory/memory-search-results.tsx

@@ -64,8 +64,11 @@ export function MemorySearchResults({ results, elapsedMs }: MemorySearchResultsP
           <Card key={`${hit.unit.id ?? hit.unit.memoryId ?? index}-${index}`}>
             <CardContent className="space-y-3 p-4">
               <div className="flex flex-wrap items-center gap-2">
+                {hit.similarity !== undefined && (
+                  <Badge variant="secondary">相似度 {hit.similarity.toFixed(4)}</Badge>
+                )}
                 {hit.score !== undefined && (
-                  <Badge variant="secondary">相似度 {hit.score.toFixed(4)}</Badge>
+                  <Badge variant="outline">相关度 {hit.score.toFixed(4)}</Badge>
                 )}
                 {hit.source && <Badge variant="outline">{hit.source}</Badge>}
                 {hit.unit.unit_type && <Badge variant="outline">{hit.unit.unit_type}</Badge>}

package/dist/dashboard-runtime/dashboard/src/components/memory/memory-settings-form.tsx

@@ -9,6 +9,7 @@ import { Textarea } from "@/components/ui/textarea";
 import {
   Form,
   FormControl,
+  FormDescription,
   FormField,
   FormItem,
   FormLabel,
@@ -19,6 +20,11 @@ import type { MemoryStore } from "@/types/memory";
 
 const updateMemoryStoreSchema = z.object({
   description: z.string().trim().max(256, "描述不超过 256 字").optional(),
+  extractInstructions: z
+    .string()
+    .trim()
+    .max(4096, "抽取指令不超过 4096 字")
+    .optional(),
 });
 
 type UpdateMemoryStoreInput = z.infer<typeof updateMemoryStoreSchema>;
@@ -32,14 +38,27 @@ export function MemorySettingsForm({ memoryStore }: MemorySettingsFormProps) {
 
   const form = useForm<UpdateMemoryStoreInput>({
     resolver: zodResolver(updateMemoryStoreSchema),
-    defaultValues: { description: memoryStore.description ?? "" },
+    defaultValues: {
+      description: memoryStore.description ?? "",
+      extractInstructions: memoryStore.extractInstructions ?? "",
+    },
   });
 
   async function onSubmit(values: UpdateMemoryStoreInput) {
     try {
+      // The currently-running server build may predate extractInstructions and
+      // rejects unknown JSON fields. Only include it in the PATCH when the user
+      // actually changed it, so description-only edits keep working. An empty
+      // string is a deliberate "clear" (the field uses pointer/PATCH semantics
+      // server-side once a newer build is deployed).
+      const initialInstructions = memoryStore.extractInstructions ?? "";
+      const nextInstructions = values.extractInstructions ?? "";
+      const instructionsChanged = nextInstructions !== initialInstructions;
+
       await updateStore.mutateAsync({
         memoryStoreName: memoryStore.memoryStoreName,
         description: values.description?.length ? values.description : undefined,
+        ...(instructionsChanged ? { extractInstructions: nextInstructions } : {}),
       });
       toast.success("记忆库设置已更新");
     } catch (error) {
@@ -76,14 +95,40 @@ export function MemorySettingsForm({ memoryStore }: MemorySettingsFormProps) {
                 </FormItem>
               )}
             />
-
-            <div className="flex justify-end border-t border-[hsl(var(--border-light))] pt-4">
-              <Button type="submit" disabled={updateStore.isPending}>
-                {updateStore.isPending ? "保存中..." : "保存更改"}
-              </Button>
-            </div>
           </div>
         </section>
+
+        <section className="rounded-lg border border-[hsl(var(--border))] bg-[hsl(var(--card))] p-5 shadow-sm">
+          <SettingsSectionHeader
+            title="记忆抽取指令"
+            description="可选。这里填写的指令会注入到后台记忆抽取的提示词中，用于引导从对话里提炼哪些信息、忽略哪些信息。留空则使用默认抽取策略。"
+          />
+          <FormField
+            control={form.control}
+            name="extractInstructions"
+            render={({ field }) => (
+              <FormItem>
+                <FormLabel>抽取指令</FormLabel>
+                <FormControl>
+                  <Textarea
+                    placeholder="例如：重点关注用户的医疗术语与用药偏好；忽略寒暄与客套内容。"
+                    className="resize-none"
+                    rows={5}
+                    {...field}
+                  />
+                </FormControl>
+                <FormDescription>最多 4096 个字符。留空表示恢复默认抽取策略。</FormDescription>
+                <FormMessage />
+              </FormItem>
+            )}
+          />
+        </section>
+
+        <div className="flex justify-end border-t border-[hsl(var(--border-light))] pt-4">
+          <Button type="submit" disabled={updateStore.isPending}>
+            {updateStore.isPending ? "保存中..." : "保存更改"}
+          </Button>
+        </div>
       </form>
     </Form>
   );

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

@@ -20,7 +20,7 @@ import {
 import { Skeleton } from "@/components/ui/skeleton";
 import { RelativeTime } from "@/components/shared/relative-time";
 import { useMemoryUnit } from "@/hooks/use-memory-unit";
-import { useMemoryMessages } from "@/hooks/use-memory-messages";
+import { useTaskSourceMessages } from "@/hooks/use-task-source-messages";
 import { cn } from "@/lib/utils";
 import type { MemoryScope, MemoryTask, MemoryTaskStatus } from "@/types/memory";
 
@@ -81,26 +81,6 @@ function formatTime(value: string | undefined): string {
   return new Date(ms).toLocaleString("zh-CN");
 }
 
-/**
- * Compute a tight time window around the task's lifetime so the original-
- * conversation panel only loads messages that plausibly fed this task.
- *
- * The buffers are wide enough to absorb engine-side timestamp skew while
- * still keeping the panel focused on "the messages that this AddMemories
- * call ingested" rather than the entire session history.
- */
-function taskMessageWindow(task: MemoryTask): {
-  minTimestamp?: string;
-  maxTimestamp?: string;
-} {
-  const created = toEpochMs(task.createdAt);
-  const finished = toEpochMs(task.finishedAt) ?? toEpochMs(task.updatedAt);
-  if (created === undefined) return {};
-  const min = new Date(created - 60_000).toISOString();
-  const max = new Date((finished ?? created) + 60_000).toISOString();
-  return { minTimestamp: min, maxTimestamp: max };
-}
-
 export function MemoryTasksTable({
   tasks,
   memoryStoreName,
@@ -286,21 +266,10 @@ function OriginalMessagesPanel({
   task: MemoryTask;
   memoryStoreName: string;
 }) {
-  const { minTimestamp, maxTimestamp } = taskMessageWindow(task);
-  const query = useMemoryMessages(
+  const { messages, isLoading, precise } = useTaskSourceMessages(
     memoryStoreName,
-    task.scope,
-    undefined,
-    50,
-    Boolean(memoryStoreName && task.scope?.appId),
-    { minTimestamp, maxTimestamp },
+    task,
   );
-
-  const messages = query.data?.items ?? [];
-  // Engine writes both the original turn and any system events in the same
-  // table, so on noisy stores this list can have non-conversational rows;
-  // we render whatever has a `content` string and let the styling do the
-  // rest. The window is already narrowed by createdAt±60s so it's small.
   const renderable = messages.filter(
     (m) => typeof m.content === "string" && m.content.length > 0,
   );
@@ -310,17 +279,19 @@ function OriginalMessagesPanel({
       <header className="mb-2 flex items-center justify-between">
         <h4 className="text-sm font-semibold">原始会话</h4>
         <span className="text-[10px] text-muted-foreground">
-          任务时间窗口内的消息
+          {precise ? "该任务提取记忆所依据的消息" : "任务时间窗口内的消息"}
         </span>
       </header>
-      {query.isLoading ? (
+      {isLoading ? (
         <div className="space-y-2">
           <Skeleton className="h-12 w-full" />
           <Skeleton className="h-12 w-full" />
         </div>
       ) : renderable.length === 0 ? (
         <p className="rounded border border-dashed border-[hsl(var(--border))] py-4 text-center text-xs italic text-muted-foreground">
-          未在该时间窗口内找到原始会话消息
+          {precise
+            ? "未找到该任务对应的原始消息（可能已被清理）"
+            : "未在该任务范围内找到原始会话消息"}
         </p>
       ) : (
         <ul className="max-h-[420px] space-y-2 overflow-auto pr-1">

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

@@ -30,7 +30,7 @@ import {
   DialogTitle,
 } from "@/components/ui/dialog";
 import { ConfirmDialog } from "@/components/shared/confirm-dialog";
-import { cn } from "@/lib/utils";
+import { cn, extractFileName } from "@/lib/utils";
 import { IMAGE_FALLBACK_DATA_PREFIX } from "@/lib/constants";
 import type {
   ChatMessage,
@@ -927,8 +927,7 @@ function GroupedChunkList({
 function ChunkCard({ chunk, index }: { chunk: RetrievedChunk; index: number }) {
   const [zoomedImage, setZoomedImage] = useState<string | null>(null);
   const [expanded, setExpanded] = useState(false);
-  const fileName =
-    chunk.ossKey?.split("/").pop() || chunk.ossKey || "未知文件";
+  const fileName = chunk.ossKey ? extractFileName(chunk.ossKey) : "未知文件";
 
   const isImageChunk = chunk.chunkType?.toLowerCase() === "image";
   const imageSrc = chunk.imageBase64

package/dist/dashboard-runtime/dashboard/src/components/retrieval/retrieval-result-card.tsx

@@ -4,6 +4,7 @@ import { Badge } from "@/components/ui/badge";
 import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
 import type { RetrievalResult } from "@/types/retrieval";
 import { IMAGE_FALLBACK_DATA_PREFIX } from "@/lib/constants";
+import { extractFileName } from "@/lib/utils";
 
 interface RetrievalResultCardProps {
   result: RetrievalResult;
@@ -12,7 +13,7 @@ interface RetrievalResultCardProps {
 
 export function RetrievalResultCard({ result, rank }: RetrievalResultCardProps) {
   const [zoomedImage, setZoomedImage] = useState<string | null>(null);
-  const fileName = result.ossKey?.split("/").pop() || result.ossKey || "未知文件";
+  const fileName = result.ossKey ? extractFileName(result.ossKey) : "未知文件";
 
   const isImageChunk = result.chunkType?.toLowerCase() === "image";
   const chunkBadgeVariant = isImageChunk ? "default" : "secondary";