@@ -83,6 +83,7 @@ const EditChannel = () => {
83
83
data . model_mapping = JSON . stringify ( JSON . parse ( data . model_mapping ) , null , 2 ) ;
84
84
}
85
85
setInputs ( data ) ;
86
+ setBasicModels ( getChannelModels ( data . type ) ) ;
86
87
} else {
87
88
showError ( message ) ;
88
89
}
@@ -99,9 +100,6 @@ const EditChannel = () => {
99
100
} ) ) ;
100
101
setOriginModelOptions ( localModelOptions ) ;
101
102
setFullModels ( res . data . data . map ( ( model ) => model . id ) ) ;
102
- setBasicModels ( res . data . data . filter ( ( model ) => {
103
- return model . id . startsWith ( 'gpt-3' ) || model . id . startsWith ( 'text-' ) ;
104
- } ) . map ( ( model ) => model . id ) ) ;
105
103
} catch ( error ) {
106
104
showError ( error . message ) ;
107
105
}
@@ -137,6 +135,9 @@ const EditChannel = () => {
137
135
useEffect ( ( ) => {
138
136
if ( isEdit ) {
139
137
loadChannel ( ) . then ( ) ;
138
+ } else {
139
+ let localModels = getChannelModels ( inputs . type ) ;
140
+ setBasicModels ( localModels ) ;
140
141
}
141
142
fetchModels ( ) . then ( ) ;
142
143
fetchGroups ( ) . then ( ) ;
@@ -355,7 +356,7 @@ const EditChannel = () => {
355
356
< div style = { { lineHeight : '40px' , marginBottom : '12px' } } >
356
357
< Button type = { 'button' } onClick = { ( ) => {
357
358
handleInputChange ( null , { name : 'models' , value : basicModels } ) ;
358
- } } > 填入基础模型 </ Button >
359
+ } } > 填入相关模型 </ Button >
359
360
< Button type = { 'button' } onClick = { ( ) => {
360
361
handleInputChange ( null , { name : 'models' , value : fullModels } ) ;
361
362
} } > 填入所有模型</ Button >
0 commit comments