@tablestore/tablestore-agent-cli 0.2.0 → 0.2.1

package/dist/dashboard-runtime/dashboard/src/components/documents/oss-browser.tsx

@@ -0,0 +1,264 @@
+"use client";
+
+import { useEffect, useMemo, useState } from "react";
+import {
+  ChevronRight,
+  Database,
+  FileText,
+  Folder,
+  RefreshCw,
+} from "lucide-react";
+
+import { Checkbox } from "@/components/ui/checkbox";
+import { EmptyState } from "@/components/ui/empty-state";
+import { Button } from "@/components/ui/button";
+import { Input } from "@/components/ui/input";
+import { useOssList } from "@/hooks/use-oss-list";
+import {
+  isFileSelected,
+  isFolderSelected,
+  toggleFile,
+  toggleFolder,
+  type OssSelection,
+} from "@/lib/oss-selection";
+import { cn } from "@/lib/utils";
+
+interface OssBrowserProps {
+  selection: OssSelection;
+  onSelectionChange: (next: OssSelection) => void;
+}
+
+const SEARCH_DEBOUNCE_MS = 250;
+
+/** Format a byte count as "1.2 MB" / "344 KB" / "12 B" with one decimal. */
+function formatBytes(n: number): string {
+  if (!Number.isFinite(n) || n <= 0) return "—";
+  const units = ["B", "KB", "MB", "GB"];
+  let i = 0;
+  let v = n;
+  while (v >= 1024 && i < units.length - 1) {
+    v /= 1024;
+    i++;
+  }
+  return `${v.toFixed(i === 0 ? 0 : 1)} ${units[i]}`;
+}
+
+/** Slice a prefix into breadcrumb segments. "" → [], "docs/2026/" → [docs, 2026]. */
+function breadcrumbSegments(prefix: string): { label: string; prefix: string }[] {
+  if (!prefix) return [];
+  const trimmed = prefix.endsWith("/") ? prefix.slice(0, -1) : prefix;
+  const parts = trimmed.split("/");
+  const out: { label: string; prefix: string }[] = [];
+  let acc = "";
+  for (const p of parts) {
+    acc += `${p}/`;
+    out.push({ label: p, prefix: acc });
+  }
+  return out;
+}
+
+/**
+ * Single-level OSS file/folder picker. The component owns navigation
+ * (`prefix`, `search`) and lifts selection up; the parent decides how the
+ * selection turns into a submission.
+ *
+ * Empty / error states render inline so a misconfigured OSS doesn't kick
+ * the user out of the upload dialog — they can switch back to a local tab.
+ */
+export function OssBrowser({ selection, onSelectionChange }: OssBrowserProps) {
+  const [prefix, setPrefix] = useState("");
+  const [searchInput, setSearchInput] = useState("");
+  const [search, setSearch] = useState("");
+
+  // Debounce search to keep keystrokes off the wire.
+  useEffect(() => {
+    const id = setTimeout(() => setSearch(searchInput.trim()), SEARCH_DEBOUNCE_MS);
+    return () => clearTimeout(id);
+  }, [searchInput]);
+
+  const {
+    folders,
+    files,
+    isLoading,
+    isFetchingNextPage,
+    hasNextPage,
+    fetchNextPage,
+    refetch,
+    error,
+    errorCode,
+  } = useOssList({ prefix, search });
+
+  const breadcrumbs = useMemo(() => breadcrumbSegments(prefix), [prefix]);
+
+  // ---------- Empty / error branches ----------
+
+  if (errorCode === "OSS_NOT_CONFIGURED") {
+    return (
+      <EmptyState
+        icon={<Database />}
+        title="尚未配置 OSS"
+        description="请先在「实例配置」中绑定 OSS 端点和 Bucket，再回到此处导入。"
+      />
+    );
+  }
+
+  if (error) {
+    return (
+      <EmptyState
+        icon={<Database />}
+        title={
+          errorCode === "OSS_FORBIDDEN"
+            ? "无访问权限"
+            : errorCode === "OSS_BUCKET_NOT_FOUND"
+              ? "Bucket 不存在"
+              : "OSS 列举失败"
+        }
+        description={error.message}
+        action={
+          <Button
+            type="button"
+            variant="outline"
+            size="sm"
+            onClick={() => refetch()}
+          >
+            <RefreshCw className="mr-1 h-3.5 w-3.5" />
+            重试
+          </Button>
+        }
+      />
+    );
+  }
+
+  // ---------- Main layout ----------
+
+  const totalRowCount = folders.length + files.length;
+
+  return (
+    <div className="rounded-md border border-[hsl(var(--border))] bg-[hsl(var(--surface))]">
+      {/* Breadcrumb + search */}
+      <div className="flex items-center gap-2 border-b border-[hsl(var(--border))] px-3 py-2 text-xs">
+        <button
+          type="button"
+          className="text-[hsl(var(--muted-foreground))] hover:text-[hsl(var(--foreground))]"
+          onClick={() => setPrefix("")}
+        >
+          /
+        </button>
+        {breadcrumbs.map((seg, i) => (
+          <span key={seg.prefix} className="flex items-center gap-1">
+            <span className="text-[hsl(var(--muted-foreground))]">/</span>
+            <button
+              type="button"
+              className={cn(
+                "hover:text-[hsl(var(--foreground))]",
+                i === breadcrumbs.length - 1
+                  ? "text-[hsl(var(--foreground))] font-medium"
+                  : "text-[hsl(var(--muted-foreground))]",
+              )}
+              onClick={() => setPrefix(seg.prefix)}
+            >
+              {seg.label}
+            </button>
+          </span>
+        ))}
+        <Input
+          value={searchInput}
+          onChange={(e) => setSearchInput(e.target.value)}
+          placeholder="搜索当前目录"
+          className="ml-auto h-7 w-44 text-xs"
+        />
+      </div>
+
+      {/* Status bar */}
+      <div className="flex items-center justify-between border-b border-[hsl(var(--border))] px-3 py-1.5 text-xs text-[hsl(var(--muted-foreground))]">
+        <span>{totalRowCount} 个对象</span>
+        <span>已选 {selection.length} 项</span>
+      </div>
+
+      {/* Rows */}
+      <div className="max-h-72 overflow-y-auto">
+        {isLoading ? (
+          <div className="px-3 py-6 text-center text-xs text-[hsl(var(--muted-foreground))]">
+            加载中…
+          </div>
+        ) : totalRowCount === 0 ? (
+          <div className="px-3 py-8 text-center text-xs text-[hsl(var(--muted-foreground))]">
+            当前目录为空
+          </div>
+        ) : (
+          <>
+            {folders.map((f) => {
+              const checked = isFolderSelected(selection, f.prefix);
+              return (
+                <div
+                  key={f.prefix}
+                  className={cn(
+                    "group flex items-center gap-2 px-3 py-1.5 text-sm transition-colors hover:bg-[hsl(var(--accent))] cursor-pointer",
+                    checked && "bg-[hsl(var(--primary-soft))]",
+                  )}
+                  onClick={() => setPrefix(f.prefix)}
+                >
+                  <Checkbox
+                    checked={checked}
+                    onCheckedChange={() =>
+                      onSelectionChange(toggleFolder(selection, { prefix: f.prefix }))
+                    }
+                    onClick={(e) => e.stopPropagation()}
+                  />
+                  <Folder className="h-3.5 w-3.5 shrink-0 text-[hsl(var(--muted-foreground))]" />
+                  <span className="flex-1 truncate">{f.name}</span>
+                  <ChevronRight className="h-3.5 w-3.5 shrink-0 text-[hsl(var(--muted-foreground))] opacity-0 transition-opacity group-hover:opacity-100" />
+                </div>
+              );
+            })}
+            {files.map((f) => {
+              const checked = isFileSelected(selection, f.ossKey);
+              return (
+                <div
+                  key={f.ossKey}
+                  className={cn(
+                    "flex items-center gap-2 px-3 py-1.5 text-sm transition-colors hover:bg-[hsl(var(--accent))] cursor-pointer",
+                    checked && "bg-[hsl(var(--primary-soft))]",
+                  )}
+                  onClick={() =>
+                    onSelectionChange(
+                      toggleFile(selection, { ossKey: f.ossKey, size: f.size }),
+                    )
+                  }
+                >
+                  <Checkbox
+                    checked={checked}
+                    onCheckedChange={() =>
+                      onSelectionChange(
+                        toggleFile(selection, { ossKey: f.ossKey, size: f.size }),
+                      )
+                    }
+                    onClick={(e) => e.stopPropagation()}
+                  />
+                  <FileText className="h-3.5 w-3.5 shrink-0 text-[hsl(var(--muted-foreground))]" />
+                  <span className="flex-1 truncate">{f.name}</span>
+                  <span className="shrink-0 text-xs text-[hsl(var(--muted-foreground))]">
+                    {formatBytes(f.size)}
+                  </span>
+                </div>
+              );
+            })}
+            {hasNextPage && (
+              <div className="px-3 py-2 text-center">
+                <Button
+                  type="button"
+                  variant="ghost"
+                  size="sm"
+                  onClick={fetchNextPage}
+                  disabled={isFetchingNextPage}
+                >
+                  {isFetchingNextPage ? "加载中…" : "加载更多"}
+                </Button>
+              </div>
+            )}
+          </>
+        )}
+      </div>
+    </div>
+  );
+}

