perf(rag): ui and related apis

2026-06-09 15:26:03 +00:00 · 2025-07-18 00:37:22 +08:00
parent bb672d8f46
commit b61bd36b14
17 changed files with 89 additions and 32 deletions
--- a/web/src/app/home/components/dynamic-form/DynamicFormItemComponent.tsx
+++ b/web/src/app/home/components/dynamic-form/DynamicFormItemComponent.tsx
@@ -51,7 +51,7 @@ export default function DynamicFormItemComponent({
        });
    }
  }, [config.type]);
-  
+
  useEffect(() => {
    if (config.type === DynamicFormItemType.KNOWLEDGE_BASE_SELECTOR) {
      httpClient
@@ -272,28 +272,11 @@ export default function DynamicFormItemComponent({
          </SelectTrigger>
          <SelectContent>
            <SelectGroup>
-              {/* <SelectItem value="">{t('knowledge.empty')}</SelectItem> */}
+              <SelectItem value="__none__">{t('knowledge.empty')}</SelectItem>
              {knowledgeBases.map((base) => (
-                <HoverCard key={base.uuid} openDelay={0} closeDelay={0}>
-                  <HoverCardTrigger asChild>
-                    <SelectItem value={base.uuid ?? ''}>{base.name}</SelectItem>
-                  </HoverCardTrigger>
-                  <HoverCardContent
-                    className="w-80 data-[state=open]:animate-none data-[state=closed]:animate-none"
-                    align="end"
-                    side="right"
-                    sideOffset={10}
-                  >
-                    <div className="space-y-2">
-                      <div className="flex items-center gap-2">
-                        <h4 className="font-medium">{base.name}</h4>
-                      </div>
-                      <p className="text-sm text-muted-foreground">
-                        {base.description}
-                      </p>
-                      </div>
-                  </HoverCardContent>
-                </HoverCard>
+                <SelectItem key={base.uuid} value={base.uuid ?? ''}>
+                  {base.name}
+                </SelectItem>
              ))}
            </SelectGroup>
          </SelectContent>
--- a/web/src/app/home/knowledge/KBDetailDialog.tsx
+++ b/web/src/app/home/knowledge/KBDetailDialog.tsx
@@ -36,6 +36,7 @@ interface KBDetailDialogProps {
  onFormCancel: () => void;
  onKbDeleted: () => void;
  onNewKbCreated: (kbId: string) => void;
+  onKbUpdated: (kbId: string) => void;
 }

 export default function KBDetailDialog({
@@ -46,6 +47,7 @@ export default function KBDetailDialog({
  onFormCancel,
  onKbDeleted,
  onNewKbCreated,
+  onKbUpdated,
 }: KBDetailDialogProps) {
  const { t } = useTranslation();
  const [kbId, setKbId] = useState<string | undefined>(propKbId);
@@ -111,6 +113,7 @@ export default function KBDetailDialog({
                  onFormCancel={onFormCancel}
                  onKbDeleted={onKbDeleted}
                  onNewKbCreated={onNewKbCreated}
+                  onKbUpdated={onKbUpdated}
                />
              )}
              {activeMenu === 'documents' && <div>documents</div>}
@@ -185,6 +188,7 @@ export default function KBDetailDialog({
                    onFormCancel={onFormCancel}
                    onKbDeleted={onKbDeleted}
                    onNewKbCreated={onNewKbCreated}
+                    onKbUpdated={onKbUpdated}
                  />
                )}
                {activeMenu === 'documents' && <KBDoc kbId={kbId} />}
--- a/web/src/app/home/knowledge/components/kb-docs/KBDoc.tsx
+++ b/web/src/app/home/knowledge/components/kb-docs/KBDoc.tsx
@@ -28,7 +28,7 @@ export default function KBDoc({ kbId }: { kbId: string }) {
    setDocumentsList(
      resp.files.map((file: KnowledgeBaseFile) => {
        return {
-          id: file.id,
+          uuid: file.uuid,
          name: file.file_name,
          status: file.status,
        };
--- a/web/src/app/home/knowledge/components/kb-docs/documents/columns.tsx
+++ b/web/src/app/home/knowledge/components/kb-docs/documents/columns.tsx
@@ -14,7 +14,7 @@ import {
 import { useTranslation } from 'react-i18next';

 export type DocumentFile = {
-  id: string;
+  uuid: string;
  name: string;
  status: string;
 };
@@ -52,7 +52,7 @@ export const columns = (
                {t('knowledge.documentsTab.actions')}
              </DropdownMenuLabel>

-              <DropdownMenuItem onClick={() => onDelete(document.id)}>
+              <DropdownMenuItem onClick={() => onDelete(document.uuid)}>
                {t('knowledge.documentsTab.delete')}
              </DropdownMenuItem>
            </DropdownMenuContent>
--- a/web/src/app/home/knowledge/components/kb-form/KBForm.tsx
+++ b/web/src/app/home/knowledge/components/kb-form/KBForm.tsx
@@ -24,6 +24,7 @@ import {
  SelectValue,
 } from '@/components/ui/select';
 import { KnowledgeBase } from '@/app/infra/entities/api';
+import { toast } from 'sonner';

 const getFormSchema = (t: (key: string) => string) =>
  z.object({
@@ -42,6 +43,7 @@ export default function KBForm({
  onFormCancel,
  onKbDeleted,
  onNewKbCreated,
+  onKbUpdated,
 }: {
  initKbId?: string;
  // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -49,6 +51,7 @@ export default function KBForm({
  onFormCancel: () => void;
  onKbDeleted: () => void;
  onNewKbCreated: (kbId: string) => void;
+  onKbUpdated: (kbId: string) => void;
 }) {
  const { t } = useTranslation();
  const formSchema = getFormSchema(t);
@@ -114,6 +117,17 @@ export default function KBForm({
        description: data.description,
        embedding_model_uuid: data.embeddingModelUUID,
      };
+      httpClient
+        .updateKnowledgeBase(initKbId, updateKb)
+        .then((res) => {
+          console.log('update knowledge base success', res);
+          onKbUpdated(res.uuid);
+          toast.success(t('knowledge.updateKnowledgeBaseSuccess'));
+        })
+        .catch((err) => {
+          console.error('update knowledge base failed', err);
+          toast.error(t('knowledge.updateKnowledgeBaseFailed'));
+        });
    } else {
      // create knowledge base
      const newKb: KnowledgeBase = {
@@ -186,6 +200,7 @@ export default function KBForm({
                  <FormControl>
                    <div className="relative">
                      <Select
+                        disabled={!!initKbId}
                        onValueChange={(value) => {
                          field.onChange(value);
                          console.log('value', value);
@@ -210,7 +225,9 @@ export default function KBForm({
                    </div>
                  </FormControl>
                  <FormDescription>
-                    {t('knowledge.embeddingModelDescription')}
+                    {initKbId
+                      ? t('knowledge.cannotChangeEmbeddingModel')
+                      : t('knowledge.embeddingModelDescription')}
                  </FormDescription>
                  <FormMessage />
                </FormItem>
--- a/web/src/app/home/knowledge/page.tsx
+++ b/web/src/app/home/knowledge/page.tsx
@@ -81,6 +81,10 @@ export default function KnowledgePage() {
    setDetailDialogOpen(false);
  };

+  const handleKbUpdated = (kbId: string) => {
+    getKnowledgeBaseList();
+  };
+
  return (
    <div>
      <KBDetailDialog
@@ -91,6 +95,7 @@ export default function KnowledgePage() {
        onFormCancel={handleFormCancel}
        onKbDeleted={handleKbDeleted}
        onNewKbCreated={handleNewKbCreated}
+        onKbUpdated={handleKbUpdated}
      />

      <div className={styles.knowledgeListContainer}>
--- a/web/src/app/infra/entities/api/index.ts
+++ b/web/src/app/infra/entities/api/index.ts
@@ -164,7 +164,7 @@ export interface ApiRespKnowledgeBaseFiles {
 }

 export interface KnowledgeBaseFile {
-  id: string;
+  uuid: string;
  file_name: string;
  status: string;
 }
--- a/web/src/app/infra/http/HttpClient.ts
+++ b/web/src/app/infra/http/HttpClient.ts
@@ -459,6 +459,13 @@ class HttpClient {
    return this.post('/api/v1/knowledge/bases', base);
  }

+  public updateKnowledgeBase(
+    uuid: string,
+    base: KnowledgeBase,
+  ): Promise<{ uuid: string }> {
+    return this.put(`/api/v1/knowledge/bases/${uuid}`, base);
+  }
+
  public uploadKnowledgeBaseFile(
    uuid: string,
    file_id: string,
--- a/web/src/i18n/locales/en-US.ts
+++ b/web/src/i18n/locales/en-US.ts
@@ -253,6 +253,10 @@ const enUS = {
    embeddingModelDescription:
      'Used to vectorize the text, you can configure it in the Models page',
    updateTime: 'Updated ',
+    cannotChangeEmbeddingModel:
+      'Knowledge base created cannot be modified embedding model',
+    updateKnowledgeBaseSuccess: 'Knowledge base updated successfully',
+    updateKnowledgeBaseFailed: 'Knowledge base update failed',
    documentsTab: {
      name: 'Name',
      status: 'Status',
--- a/web/src/i18n/locales/ja-JP.ts
+++ b/web/src/i18n/locales/ja-JP.ts
@@ -255,6 +255,10 @@ const jaJP = {
    embeddingModelDescription:
      'テキストのベクトル化に使用する埋め込みモデルを管理します',
    updateTime: '更新日時',
+    cannotChangeEmbeddingModel:
+      '知識ベース作成後は埋め込みモデルを変更できません',
+    updateKnowledgeBaseSuccess: '知識ベースの更新に成功しました',
+    updateKnowledgeBaseFailed: '知識ベースの更新に失敗しました',
    documentsTab: {
      name: '名前',
      status: 'ステータス',
--- a/web/src/i18n/locales/zh-Hans.ts
+++ b/web/src/i18n/locales/zh-Hans.ts
@@ -247,6 +247,9 @@ const zhHans = {
    selectEmbeddingModel: '选择嵌入模型',
    embeddingModelDescription: '用于向量化文本，可在模型配置页面配置',
    updateTime: '更新于',
+    cannotChangeEmbeddingModel: '知识库创建后不可修改嵌入模型',
+    updateKnowledgeBaseSuccess: '知识库更新成功',
+    updateKnowledgeBaseFailed: '知识库更新失败',
    documentsTab: {
      name: '名称',
      status: '状态',