From b23a4a8fead3877a03aa8c1d092a928f12d4e6cb Mon Sep 17 00:00:00 2001
From: balibabu <cike8899@users.noreply.github.com>
Date: Thu, 23 Jan 2025 15:58:22 +0800
Subject: [PATCH] Feat: Add keyword item to AssistantSetting #4543 (#4603)

### What problem does this PR solve?

Feat: Add keyword item to AssistantSetting #4543

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
---
 web/src/components/knowledge-base-item.tsx      |  7 ++++++-
 web/src/constants/knowledge.ts                  | 17 +++++++++++++++++
 web/src/locales/en.ts                           |  2 ++
 web/src/locales/zh-traditional.ts               |  2 ++
 web/src/locales/zh.ts                           |  2 ++
 .../assistant-setting.tsx                       |  9 +++++++++
 6 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/web/src/components/knowledge-base-item.tsx b/web/src/components/knowledge-base-item.tsx
index 57fcfef3a..4ab096ec9 100644
--- 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
index 63d403c94..7752af4ce 100644
--- 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
index 93cb2800b..cb9ad59d2 100644
--- 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
index aa8b9b725..5cd511023 100644
--- 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
index 6ed338573..ee1f9659d 100644
--- 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
index d3e302f17..081e78da2 100644
--- 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"