package/dist/dashboard-runtime/dashboard/src/components/documents/oss-import-tab.tsx

@@ -0,0 +1,143 @@
+"use client";
+
+import { useMemo } from "react";
+
+import { Input } from "@/components/ui/input";
+import { Label } from "@/components/ui/label";
+import { partitionSelection, type OssSelection } from "@/lib/oss-selection";
+
+import { OssBrowser } from "./oss-browser";
+
+/**
+ * Filter input shape — four free-text comma-separated fields. The translation
+ * to the server's glob form (`prefix*` / `*suffix`) lives in
+ * `@/lib/glob-filter`; this state is the raw text the user typed.
+ */
+export interface OssImportFilterState {
+  prefixIncludeInput: string;
+  prefixExcludeInput: string;
+  suffixIncludeInput: string;
+  suffixExcludeInput: string;
+}
+
+export const EMPTY_OSS_FILTER: OssImportFilterState = {
+  prefixIncludeInput: "",
+  prefixExcludeInput: "",
+  suffixIncludeInput: "",
+  suffixExcludeInput: "",
+};
+
+interface OssImportTabProps {
+  selection: OssSelection;
+  onSelectionChange: (next: OssSelection) => void;
+  filter: OssImportFilterState;
+  onFilterChange: (next: OssImportFilterState) => void;
+}
+
+/**
+ * The "从 OSS 导入" tab body. Hosts the OSS browser, exposes the four
+ * filter knobs (prefix-include / prefix-exclude / suffix-include /
+ * suffix-exclude), and renders a small selection summary.
+ *
+ * Filters apply only to folder selections — the server recurses into a
+ * folder ossKey and applies the include / exclude globs server-side.
+ * Individually-checked files are imported as-is, bypassing filters
+ * (explicit user intent overrides any default rule). The four-field split
+ * is what masks the server's glob constraint that `*` may only appear at
+ * one anchor: `prefix*` or `*suffix` — users type the plain text, the
+ * dialog adds the `*`.
+ */
+export function OssImportTab({
+  selection,
+  onSelectionChange,
+  filter,
+  onFilterChange,
+}: OssImportTabProps) {
+  const { fileKeys, folderPrefixes } = useMemo(
+    () => partitionSelection(selection),
+    [selection],
+  );
+
+  const hasFolderSelection = folderPrefixes.length > 0;
+
+  return (
+    <div className="space-y-3">
+      <OssBrowser selection={selection} onSelectionChange={onSelectionChange} />
+
+      <div
+        className={
+          "rounded-lg border border-[hsl(var(--border))] bg-[hsl(var(--secondary))]/30 p-3 space-y-3 " +
+          (hasFolderSelection ? "" : "opacity-60")
+        }
+        aria-disabled={!hasFolderSelection}
+      >
+        <div className="text-xs font-medium text-foreground">
+          过滤选项
+          <span className="ml-1 font-normal text-[hsl(var(--muted-foreground))]">
+            （仅作用于「选中的文件夹」，单选文件直接导入；服务端递归子目录）
+          </span>
+        </div>
+
+        <div className="grid grid-cols-2 gap-3">
+          <div className="space-y-1">
+            <Label className="text-xs text-muted-foreground">前缀 包含</Label>
+            <Input
+              placeholder="archive/2026/, drafts/"
+              value={filter.prefixIncludeInput}
+              onChange={(e) =>
+                onFilterChange({ ...filter, prefixIncludeInput: e.target.value })
+              }
+              className="h-8 bg-background"
+              disabled={!hasFolderSelection}
+            />
+          </div>
+          <div className="space-y-1">
+            <Label className="text-xs text-muted-foreground">前缀 排除</Label>
+            <Input
+              placeholder="tmp/, _archive/"
+              value={filter.prefixExcludeInput}
+              onChange={(e) =>
+                onFilterChange({ ...filter, prefixExcludeInput: e.target.value })
+              }
+              className="h-8 bg-background"
+              disabled={!hasFolderSelection}
+            />
+          </div>
+          <div className="space-y-1">
+            <Label className="text-xs text-muted-foreground">后缀 包含</Label>
+            <Input
+              placeholder=".md, .pdf, .txt"
+              value={filter.suffixIncludeInput}
+              onChange={(e) =>
+                onFilterChange({ ...filter, suffixIncludeInput: e.target.value })
+              }
+              className="h-8 bg-background"
+              disabled={!hasFolderSelection}
+            />
+          </div>
+          <div className="space-y-1">
+            <Label className="text-xs text-muted-foreground">后缀 排除</Label>
+            <Input
+              placeholder=".log, .tmp"
+              value={filter.suffixExcludeInput}
+              onChange={(e) =>
+                onFilterChange({ ...filter, suffixExcludeInput: e.target.value })
+              }
+              className="h-8 bg-background"
+              disabled={!hasFolderSelection}
+            />
+          </div>
+        </div>
+      </div>
+
+      {selection.length > 0 && (
+        <div className="rounded-md border border-[hsl(var(--border))] bg-[hsl(var(--surface))] px-3 py-2 text-xs text-[hsl(var(--muted-foreground))]">
+          已选 <span className="font-medium text-foreground">{fileKeys.length}</span>{" "}
+          个文件 +{" "}
+          <span className="font-medium text-foreground">{folderPrefixes.length}</span>{" "}
+          个文件夹（文件夹由服务端递归并按上方规则过滤）
+        </div>
+      )}
+    </div>
+  );
+}

