diff --git a/web/src/components/llm-select/index.tsx b/web/src/components/llm-select/index.tsx index 0da49e814..1100390b8 100644 --- a/web/src/components/llm-select/index.tsx +++ b/web/src/components/llm-select/index.tsx @@ -1,5 +1,5 @@ import { LlmModelType } from '@/constants/knowledge'; -import { useSelectLlmOptionsByModelType } from '@/hooks/llm-hooks'; +import { useComposeLlmOptionsByModelTypes } from '@/hooks/llm-hooks'; import { Popover, Select } from 'antd'; import LlmSettingItems from '../llm-setting-items'; @@ -10,7 +10,10 @@ interface IProps { } const LLMSelect = ({ id, value, onChange }: IProps) => { - const modelOptions = useSelectLlmOptionsByModelType(); + const modelOptions = useComposeLlmOptionsByModelTypes([ + LlmModelType.Chat, + LlmModelType.Image2text, + ]); const content = (
@@ -29,10 +32,7 @@ const LLMSelect = ({ id, value, onChange }: IProps) => { destroyTooltipOnHide > +