Feat: Add keyword item to AssistantSetting infiniflow#4543 (infiniflo…

…w#4603) ### What problem does this PR solve? Feat: Add keyword item to AssistantSetting infiniflow#4543 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
cike8899 · Jan 23, 2025 · b23a4a8 · b23a4a8
1 parent 240e7d7
commit b23a4a8
Show file tree

Hide file tree

Showing 6 changed files with 38 additions and 1 deletion.
diff --git a/web/src/components/knowledge-base-item.tsx b/web/src/components/knowledge-base-item.tsx
@@ -1,3 +1,4 @@
+import { DocumentParserType } from '@/constants/knowledge';
 import { useTranslate } from '@/hooks/common-hooks';
 import { useFetchKnowledgeList } from '@/hooks/knowledge-hooks';
 import { UserOutlined } from '@ant-design/icons';
@@ -8,7 +9,11 @@ const KnowledgeBaseItem = () => {
 
   const { list: knowledgeList } = useFetchKnowledgeList(true);
 
-  const knowledgeOptions = knowledgeList.map((x) => ({
+  const filteredKnowledgeList = knowledgeList.filter(
+    (x) => x.parser_id !== DocumentParserType.Tag,
+  );
+
+  const knowledgeOptions = filteredKnowledgeList.map((x) => ({
     label: (
       <Space>
         <Avatar size={20} icon={<UserOutlined />} src={x.avatar} />

diff --git a/web/src/constants/knowledge.ts b/web/src/constants/knowledge.ts
@@ -63,3 +63,20 @@ export enum DocumentType {
   Virtual = 'virtual',
   Visual = 'visual',
 }
+
+export enum DocumentParserType {
+  Naive = 'naive',
+  Qa = 'qa',
+  Resume = 'resume',
+  Manual = 'manual',
+  Table = 'table',
+  Paper = 'paper',
+  Book = 'book',
+  Laws = 'laws',
+  Presentation = 'presentation',
+  Picture = 'picture',
+  One = 'one',
+  Audio = 'audio',
+  Email = 'email',
+  Tag = 'tag',
+}
diff --git a/web/src/locales/en.ts b/web/src/locales/en.ts
@@ -520,6 +520,8 @@ This procedure will improve precision of retrieval by adding more information to
       useKnowledgeGraph: 'Use knowledge graph',
       useKnowledgeGraphTip:
         'It will retrieve descriptions of relevant entities,relations and community reports, which will enhance inference of multi-hop and complex question.',
+      keyword: 'Keyword analysis',
+      keywordTip: `Apply LLM to analyze user's questions, extract keywords which will be emphesize during the relevance omputation.`,
     },
     setting: {
       profile: 'Profile',

diff --git a/web/src/locales/zh-traditional.ts b/web/src/locales/zh-traditional.ts
@@ -504,6 +504,8 @@ export default {
       useKnowledgeGraph: '使用知識圖譜',
       useKnowledgeGraph提示:
         '它將檢索相關實體、關係和社區報告的描述，這將增強多跳和複雜問題的推理。',
+      keyword: '關鍵字分析',
+      keywordTip: `應用LLM分析使用者的問題，提取在相關性計算中需要強調的關鍵字。`,
     },
     setting: {
       profile: '概述',

diff --git a/web/src/locales/zh.ts b/web/src/locales/zh.ts
@@ -521,6 +521,8 @@ General：实体和关系提取提示来自 GitHub - microsoft/graphrag：基于
       useKnowledgeGraph: '使用知识图谱',
       useKnowledgeGraphTip:
         '它将检索相关实体、关系和社区报告的描述，这将增强多跳和复杂问题的推理。',
+      keyword: '关键词分析',
+      keywordTip: `应用 LLM 分析用户的问题，提取在相关性计算中要强调的关键词。`,
     },
     setting: {
       profile: '概要',

diff --git a/web/src/pages/chat/chat-configuration-modal/assistant-setting.tsx b/web/src/pages/chat/chat-configuration-modal/assistant-setting.tsx
@@ -107,6 +107,15 @@ const AssistantSetting = ({ show, form }: ISegmentedContentProps) => {
       >
         <Switch />
       </Form.Item>
+      <Form.Item
+        label={t('keyword')}
+        valuePropName="checked"
+        name={['prompt_config', 'keyword']}
+        tooltip={t('keywordTip')}
+        initialValue={false}
+      >
+        <Switch />
+      </Form.Item>
       {/* <Form.Item
         label={t('selfRag')}
         valuePropName="checked"