feat: add skill and assistant knowledge bases

2026-09-02 00:49:13 +08:00
1 parent dd0ee282df
commit fbd8dec48f
@@ -3,6 +3,7 @@
 import {
  ChevronLeftIcon,
  ChevronRightIcon,
+  DatabaseZap,
  Replace,
  SearchIcon,
  SparklesIcon,
@@ -18,9 +19,11 @@ import {
  WorkspaceHeader,
 } from "@/components/workspace/workspace-container";
 import { SkillRow } from "@/components/workspace/settings/skill-settings-page";
+import { SkillKnowledgePanel } from "@/components/workspace/skills/SkillKnowledgePanel";
 import { getStoredAuth } from "@/core/auth";
 import { useAllSkillsAdmin } from "@/core/skills/hooks";
 import { desensitize } from "@/lib/desensitize";
+import { useSkillKnowledgeBindings } from "@/strategy-components/api/skill-knowledge";
 import { SkillAddressesPanel } from "./SkillAddressesPage";

 const PAGE_SIZE = 10;
@@ -29,6 +32,7 @@ export default function AdminSkillsPage() {
  const [search, setSearch] = useState("");
  const [page, setPage] = useState(1);
  const { skills, isLoading, error } = useAllSkillsAdmin(true);
+  const knowledgeBindings = useSkillKnowledgeBindings();

  const storedAuth = useMemo(() => getStoredAuth(), []);
  const currentUserId = storedAuth?.user_id ?? null;
@@ -71,6 +75,10 @@ export default function AdminSkillsPage() {
                  <Replace />
                  技能地址管理
                </TabsTrigger>
+                <TabsTrigger value="knowledge">
+                  <DatabaseZap />
+                  技能知识归纳
+                </TabsTrigger>
              </TabsList>
            </div>
          </div>
@@ -126,6 +134,9 @@ export default function AdminSkillsPage() {
                    skill={skill}
                    currentUserId={currentUserId}
                    isAdmin
+                    knowledgeBindings={(knowledgeBindings.data ?? []).filter(
+                      (binding) => binding.skill_name === skill.name,
+                    )}
                  />
                ))}
              </>
@@ -164,6 +175,12 @@ export default function AdminSkillsPage() {
          >
            <SkillAddressesPanel />
          </TabsContent>
+          <TabsContent
+            value="knowledge"
+            className="mt-0 min-h-0 flex-1 overflow-hidden"
+          >
+            <SkillKnowledgePanel />
+          </TabsContent>
        </Tabs>
      </WorkspaceBody>
    </WorkspaceContainer>
@@ -0,0 +1,74 @@
+import { Database, FileUp, History, Loader2, RefreshCw, Search, ShieldCheck, Trash2 } from "lucide-react";
+import { useMemo, useRef, useState } from "react";
+import { toast } from "sonner";
+
+import { Badge } from "@/components/ui/badge";
+import { Button } from "@/components/ui/button";
+import { Input } from "@/components/ui/input";
+import { ScrollArea } from "@/components/ui/scroll-area";
+import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
+import { MarkdownContent } from "@/components/workspace/messages/markdown-content";
+import { KnowledgeProductNav } from "@/components/workspace/knowledge-product-nav";
+import { getStoredAuth } from "@/core/auth";
+import {
+  listAssistantWikiRevisions,
+  type AssistantWikiPage,
+  type AssistantWikiRevision,
+  useAssistantImportJobs,
+  useAssistantKnowledgeActions,
+  useAssistantKnowledgeBase,
+  useAssistantKnowledgeSources,
+  useAssistantWikiPages,
+} from "@/strategy-components/api/assistant-knowledge";
+
+export default function AssistantKnowledgePage() {
+  const isAdmin = getStoredAuth()?.system_role === "admin";
+  const [query, setQuery] = useState("");
+  const [selected, setSelected] = useState<AssistantWikiPage | null>(null);
+  const [revisions, setRevisions] = useState<AssistantWikiRevision[]>([]);
+  const fileRef = useRef<HTMLInputElement>(null);
+  const base = useAssistantKnowledgeBase();
+  const pages = useAssistantWikiPages(query);
+  const sources = useAssistantKnowledgeSources();
+  const jobs = useAssistantImportJobs();
+  const actions = useAssistantKnowledgeActions();
+  const activePage = selected ?? pages.data?.[0] ?? null;
+
+  const counts = useMemo(() => ({ pages: pages.data?.length ?? 0, sources: sources.data?.length ?? 0, jobs: jobs.data?.length ?? 0 }), [pages.data, sources.data, jobs.data]);
+
+  if (base.isLoading) return <div className="flex size-full flex-col"><KnowledgeProductNav /><div className="flex flex-1 items-center justify-center"><Loader2 className="animate-spin" /></div></div>;
+  if (!base.data) {
+    return <div className="flex size-full flex-col"><KnowledgeProductNav /><div className="flex flex-1 items-center justify-center p-8"><div className="max-w-lg rounded-xl border bg-card p-8 text-center"><Database className="mx-auto mb-4 size-10 text-primary" /><h1 className="text-xl font-semibold">创建“知识梳理”助手知识库</h1><p className="mt-2 text-sm text-muted-foreground">无需配置普通知识库服务。创建后所有用户可查看 Wiki 与搜索，只有管理员可导入和维护。</p>{isAdmin ? <Button className="mt-5" disabled={actions.initialize.isPending} onClick={() => actions.initialize.mutate("知识梳理")}>创建知识梳理库</Button> : <p className="mt-5 text-sm">请联系管理员初始化。</p>}</div></div></div>;
+  }
+
+  const uploadFiles = async (files: FileList | null) => {
+    if (!files?.length) return;
+    try {
+      await actions.upload.mutateAsync(Array.from(files));
+      toast.success("文件已安全扫描并导入");
+    } catch (error) {
+      toast.error(error instanceof Error ? error.message : String(error));
+    } finally {
+      if (fileRef.current) fileRef.current.value = "";
+    }
+  };
+
+  const openHistory = async () => {
+    if (!activePage) return;
+    setRevisions(await listAssistantWikiRevisions(activePage.slug));
+  };
+
+  return (
+    <div className="flex size-full min-h-0 flex-col bg-background">
+      <KnowledgeProductNav />
+      <header className="flex items-center gap-4 border-b px-6 py-4"><div><h1 className="text-xl font-semibold">{base.data.name}</h1><p className="text-xs text-muted-foreground">全局助手知识库 · 普通用户只读</p></div><Badge variant="outline"><ShieldCheck className="mr-1 size-3" />实体对齐与版本管理</Badge><div className="ml-auto flex gap-2">{isAdmin && <><input ref={fileRef} className="hidden" type="file" multiple onChange={(event) => void uploadFiles(event.target.files)} /><Button variant="outline" onClick={() => fileRef.current?.click()}><FileUp />导入本地文件</Button></>}<Button variant="outline" disabled={!activePage} onClick={() => void openHistory()}><History />页面版本</Button></div></header>
+      <Tabs defaultValue="wiki" className="flex min-h-0 flex-1 flex-col">
+        <div className="flex items-center gap-4 border-b px-6"><TabsList className="my-2"><TabsTrigger value="wiki">Wiki（{counts.pages}）</TabsTrigger><TabsTrigger value="sources">来源（{counts.sources}）</TabsTrigger><TabsTrigger value="jobs">导入记录（{counts.jobs}）</TabsTrigger><TabsTrigger value="revisions">版本（{revisions.length}）</TabsTrigger></TabsList><div className="relative ml-auto w-72"><Search className="absolute left-2.5 top-1/2 size-4 -translate-y-1/2 text-muted-foreground" /><Input value={query} onChange={(event) => setQuery(event.target.value)} placeholder="搜索标题、实体别名和正文" className="pl-8" /></div></div>
+        <TabsContent value="wiki" className="mt-0 grid min-h-0 flex-1 grid-cols-[300px_1fr]"><ScrollArea className="border-r"><div className="space-y-1 p-3">{pages.data?.map((page) => <button key={page.id} type="button" onClick={() => setSelected(page)} className={`w-full rounded-md px-3 py-2 text-left ${activePage?.id === page.id ? "bg-primary/10 text-primary" : "hover:bg-muted"}`}><div className="truncate text-sm font-medium">{page.title}</div><div className="mt-1 flex items-center gap-2 text-[11px] text-muted-foreground"><span>{page.page_type}</span><span>{new Date(page.updated_at).toLocaleString()}</span></div></button>)}</div></ScrollArea><ScrollArea><article className="mx-auto max-w-5xl p-8">{activePage ? <><h1 className="mb-1 text-2xl font-semibold">{activePage.title}</h1><p className="mb-6 text-xs text-muted-foreground">{activePage.slug}</p><MarkdownContent content={activePage.content_markdown} isLoading={false} rehypePlugins={[]} /></> : <p className="text-sm text-muted-foreground">暂无 Wiki 页面。</p>}</article></ScrollArea></TabsContent>
+        <TabsContent value="sources" className="mt-0 min-h-0 flex-1 overflow-auto p-6"><div className="grid gap-3 md:grid-cols-2 xl:grid-cols-3">{sources.data?.map((source) => <div key={source.id} className="rounded-lg border bg-card p-4"><div className="flex items-center gap-2"><Database className="size-4" /><span className="font-medium">{source.source_name}</span><Badge variant="outline" className="ml-auto">{source.source_type}</Badge></div><p className="mt-3 text-xs text-muted-foreground">业务键：{source.source_key}</p>{isAdmin && <div className="mt-4 flex justify-end gap-2"><Button size="sm" variant="outline" disabled={source.source_type === "local_file" || actions.reimportSource.isPending} onClick={() => actions.reimportSource.mutate(source.id, { onSuccess: () => toast.success("已发起重新导入"), onError: (error) => toast.error(error.message) })}><RefreshCw />重新导入</Button><Button size="sm" variant="destructive" disabled={actions.deleteSource.isPending} onClick={() => { if (window.confirm(`确认删除来源“${source.source_name}”的贡献？仍被其他来源引用的知识会保留。`)) actions.deleteSource.mutate(source.id, { onSuccess: () => toast.success("来源贡献已删除"), onError: (error) => toast.error(error.message) }); }}><Trash2 />删除来源</Button></div>}</div>)}</div></TabsContent>
+        <TabsContent value="jobs" className="mt-0 min-h-0 flex-1 overflow-auto p-6"><div className="space-y-3">{jobs.data?.map((job) => <div key={job.id} className="flex items-center gap-3 rounded-lg border bg-card p-4"><Badge>{job.status}</Badge><span className="text-sm">{job.trigger}</span><span className="text-xs text-muted-foreground">{Object.entries(job.counts || {}).map(([key, value]) => `${key} ${value}`).join(" · ")}</span><span className="ml-auto text-xs">{new Date(job.created_at).toLocaleString()}</span></div>)}</div></TabsContent>
+        <TabsContent value="revisions" className="mt-0 min-h-0 flex-1 overflow-auto p-6"><div className="space-y-3">{revisions.map((revision) => <div key={revision.id} className="flex items-center gap-3 rounded-lg border bg-card p-4"><Badge variant="outline">v{revision.revision_no}</Badge><span className="text-sm">{revision.change_type}</span><span className="text-xs text-muted-foreground">{revision.summary}</span><span className="ml-auto text-xs">{new Date(revision.created_at).toLocaleString()}</span>{isAdmin && activePage && revision.id !== activePage.current_revision_id && <Button size="sm" variant="outline" onClick={() => actions.rollback.mutate({ slug: activePage.slug, revisionId: revision.id })}>回滚到此版本</Button>}</div>)}{!revisions.length && <p className="text-sm text-muted-foreground">请先在 Wiki 页选择页面并点击“页面版本”。</p>}</div></TabsContent>
+      </Tabs>
+    </div>
+  );
+}
@@ -3,6 +3,7 @@ import { useState } from "react";
 import { Navigate, useParams, useSearchParams } from "react-router-dom";

 import { Button } from "@/components/ui/button";
+import { KnowledgeProductNav, useKnowledgeBaseProductVersion } from "@/components/workspace/knowledge-product-nav";
 import {
  Dialog,
  DialogContent,
@@ -70,6 +71,7 @@ export default function LlmWikiKnowledgePage({
  const { noteId = "" } = useParams();
  const [searchParams] = useSearchParams();
  const runtime = useLlmWikiRuntime();
+  const productVersion = useKnowledgeBaseProductVersion();
  const wikiSlug = searchParams.get("wikiSlug")?.trim() || "";

  if (mode === "detail" && noteId && wikiSlug) {
@@ -80,11 +82,15 @@ export default function LlmWikiKnowledgePage({
    return <RuntimeLoading />;
  }

+  if (mode === "list" && productVersion.data === "v2") {
+    return <Navigate to="/page/workspace/knowledge/assistant" replace />;
+  }
+
  if (runtime.isError || runtime.data?.provider !== "weknora") {
    if (mode === "graph") {
      return <KnowledgeGraphPage />;
    }
-    return <LegacyKnowledgePage />;
+    return mode === "list" ? <div className="flex size-full min-h-0 flex-col"><KnowledgeProductNav /><div className="min-h-0 flex-1"><LegacyKnowledgePage /></div></div> : <LegacyKnowledgePage />;
  }

  if (mode === "graph") {
@@ -95,5 +101,5 @@ export default function LlmWikiKnowledgePage({
    return <WeKnoraKnowledgeDetailPage />;
  }

-  return <WeKnoraKnowledgeWorkspace />;
+  return <div className="flex size-full min-h-0 flex-col"><KnowledgeProductNav /><div className="min-h-0 flex-1"><WeKnoraKnowledgeWorkspace /></div></div>;
 }
@@ -40,6 +40,7 @@ import AdminSkillsPage from './AdminSkillsPage';
 import ScheduledTaskRunDetailPage from './ScheduledTaskRunDetailPage';
 import ScheduledTasksPage from './ScheduledTasksPage';
 import LlmWikiKnowledgePage from './LlmWikiKnowledgePage';
+import AssistantKnowledgePage from './AssistantKnowledgePage';
 import WeKnoraWikiManagementPage from '@/strategy-components/components/resource-management/llmwiki/WeKnoraWikiManagementPage';
 import WeKnoraWikiPageView from '@/strategy-components/components/resource-management/llmwiki/WeKnoraWikiPageView';
 import MarkdownDocxEditorPage from './MarkdownDocxEditorPage';
@@ -689,6 +690,14 @@ export default function WorkspaceRoutes({
            </WorkspaceLayout>
          }
        />
+        <Route
+          path="knowledge/assistant"
+          element={
+            <WorkspaceLayout {...layoutProps}>
+              <AssistantKnowledgePage />
+            </WorkspaceLayout>
+          }
+        />
        <Route
          path="knowledge/:noteId/wiki/view"
          element={