package/dist/dashboard-runtime/dashboard/src/components/instance/connect-instance-form.tsx

@@ -219,6 +219,66 @@ export function ConnectInstanceForm({
               )}
             />
 
+            <FormField
+              control={manualForm.control}
+              name="region"
+              render={({ field }) => (
+                <FormItem>
+                  <FormLabel>Region</FormLabel>
+                  {regionMode === "__custom__" ? (
+                    <div className="flex gap-2">
+                      <FormControl>
+                        <Input
+                          {...field}
+                          placeholder="cn-xxx"
+                          disabled={busy}
+                        />
+                      </FormControl>
+                      <Button
+                        type="button"
+                        variant="outline"
+                        size="sm"
+                        onClick={() => {
+                          setRegionMode("cn-hangzhou");
+                          field.onChange("cn-hangzhou");
+                        }}
+                        disabled={busy}
+                      >
+                        使用预设
+                      </Button>
+                    </div>
+                  ) : (
+                    <Select
+                      value={regionMode}
+                      onValueChange={(val) => {
+                        setRegionMode(val);
+                        if (val !== "__custom__") {
+                          field.onChange(val);
+                        } else {
+                          field.onChange("");
+                        }
+                      }}
+                      disabled={busy}
+                    >
+                      <FormControl>
+                        <SelectTrigger>
+                          <SelectValue />
+                        </SelectTrigger>
+                      </FormControl>
+                      <SelectContent>
+                        {REGION_OPTIONS.map((opt) => (
+                          <SelectItem key={opt.value} value={opt.value}>
+                            {opt.label}
+                          </SelectItem>
+                        ))}
+                      </SelectContent>
+                    </Select>
+                  )}
+                  <FormMessage />
+                </FormItem>
+              )}
+            />
+
             <div className="grid gap-4 sm:grid-cols-2">
               <FormField
                 control={manualForm.control}
