1- import { Select , type SelectProps , TooltipGroup } from '@lobehub/ui' ;
1+ import { LobeSelect , type LobeSelectProps , TooltipGroup } from '@lobehub/ui' ;
22import { createStaticStyles } from 'antd-style' ;
3- import { memo , useMemo } from 'react' ;
3+ import { type ReactNode , memo , useMemo } from 'react' ;
44
55import { ModelItemRender , ProviderItemRender } from '@/components/ModelSelect' ;
66import { useEnabledChatModels } from '@/hooks/useEnabledChatModels' ;
@@ -18,12 +18,12 @@ const styles = createStaticStyles(({ css }) => ({
1818} ) ) ;
1919
2020interface 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 {
3232const 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