Feat: Add the MessageHistoryWindowSizeItem to RewriteQuestionForm #1739…

… (#4502) ### What problem does this PR solve? Feat: Add the MessageHistoryWindowSizeItem to RewriteQuestionForm #1739 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
infiniflow · Jan 16, 2025 · 39be08c · 39be08c
1 parent 3805621
commit 39be08c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/web/src/pages/flow/constant.tsx b/web/src/pages/flow/constant.tsx
@@ -426,6 +426,7 @@ export const initialGenerateValues = {
 
 export const initialRewriteQuestionValues = {
   ...initialLlmBaseValues,
+  message_history_window_size: 6,
 };
 
 export const initialRelevantValues = {
@@ -445,7 +446,7 @@ export const initialMessageValues = {
 
 export const initialKeywordExtractValues = {
   ...initialLlmBaseValues,
-  top_n: 1,
+  top_n: 3,
   ...initialQueryBaseValues,
 };
 export const initialDuckValues = {

diff --git a/web/src/pages/flow/form/keyword-extract-form/index.tsx b/web/src/pages/flow/form/keyword-extract-form/index.tsx
@@ -24,7 +24,7 @@ const KeywordExtractForm = ({ onValuesChange, form, node }: IOperatorForm) => {
       >
         <LLMSelect></LLMSelect>
       </Form.Item>
-      <TopNItem initialValue={1}></TopNItem>
+      <TopNItem initialValue={3}></TopNItem>
     </Form>
   );
 };

diff --git a/web/src/pages/flow/form/rewrite-question-form/index.tsx b/web/src/pages/flow/form/rewrite-question-form/index.tsx
@@ -1,4 +1,5 @@
 import LLMSelect from '@/components/llm-select';
+import MessageHistoryWindowSizeItem from '@/components/message-history-window-size-item';
 import { useTranslate } from '@/hooks/common-hooks';
 import { Form } from 'antd';
 import { IOperatorForm } from '../../interface';
@@ -9,8 +10,8 @@ const RewriteQuestionForm = ({ onValuesChange, form }: IOperatorForm) => {
   return (
     <Form
       name="basic"
-      labelCol={{ span: 4 }}
-      wrapperCol={{ span: 20 }}
+      labelCol={{ span: 8 }}
+      wrapperCol={{ span: 16 }}
       onValuesChange={onValuesChange}
       autoComplete="off"
       form={form}
@@ -22,6 +23,9 @@ const RewriteQuestionForm = ({ onValuesChange, form }: IOperatorForm) => {
       >
         <LLMSelect></LLMSelect>
       </Form.Item>
+      <MessageHistoryWindowSizeItem
+        initialValue={6}
+      ></MessageHistoryWindowSizeItem>
     </Form>
   );
 };