@@ -341,66 +401,6 @@ export function ConnectInstanceForm({
               />
             </div>
 
-            <FormField
-              control={manualForm.control}
-              name="region"
-              render={({ field }) => (
-                <FormItem>
-                  <FormLabel>Region</FormLabel>
-                  {regionMode === "__custom__" ? (
-                    <div className="flex gap-2">
-                      <FormControl>
-                        <Input
-                          {...field}
-                          placeholder="cn-xxx"
-                          disabled={busy}
-                        />
-                      </FormControl>
-                      <Button
-                        type="button"
-                        variant="outline"
-                        size="sm"
-                        onClick={() => {
-                          setRegionMode("cn-hangzhou");
-                          field.onChange("cn-hangzhou");
-                        }}
-                        disabled={busy}
-                      >
-                        使用预设
-                      </Button>
-                    </div>
-                  ) : (
-                    <Select
-                      value={regionMode}
-                      onValueChange={(val) => {
-                        setRegionMode(val);
-                        if (val !== "__custom__") {
-                          field.onChange(val);
-                        } else {
-                          field.onChange("");
-                        }
-                      }}
-                      disabled={busy}
-                    >
-                      <FormControl>
-                        <SelectTrigger>
-                          <SelectValue />
-                        </SelectTrigger>
-                      </FormControl>
-                      <SelectContent>
-                        {REGION_OPTIONS.map((opt) => (
-                          <SelectItem key={opt.value} value={opt.value}>
-                            {opt.label}
-                          </SelectItem>
-                        ))}
-                      </SelectContent>
-                    </Select>
-                  )}
-                  <FormMessage />
-                </FormItem>
-              )}
-            />
-
             <div className="flex justify-end pt-2">
               <Button type="submit" disabled={busy}>
                 {busy

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

@@ -35,6 +35,8 @@ import { ToggleGroup, ToggleGroupItem } from "@/components/ui/toggle-group";
 import { TagInput } from "@/components/shared/tag-input";
 import { MetadataForm } from "@/components/shared/metadata-form";
 import { EmbeddingConfigForm } from "./embedding-config-form";
+import { RetrievalResultCountField } from "./retrieval-result-count-field";
+import { buildKnowledgeBaseRetrievalConfiguration } from "./knowledge-base-retrieval-config";
 import { useCreateKB } from "@/hooks/use-create-kb";
 import { createKBSchema, type CreateKBInput } from "@/schemas/knowledge-base";
 
@@ -57,6 +59,12 @@ export function KBCreateForm() {
       },
       retrievalConfiguration: {
         searchType: ["DENSE_VECTOR", "FULL_TEXT"],
+        denseVectorSearchConfiguration: {
+          numberOfResults: 20,
+        },
+        fullTextSearchConfiguration: {
+          numberOfResults: 20,
+        },
         rerankingConfiguration: {
           type: "WEIGHT",
           numberOfResults: 20,
@@ -93,16 +101,7 @@ export function KBCreateForm() {
       const request: Parameters<typeof createKB.mutateAsync>[0] = {
         ...values,
         retrievalConfiguration: values.retrievalConfiguration
-          ? {
-              ...values.retrievalConfiguration,
-              rerankingConfiguration: values.retrievalConfiguration.rerankingConfiguration
-                ? {
-                    ...values.retrievalConfiguration.rerankingConfiguration,
-                    numberOfResults:
-                      values.retrievalConfiguration.rerankingConfiguration.numberOfResults ?? 20,
-                  }
-                : undefined,
-            }
+          ? buildKnowledgeBaseRetrievalConfiguration(values.retrievalConfiguration)
           : undefined,
       };
       await createKB.mutateAsync(request);
@@ -278,31 +277,49 @@ export function KBCreateForm() {
                 )}
               />
 
-              <FormField
-                control={form.control}
-                name="retrievalConfiguration.rerankingConfiguration.numberOfResults"
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>返回数量 (Top K)</FormLabel>
-                    <FormControl>
-                      <Input
-                        type="number"
-                        placeholder="20"
-                        {...field}
-                        value={field.value ?? ""}
-                        onChange={(event) =>
-                          field.onChange(
-                            event.target.value
-                              ? Number(event.target.value)
-                              : undefined
-                          )
-                        }
-                      />
-                    </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>
 
               <FormField
                 control={form.control}