Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-07 05:39:05 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

feat: add document count in create popup

Browse Source
This commit is contained in:
eeee0717 2025-04-25 18:21:29 +08:00 committed by 亢奋猫
parent ec0836c471
commit 0fa8ae9560
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/renderer/src/pages/knowledge/components/AddKnowledgePopup.tsx
View File

@ -1,4 +1,5 @@
import { TopView } from '@renderer/components/TopView' import { TopView } from '@renderer/components/TopView'
import { DEFAULT_KNOWLEDGE_DOCUMENT_COUNT } from '@renderer/config/constant'
import { isEmbeddingModel, isRerankModel } from '@renderer/config/models' import { isEmbeddingModel, isRerankModel } from '@renderer/config/models'
import { SUPPORTED_REANK_PROVIDERS } from '@renderer/config/providers' import { SUPPORTED_REANK_PROVIDERS } from '@renderer/config/providers'
import { useKnowledgeBases } from '@renderer/hooks/useKnowledge' import { useKnowledgeBases } from '@renderer/hooks/useKnowledge'
@ -9,7 +10,7 @@ import { getKnowledgeBaseParams } from '@renderer/services/KnowledgeService'
import { getModelUniqId } from '@renderer/services/ModelService' import { getModelUniqId } from '@renderer/services/ModelService'
import { Model } from '@renderer/types' import { Model } from '@renderer/types'
import { getErrorMessage } from '@renderer/utils/error' import { getErrorMessage } from '@renderer/utils/error'
import { Form, Input, Modal, Select } from 'antd' import { Form, Input, Modal, Select, Slider } from 'antd'
import { find, sortBy } from 'lodash' import { find, sortBy } from 'lodash'
import { nanoid } from 'nanoid' import { nanoid } from 'nanoid'
import { useRef, useState } from 'react' import { useRef, useState } from 'react'
@ -23,6 +24,7 @@ interface FormData {
name: string name: string
model: string model: string
rerankModel: string | undefined rerankModel: string | undefined
documentCount: number | undefined
} }
interface Props extends ShowParams { interface Props extends ShowParams {
@ -113,6 +115,7 @@ const PopupContainer: React.FC<Props> = ({ title, resolve }) => {
model: selectedModel, model: selectedModel,
rerankModel: selectedRerankModel, rerankModel: selectedRerankModel,
dimensions, dimensions,
documentCount: values.documentCount || DEFAULT_KNOWLEDGE_DOCUMENT_COUNT,
items: [], items: [],
created_at: Date.now(), created_at: Date.now(),
updated_at: Date.now(), updated_at: Date.now(),
@ -177,6 +180,19 @@ const PopupContainer: React.FC<Props> = ({ title, resolve }) => {
provider: SUPPORTED_REANK_PROVIDERS.map((id) => t(`provider.${id}`)) provider: SUPPORTED_REANK_PROVIDERS.map((id) => t(`provider.${id}`))
})} })}
</SettingHelpText> </SettingHelpText>
<Form.Item
name="documentCount"
label={t('knowledge.document_count')}
initialValue={DEFAULT_KNOWLEDGE_DOCUMENT_COUNT} // 设置初始值
tooltip={{ title: t('knowledge.document_count_help') }}>
<Slider
style={{ width: '100%' }}
min={1}
max={30}
step={1}
marks={{ 1: '1', 6: t('knowledge.document_count_default'), 30: '30' }}
/>
</Form.Item>
</Form> </Form>
</Modal> </Modal>
) )