feat: support Self-RAG #1069 (#1079)

### What problem does this PR solve? feat: support Self-RAG #1069 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
infiniflow · Jun 6, 2024 · 1552dca · 1552dca
1 parent db35e9d
commit 1552dca
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 0 deletions.
diff --git a/web/src/locales/en.ts b/web/src/locales/en.ts
@@ -379,6 +379,8 @@ The above is the content you need to summarize.`,
         'This sets the maximum length of the model’s output, measured in the number of tokens (words or pieces of words).',
       quote: 'Show Quote',
       quoteTip: 'Should the source of the original text be displayed?',
+      selfRag: 'Self-RAG',
+      selfRagTip: 'Please refer to: https://huggingface.co/papers/2310.11511',
       overview: 'Chat Bot API',
       pv: 'Number of messages',
       uv: 'Active user number',

diff --git a/web/src/locales/zh-traditional.ts b/web/src/locales/zh-traditional.ts
@@ -350,6 +350,8 @@ export default {
         '這設置了模型輸出的最大長度，以標記（單詞或單詞片段）的數量來衡量。',
       quote: '顯示引文',
       quoteTip: '是否應該顯示原文出處？',
+      selfRag: '自反令牌',
+      selfRagTip: '請參考: https://huggingface.co/papers/2310.11511',
       overview: '聊天 API',
       pv: '消息數',
       uv: '活躍用戶數',

diff --git a/web/src/locales/zh.ts b/web/src/locales/zh.ts
@@ -367,6 +367,8 @@ export default {
         '这设置了模型输出的最大长度，以标记（单词或单词片段）的数量来衡量。',
       quote: '显示引文',
       quoteTip: '是否应该显示原文出处？',
+      selfRag: '自反令牌',
+      selfRagTip: '请参考: https://huggingface.co/papers/2310.11511',
       overview: '聊天 API',
       pv: '消息数',
       uv: '活跃用户数',

diff --git a/web/src/pages/chat/chat-configuration-modal/assistant-setting.tsx b/web/src/pages/chat/chat-configuration-modal/assistant-setting.tsx
@@ -90,6 +90,15 @@ const AssistantSetting = ({ show }: ISegmentedContentProps) => {
       >
         <Switch />
       </Form.Item>
+      <Form.Item
+        label={t('selfRag')}
+        valuePropName="checked"
+        name={['prompt_config', 'self_rag']}
+        tooltip={t('selfRagTip')}
+        initialValue={true}
+      >
+        <Switch />
+      </Form.Item>
       <KnowledgeBaseItem></KnowledgeBaseItem>
     </section>
   );