Adjust the parameters of gpt-40-mini model, custom for QA mode Add ch…

…unkSize setting input (#2139) * Adjust the parameters of gpt-40-mini model, custom for QA mode Add chunkSize setting input * Adjust the parameters of gpt-40-mini model
labring · Aug 7, 2024 · 7b388b2 · 7b388b2
1 parent e9ba00d
commit 7b388b2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/projects/app/data/config.json b/projects/app/data/config.json
@@ -14,7 +14,7 @@
       "name": "gpt-4o-mini", // 模型别名
       "avatar": "/imgs/model/openai.svg", // 模型的logo
       "maxContext": 125000, // 最大上下文
-      "maxResponse": 4000, // 最大回复
+      "maxResponse": 16000, // 最大回复
       "quoteMaxToken": 120000, // 最大引用内容
       "maxTemperature": 1.2, // 最大温度
       "charsPointsPrice": 0, // n积分/1k token（商业版）

diff --git a/projects/app/src/pages/dataset/detail/components/Import/Context.tsx b/projects/app/src/pages/dataset/detail/components/Import/Context.tsx
@@ -201,12 +201,13 @@ const DatasetImportContextProvider = ({ children }: { children: React.ReactNode
       uploadRate: 150
     },
     [TrainingModeEnum.qa]: {
+      chunkSizeField: 'embeddingChunkSize' as ChunkSizeFieldType,
       chunkOverlapRatio: 0,
-      maxChunkSize: 8000,
-      minChunkSize: 3000,
+      maxChunkSize: 4096,
+      minChunkSize: 512,
       autoChunkSize: agentModel.maxContext * 0.55 || 6000,
-      chunkSize: agentModel.maxContext * 0.55 || 6000,
-      showChunkInput: false,
+      chunkSize: embeddingChunkSize || agentModel.maxContext * 0.55 || 6000,
+      showChunkInput: true,
       showPromptInput: true,
       charsPointsPrice: agentModel.charsPointsPrice,
       priceTip: t('core.dataset.import.QA Estimated Price Tips', {