fix: Add model by ollama in model provider page, user can't choose the model in chat window. #2479 (#2529)

### What problem does this PR solve?

fix: Add model by ollama in model provider page, user can't choose the
model in chat window. #2479

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
balibabu 2024-09-23 10:53:18 +08:00 committed by GitHub
parent d8a43416f5
commit f6ceb43e36
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 1 deletions

View File

@ -1,3 +1,5 @@
import { LlmModelType } from '@/constants/knowledge';
import { useSelectLlmOptionsByModelType } from '@/hooks/llm-hooks';
import { Popover, Select } from 'antd';
import LlmSettingItems from '../llm-setting-items';
@ -8,6 +10,8 @@ interface IProps {
}
const LLMSelect = ({ id, value, onChange }: IProps) => {
const modelOptions = useSelectLlmOptionsByModelType();
const content = (
<div style={{ width: 400 }}>
<LlmSettingItems
@ -25,6 +29,10 @@ const LLMSelect = ({ id, value, onChange }: IProps) => {
destroyTooltipOnHide
>
<Select
options={[
...modelOptions[LlmModelType.Chat],
...modelOptions[LlmModelType.Image2text],
]}
style={{ width: '100%' }}
dropdownStyle={{ display: 'none' }}
id={id}

View File

@ -71,7 +71,7 @@ export const useSelectLlmOptionsByModelType = () => {
)
.map((x) => ({
label: x.llm_name,
value: x.llm_name,
value: `${x.llm_name}@${x.fid}`,
disabled: !x.available,
})),
};