@tablestore/tablestore-agent-cli 0.2.1 → 0.2.4

package/dist/dashboard-runtime/dashboard/src/components/knowledge-base/kb-settings-form.tsx

@@ -1,9 +1,11 @@
 'use client';
 
+import { useState } from "react";
 import { useForm } from "react-hook-form";
 import { zodResolver } from "@hookform/resolvers/zod";
 import { toast } from "sonner";
-import { Check } from "lucide-react";
+import { Check, ChevronRight } from "lucide-react";
+import { cn } from "@/lib/utils";
 import { handleNumberInput } from "@/lib/number-input-handler";
 import { Button } from "@/components/ui/button";
 import { Input } from "@/components/ui/input";
@@ -85,6 +87,9 @@ export function KBSettingsForm({ knowledgeBase }: KBSettingsFormProps) {
     },
   });
 
+  const [retrievalOpen, setRetrievalOpen] = useState(false);
+  const [rerankOpen, setRerankOpen] = useState(false);
+
   // Highlight the parameter card matching the currently selected strategy
   // so users can see at a glance which block their toggle affects.
   const rerankType = form.watch("retrievalConfiguration.rerankingConfiguration.type");
@@ -96,11 +101,21 @@ export function KBSettingsForm({ knowledgeBase }: KBSettingsFormProps) {
 
   async function onSubmit(values: UpdateKBInput) {
     try {
-      // Explicitly construct the request to align Zod output with API types
+      // Explicitly map the nested form fields (modelConfiguration.provider/model)
+      // to the flat fields (modelProvider/modelName) that
+      // buildKnowledgeBaseRetrievalConfiguration reads first; otherwise the
+      // user-selected rerank model silently falls back to the default and
+      // the update appears successful but has no effect.
+      const rc = values.retrievalConfiguration;
+      const rerank = rc?.rerankingConfiguration;
       const request: Parameters<typeof updateKB.mutateAsync>[0] = {
         ...values,
-        retrievalConfiguration: values.retrievalConfiguration
-          ? buildKnowledgeBaseRetrievalConfiguration(values.retrievalConfiguration)
+        retrievalConfiguration: rc
+          ? buildKnowledgeBaseRetrievalConfiguration({
+              ...rc,
+              modelProvider: rerank?.modelConfiguration?.provider,
+              modelName: rerank?.modelConfiguration?.model,
+            })
           : undefined,
       };
       await updateKB.mutateAsync(request);
@@ -169,292 +184,316 @@ export function KBSettingsForm({ knowledgeBase }: KBSettingsFormProps) {
               description="控制默认召回方式、返回规模和 Rerank 策略。"
             />
             <div className="space-y-4">
-              <FormField
-                control={form.control}
-                name="retrievalConfiguration.searchType"
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>搜索类型</FormLabel>
-                    <FormControl>
-                      <ToggleGroup
-                        type="multiple"
-                        value={field.value ?? []}
-                        onValueChange={(v) => {
-                          // Require at least one type to keep the form valid.
-                          if (v?.length) field.onChange(v);
-                        }}
-                        className="grid w-full grid-cols-2 gap-0 overflow-hidden rounded-lg border border-border bg-[hsl(var(--muted)/0.4)] p-0 sm:max-w-md"
-                      >
-                        <ToggleGroupItem value="DENSE_VECTOR" className="h-9 gap-1.5">
-                          {field.value?.includes("DENSE_VECTOR") && (
-                            <Check className="h-3.5 w-3.5 text-[hsl(var(--primary))]" />
-                          )}
-                          向量检索
-                        </ToggleGroupItem>
-                        <ToggleGroupItem value="FULL_TEXT" className="h-9 gap-1.5">
-                          {field.value?.includes("FULL_TEXT") && (
-                            <Check className="h-3.5 w-3.5 text-[hsl(var(--primary))]" />
-                          )}
-                          全文检索
-                        </ToggleGroupItem>
-                      </ToggleGroup>
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <div className="space-y-3">
-                <div className="space-y-1">
-                  <p className="text-sm font-medium text-[hsl(var(--foreground))]">
-                    返回数量
-                  </p>
-                  <p className="text-xs text-[hsl(var(--muted-foreground))]">
-                    控制每路召回参与排序的候选数，以及 Rerank 后实际返回的片段数。
-                  </p>
-                </div>
-                <div className="grid grid-cols-[repeat(auto-fit,minmax(min(260px,100%),1fr))] gap-3">
-                  <RetrievalResultCountField
-                    control={form.control}
-                    name="retrievalConfiguration.denseVectorSearchConfiguration.numberOfResults"
-                    label="向量召回数"
-                    description="向量检索参与后续排序的候选片段数。"
-                    max={100}
-                    muted={
-                      !form
-                        .watch("retrievalConfiguration.searchType")
-                        ?.includes("DENSE_VECTOR")
-                    }
-                  />
-                  <RetrievalResultCountField
-                    control={form.control}
-                    name="retrievalConfiguration.fullTextSearchConfiguration.numberOfResults"
-                    label="全文召回数"
-                    description="全文检索参与后续排序的候选片段数。"
-                    max={100}
-                    muted={
-                      !form
-                        .watch("retrievalConfiguration.searchType")
-                        ?.includes("FULL_TEXT")
-                    }
-                  />
-                </div>
-                <RetrievalResultCountField
-                  control={form.control}
-                  name="retrievalConfiguration.rerankingConfiguration.numberOfResults"
-                  label="重排序返回数"
-                  description="Rerank 后保留并返回的片段数。"
-                  max={200}
-                />
-              </div>
+              {/* 面板 1: 检索参数 */}
+              <div>
+                <button
+                  type="button"
+                  onClick={() => setRetrievalOpen(!retrievalOpen)}
+                  className="flex items-center gap-2 w-full py-2 text-sm font-medium text-foreground hover:text-primary transition-colors"
+                >
+                  <ChevronRight className={cn("h-4 w-4 transition-transform", retrievalOpen && "rotate-90")} />
+                  检索参数
+                </button>
+                {retrievalOpen && (
+                  <div className="pl-6 space-y-4">
+                    <FormField
+                      control={form.control}
+                      name="retrievalConfiguration.searchType"
+                      render={({ field }) => (
+                        <FormItem>
+                          <FormLabel>搜索类型</FormLabel>
+                          <FormControl>
+                            <ToggleGroup
+                              type="multiple"
+                              value={field.value ?? []}
+                              onValueChange={(v) => {
+                                // Require at least one type to keep the form valid.
+                                if (v?.length) field.onChange(v);
+                              }}
+                              className="grid w-full grid-cols-2 gap-0 overflow-hidden rounded-lg border border-border bg-[hsl(var(--muted)/0.4)] p-0 sm:max-w-md"
+                            >
+                              <ToggleGroupItem value="DENSE_VECTOR" className="h-9 gap-1.5">
+                                {field.value?.includes("DENSE_VECTOR") && (
+                                  <Check className="h-3.5 w-3.5 text-[hsl(var(--primary))]" />
+                                )}
+                                向量检索
+                              </ToggleGroupItem>
+                              <ToggleGroupItem value="FULL_TEXT" className="h-9 gap-1.5">
+                                {field.value?.includes("FULL_TEXT") && (
+                                  <Check className="h-3.5 w-3.5 text-[hsl(var(--primary))]" />
+                                )}
+                                全文检索
+                              </ToggleGroupItem>
+                            </ToggleGroup>
+                          </FormControl>
+                          <FormMessage />
+                        </FormItem>
+                      )}
+                    />
 
-              <FormField
-                control={form.control}
-                name="retrievalConfiguration.rerankingConfiguration.type"
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Rerank 策略</FormLabel>
-                    <FormControl>
-                      <ToggleGroup
-                        type="single"
-                        value={field.value ?? ""}
-                        onValueChange={(v) => {
-                          // Radix emits "" when clicking the active item;
-                          // keep the selection stable.
-                          if (v) field.onChange(v);
-                        }}
-                        className="grid w-full grid-cols-3 gap-0 overflow-hidden rounded-lg border border-border bg-[hsl(var(--muted)/0.4)] p-0 sm:max-w-md"
-                      >
-                        <ToggleGroupItem value="MODEL" className="h-9">
-                          Model
-                        </ToggleGroupItem>
-                        <ToggleGroupItem value="WEIGHT" className="h-9">
-                          Weight
-                        </ToggleGroupItem>
-                        <ToggleGroupItem value="RRF" className="h-9">
-                          RRF
-                        </ToggleGroupItem>
-                      </ToggleGroup>
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
+                    <div className="grid grid-cols-[repeat(auto-fit,minmax(min(260px,100%),1fr))] gap-3">
+                      <RetrievalResultCountField
+                        control={form.control}
+                        name="retrievalConfiguration.denseVectorSearchConfiguration.numberOfResults"
+                        label="向量召回数"
+                        description="向量检索参与后续排序的候选片段数。"
+                        max={100}
+                        muted={
+                          !form
+                            .watch("retrievalConfiguration.searchType")
+                            ?.includes("DENSE_VECTOR")
+                        }
+                      />
+                      <RetrievalResultCountField
+                        control={form.control}
+                        name="retrievalConfiguration.fullTextSearchConfiguration.numberOfResults"
+                        label="全文召回数"
+                        description="全文检索参与后续排序的候选片段数。"
+                        max={100}
+                        muted={
+                          !form
+                            .watch("retrievalConfiguration.searchType")
+                            ?.includes("FULL_TEXT")
+                        }
+                      />
+                    </div>
+                  </div>
                 )}
-              />
-
-              {/* RRF 配置参数 */}
-              <div className={cardClass(rerankType === "RRF")}>
-                <p className="text-sm font-medium text-[hsl(var(--muted-foreground))]">RRF 参数</p>
-                <div className="grid grid-cols-3 gap-4">
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.rrfConfiguration.denseVectorSearchWeight"
-                    render={({ field }) => (
-                      <FormItem>
-                        <FormLabel>向量检索权重</FormLabel>
-                        <FormControl>
-                          <Input
-                            type="number"
-                            step="0.1"
-                            placeholder="0.6"
-                            {...field}
-                            value={field.value ?? ""}
-                            onChange={handleNumberInput(field.onChange)}
-                          />
-                        </FormControl>
-                        <FormMessage />
-                      </FormItem>
-                    )}
-                  />
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.rrfConfiguration.fullTextSearchWeight"
-                    render={({ field }) => (
-                      <FormItem>
-                        <FormLabel>全文检索权重</FormLabel>
-                        <FormControl>
-                          <Input
-                            type="number"
-                            step="0.1"
-                            placeholder="0.4"
-                            {...field}
-                            value={field.value ?? ""}
-                            onChange={handleNumberInput(field.onChange)}
-                          />
-                        </FormControl>
-                        <FormMessage />
-                      </FormItem>
-                    )}
-                  />
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.rrfConfiguration.k"
-                    render={({ field }) => (
-                      <FormItem>
-                        <FormLabel>K 值</FormLabel>
-                        <FormControl>
-                          <Input
-                            type="number"
-                            placeholder="60"
-                            {...field}
-                            value={field.value ?? ""}
-                            onChange={handleNumberInput(field.onChange)}
-                          />
-                        </FormControl>
-                        <FormMessage />
-                      </FormItem>
-                    )}
-                  />
-                </div>
               </div>
 
-              {/* Weight 配置参数 */}
-              <div className={cardClass(rerankType === "WEIGHT")}>
-                <p className="text-sm font-medium text-[hsl(var(--muted-foreground))]">Weight 参数</p>
-                <div className="grid grid-cols-2 gap-4">
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.weightConfiguration.denseVectorSearchWeight"
-                    render={({ field }) => (
-                      <FormItem>
-                        <FormLabel>向量检索权重</FormLabel>
-                        <FormControl>
-                          <Input
-                            type="number"
-                            step="0.1"
-                            min="0"
-                            max="1"
-                            placeholder="0.7"
-                            {...field}
-                            value={field.value ?? ""}
-                            onChange={handleNumberInput(field.onChange)}
-                          />
-                        </FormControl>
-                        <FormMessage />
-                      </FormItem>
-                    )}
-                  />
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.weightConfiguration.fullTextSearchWeight"
-                    render={({ field }) => (
-                      <FormItem>
-                        <FormLabel>全文检索权重</FormLabel>
-                        <FormControl>
-                          <Input
-                            type="number"
-                            step="0.1"
-                            min="0"
-                            max="1"
-                            placeholder="0.3"
-                            {...field}
-                            value={field.value ?? ""}
-                            onChange={handleNumberInput(field.onChange)}
-                          />
-                        </FormControl>
-                        <FormMessage />
-                      </FormItem>
-                    )}
-                  />
-                </div>
-              </div>
-
-              {/* Model 配置参数 */}
-              <div className={cardClass(rerankType === "MODEL")}>
-                <p className="text-sm font-medium text-[hsl(var(--muted-foreground))]">Model 参数</p>
-                <div className="grid grid-cols-2 gap-4">
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.modelConfiguration.provider"
-                    render={({ field }) => (
-                      <FormItem>
-                        <FormLabel>提供方</FormLabel>
-                        <Select onValueChange={field.onChange} value={field.value ?? ""}>
-                          <FormControl>
-                            <SelectTrigger>
-                              <SelectValue placeholder="选择提供方" />
-                            </SelectTrigger>
-                          </FormControl>
-                          <SelectContent>
-                            <SelectItem value="bailian">百炼 (Bailian)</SelectItem>
-                          </SelectContent>
-                        </Select>
-                        <FormMessage />
-                      </FormItem>
-                    )}
-                  />
-                  <FormField
-                    control={form.control}
-                    name="retrievalConfiguration.rerankingConfiguration.modelConfiguration.model"
-                    render={({ field }) => {
-                      const rerankProvider = form.watch("retrievalConfiguration.rerankingConfiguration.modelConfiguration.provider");
-                      return (
+              {/* 面板 2: Rerank 参数 */}
+              <div>
+                <button
+                  type="button"
+                  onClick={() => setRerankOpen(!rerankOpen)}
+                  className="flex items-center gap-2 w-full py-2 text-sm font-medium text-foreground hover:text-primary transition-colors"
+                >
+                  <ChevronRight className={cn("h-4 w-4 transition-transform", rerankOpen && "rotate-90")} />
+                  Rerank 参数
+                </button>
+                {rerankOpen && (
+                  <div className="pl-6 space-y-4">
+                    <FormField
+                      control={form.control}
+                      name="retrievalConfiguration.rerankingConfiguration.type"
+                      render={({ field }) => (
                         <FormItem>
-                          <FormLabel>模型名称</FormLabel>
+                          <FormLabel>Rerank 策略</FormLabel>
                           <FormControl>
-                            {rerankProvider === "bailian" ? (
+                            <ToggleGroup
+                              type="single"
+                              value={field.value ?? ""}
+                              onValueChange={(v) => {
+                                // Radix emits "" when clicking the active item;
+                                // keep the selection stable.
+                                if (v) field.onChange(v);
+                              }}
+                              className="grid w-full grid-cols-3 gap-0 overflow-hidden rounded-lg border border-border bg-[hsl(var(--muted)/0.4)] p-0 sm:max-w-md"
+                            >
+                              <ToggleGroupItem value="MODEL" className="h-9">
+                                Model
+                              </ToggleGroupItem>
+                              <ToggleGroupItem value="WEIGHT" className="h-9">
+                                Weight
+                              </ToggleGroupItem>
+                              <ToggleGroupItem value="RRF" className="h-9">
+                                RRF
+                              </ToggleGroupItem>
+                            </ToggleGroup>
+                          </FormControl>
+                          <FormMessage />
+                        </FormItem>
+                      )}
+                    />
+
+                    <RetrievalResultCountField
+                      control={form.control}
+                      name="retrievalConfiguration.rerankingConfiguration.numberOfResults"
+                      label="重排序返回数"
+                      description="Rerank 后保留并返回的片段数。"
+                      max={200}
+                    />
+
+                    {/* Model 配置参数 */}
+                    <div className={cardClass(rerankType === "MODEL")}>
+                      <p className="text-sm font-medium text-[hsl(var(--muted-foreground))]">Model 参数</p>
+                      <div className="grid grid-cols-2 gap-4">
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.modelConfiguration.provider"
+                          render={({ field }) => (
+                            <FormItem>
+                              <FormLabel>提供方</FormLabel>
                               <Select onValueChange={field.onChange} value={field.value ?? ""}>
-                                <SelectTrigger>
-                                  <SelectValue placeholder="选择模型" />
-                                </SelectTrigger>
+                                <FormControl>
+                                  <SelectTrigger>
+                                    <SelectValue placeholder="选择提供方" />
+                                  </SelectTrigger>
+                                </FormControl>
                                 <SelectContent>
-                                  <SelectItem value="gte-rerank-v2">gte-rerank-v2</SelectItem>
-                                  <SelectItem value="qwen3-rerank">qwen3-rerank</SelectItem>
+                                  <SelectItem value="bailian">百炼 (Bailian)</SelectItem>
                                 </SelectContent>
                               </Select>
-                            ) : (
-                              <Input
-                                placeholder="输入模型名称"
-                                {...field}
-                                value={field.value ?? ""}
-                              />
-                            )}
-                          </FormControl>
-                          <FormMessage />
-                        </FormItem>
-                      );
-                    }}
-                  />
-                </div>
+                              <FormMessage />
+                            </FormItem>
+                          )}
+                        />
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.modelConfiguration.model"
+                          render={({ field }) => {
+                            const rerankProvider = form.watch("retrievalConfiguration.rerankingConfiguration.modelConfiguration.provider");
+                            return (
+                              <FormItem>
+                                <FormLabel>模型名称</FormLabel>
+                                <FormControl>
+                                  {rerankProvider === "bailian" ? (
+                                    <Select onValueChange={field.onChange} value={field.value ?? ""}>
+                                      <SelectTrigger>
+                                        <SelectValue placeholder="选择模型" />
+                                      </SelectTrigger>
+                                      <SelectContent>
+                                        <SelectItem value="gte-rerank-v2">gte-rerank-v2</SelectItem>
+                                        <SelectItem value="qwen3-rerank">qwen3-rerank</SelectItem>
+                                        <SelectItem value="qwen3-vl-rerank">
+                                          qwen3-vl-rerank (多模态)
+                                        </SelectItem>
+                                      </SelectContent>
+                                    </Select>
+                                  ) : (
+                                    <Input
+                                      placeholder="输入模型名称"
+                                      {...field}
+                                      value={field.value ?? ""}
+                                    />
+                                  )}
+                                </FormControl>
+                                <FormMessage />
+                              </FormItem>
+                            );
+                          }}
+                        />
+                      </div>
+                    </div>
+
+                    {/* Weight 配置参数 */}
+                    <div className={cardClass(rerankType === "WEIGHT")}>
+                      <p className="text-sm font-medium text-[hsl(var(--muted-foreground))]">Weight 参数</p>
+                      <div className="grid grid-cols-2 gap-4">
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.weightConfiguration.denseVectorSearchWeight"
+                          render={({ field }) => (
+                            <FormItem>
+                              <FormLabel>向量检索权重</FormLabel>
+                              <FormControl>
+                                <Input
+                                  type="number"
+                                  step="0.1"
+                                  min="0"
+                                  max="1"
+                                  placeholder="0.7"
+                                  {...field}
+                                  value={field.value ?? ""}
+                                  onChange={handleNumberInput(field.onChange)}
+                                />
+                              </FormControl>
+                              <FormMessage />
+                            </FormItem>
+                          )}
+                        />
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.weightConfiguration.fullTextSearchWeight"
+                          render={({ field }) => (
+                            <FormItem>
+                              <FormLabel>全文检索权重</FormLabel>
+                              <FormControl>
+                                <Input
+                                  type="number"
+                                  step="0.1"
+                                  min="0"
+                                  max="1"
+                                  placeholder="0.3"
+                                  {...field}
+                                  value={field.value ?? ""}
+                                  onChange={handleNumberInput(field.onChange)}
+                                />
+                              </FormControl>
+                              <FormMessage />
+                            </FormItem>
+                          )}
+                        />
+                      </div>
+                    </div>
+
+                    {/* RRF 配置参数 */}
+                    <div className={cardClass(rerankType === "RRF")}>
+                      <p className="text-sm font-medium text-[hsl(var(--muted-foreground))]">RRF 参数</p>
+                      <div className="grid grid-cols-3 gap-4">
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.rrfConfiguration.denseVectorSearchWeight"
+                          render={({ field }) => (
+                            <FormItem>
+                              <FormLabel>向量检索权重</FormLabel>
+                              <FormControl>
+                                <Input
+                                  type="number"
+                                  step="0.1"
+                                  placeholder="0.6"
+                                  {...field}
+                                  value={field.value ?? ""}
+                                  onChange={handleNumberInput(field.onChange)}
+                                />
+                              </FormControl>
+                              <FormMessage />
+                            </FormItem>
+                          )}
+                        />
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.rrfConfiguration.fullTextSearchWeight"
+                          render={({ field }) => (
+                            <FormItem>
+                              <FormLabel>全文检索权重</FormLabel>
+                              <FormControl>
+                                <Input
+                                  type="number"
+                                  step="0.1"
+                                  placeholder="0.4"
+                                  {...field}
+                                  value={field.value ?? ""}
+                                  onChange={handleNumberInput(field.onChange)}
+                                />
+                              </FormControl>
+                              <FormMessage />
+                            </FormItem>
+                          )}
+                        />
+                        <FormField
+                          control={form.control}
+                          name="retrievalConfiguration.rerankingConfiguration.rrfConfiguration.k"
+                          render={({ field }) => (
+                            <FormItem>
+                              <FormLabel>K 值</FormLabel>
+                              <FormControl>
+                                <Input
+                                  type="number"
+                                  placeholder="60"
+                                  {...field}
+                                  value={field.value ?? ""}
+                                  onChange={handleNumberInput(field.onChange)}
+                                />
+                              </FormControl>
+                              <FormMessage />
+                            </FormItem>
+                          )}
+                        />
+                      </div>
+                    </div>
+                  </div>
+                )}
               </div>
             </div>