Skip to content

Commit ad51305

Browse files
authored
♻️ refactor(model-select): migrate FunctionCallingModelSelect to LobeSelect (lobehub#11664)
- Replace Select with LobeSelect component - Update types from SelectProps to LobeSelectProps - Fix ModelOption label type from any to ReactNode
1 parent 70e1d99 commit ad51305

File tree

1 file changed

+6
-6
lines changed
  • src/features/ChatInput/ActionBar/Search/FunctionCallingModelSelect

1 file changed

+6
-6
lines changed

src/features/ChatInput/ActionBar/Search/FunctionCallingModelSelect/index.tsx

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
import { Select, type SelectProps, TooltipGroup } from '@lobehub/ui';
1+
import { LobeSelect, type LobeSelectProps, TooltipGroup } from '@lobehub/ui';
22
import { createStaticStyles } from 'antd-style';
3-
import { memo, useMemo } from 'react';
3+
import { type ReactNode, memo, useMemo } from 'react';
44

55
import { ModelItemRender, ProviderItemRender } from '@/components/ModelSelect';
66
import { useEnabledChatModels } from '@/hooks/useEnabledChatModels';
@@ -18,12 +18,12 @@ const styles = createStaticStyles(({ css }) => ({
1818
}));
1919

2020
interface ModelOption {
21-
label: any;
21+
label: ReactNode;
2222
provider: string;
2323
value: string;
2424
}
2525

26-
interface ModelSelectProps extends SelectProps {
26+
interface ModelSelectProps extends Omit<LobeSelectProps, 'onChange' | 'value'> {
2727
onChange?: (props: WorkingModel) => void;
2828
showAbility?: boolean;
2929
value?: WorkingModel;
@@ -32,7 +32,7 @@ interface ModelSelectProps extends SelectProps {
3232
const ModelSelect = memo<ModelSelectProps>(({ value, onChange, ...rest }) => {
3333
const enabledList = useEnabledChatModels();
3434

35-
const options = useMemo<SelectProps['options']>(() => {
35+
const options = useMemo<LobeSelectProps['options']>(() => {
3636
const getChatModels = (provider: EnabledProviderWithModels) =>
3737
provider.children
3838
.filter((model) => !!model.abilities.functionCall)
@@ -69,7 +69,7 @@ const ModelSelect = memo<ModelSelectProps>(({ value, onChange, ...rest }) => {
6969

7070
return (
7171
<TooltipGroup>
72-
<Select
72+
<LobeSelect
7373
onChange={(value, option) => {
7474
const model = value.split('/').slice(1).join('/');
7575
onChange?.({ model, provider: (option as unknown as ModelOption).provider });

0 commit comments

Comments
 (0)