balibabu commited on
Commit
eba9f1d
·
1 Parent(s): 59ca130

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

Browse files

### 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)

web/src/components/llm-select/index.tsx CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import { Popover, Select } from 'antd';
2
  import LlmSettingItems from '../llm-setting-items';
3
 
@@ -8,6 +10,8 @@ interface IProps {
8
  }
9
 
10
  const LLMSelect = ({ id, value, onChange }: IProps) => {
 
 
11
  const content = (
12
  <div style={{ width: 400 }}>
13
  <LlmSettingItems
@@ -25,6 +29,10 @@ const LLMSelect = ({ id, value, onChange }: IProps) => {
25
  destroyTooltipOnHide
26
  >
27
  <Select
 
 
 
 
28
  style={{ width: '100%' }}
29
  dropdownStyle={{ display: 'none' }}
30
  id={id}
 
1
+ import { LlmModelType } from '@/constants/knowledge';
2
+ import { useSelectLlmOptionsByModelType } from '@/hooks/llm-hooks';
3
  import { Popover, Select } from 'antd';
4
  import LlmSettingItems from '../llm-setting-items';
5
 
 
10
  }
11
 
12
  const LLMSelect = ({ id, value, onChange }: IProps) => {
13
+ const modelOptions = useSelectLlmOptionsByModelType();
14
+
15
  const content = (
16
  <div style={{ width: 400 }}>
17
  <LlmSettingItems
 
29
  destroyTooltipOnHide
30
  >
31
  <Select
32
+ options={[
33
+ ...modelOptions[LlmModelType.Chat],
34
+ ...modelOptions[LlmModelType.Image2text],
35
+ ]}
36
  style={{ width: '100%' }}
37
  dropdownStyle={{ display: 'none' }}
38
  id={id}
web/src/hooks/llm-hooks.ts CHANGED
@@ -71,7 +71,7 @@ export const useSelectLlmOptionsByModelType = () => {
71
  )
72
  .map((x) => ({
73
  label: x.llm_name,
74
- value: x.llm_name,
75
  disabled: !x.available,
76
  })),
77
  };
 
71
  )
72
  .map((x) => ({
73
  label: x.llm_name,
74
+ value: `${x.llm_name}@${x.fid}`,
75
  disabled: !x.available,
76
  })),
77
  };