AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-05-06 07:15:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Add rerank option to huggingface's model type drop-down box. #5658 (#5689)

Browse Source 
### What problem does this PR solve?

Feat: Add rerank option to huggingface's model type drop-down box. #5658

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu 2025-03-06 11:03:08 +08:00 committed by GitHub
parent b8da2eeb69
commit 1b2fc3cc9a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
web/src/pages/user-setting/setting-model/ollama-modal/index.tsx
View File

@ -74,6 +74,7 @@ const OllamaModal = ({
[LLMFactory.HuggingFace]: [ [LLMFactory.HuggingFace]: [
{ value: 'embedding', label: 'embedding' }, { value: 'embedding', label: 'embedding' },
{ value: 'chat', label: 'chat' }, { value: 'chat', label: 'chat' },
{ value: 'rerank', label: 'rerank' },
], ],
[LLMFactory.Xinference]: [ [LLMFactory.Xinference]: [
{ value: 'chat', label: 'chat' }, { value: 'chat', label: 'chat' },
@ -170,7 +171,7 @@ const OllamaModal = ({
type: 'number', type: 'number',
message: t('maxTokensInvalidMessage'), message: t('maxTokensInvalidMessage'),
}, },
({ getFieldValue }) => ({ ({}) => ({
validator(_, value) { validator(_, value) {
if (value < 0) { if (value < 0) {
return Promise.reject(new Error(t('maxTokensMinMessage'))); return Promise.reject(new Error(t('maxTokensMinMessage')));