Просмотр исходного кода

feat(channel model list): modify fetching model list in add channel to fetch by type

mango 11 месяцев назад
Родитель
Сommit
f5be2868df
2 измененных файлов с 6 добавлено и 4 удалено
  1. 2 1
      controller/channel.go
  2. 4 3
      web/src/pages/Channel/EditChannel.js

+ 2 - 1
controller/channel.go

@@ -510,6 +510,7 @@ func UpdateChannel(c *gin.Context) {
 func FetchModels(c *gin.Context) {
 	var req struct {
 		BaseURL string `json:"base_url"`
+		Type    int    `json:"type"`
 		Key     string `json:"key"`
 	}
 
@@ -523,7 +524,7 @@ func FetchModels(c *gin.Context) {
 
 	baseURL := req.BaseURL
 	if baseURL == "" {
-		baseURL = "https://api.openai.com"
+		baseURL = common.ChannelBaseURLs[req.Type]
 	}
 
 	client := &http.Client{}

+ 4 - 3
web/src/pages/Channel/EditChannel.js

@@ -218,6 +218,7 @@ const EditChannel = (props) => {
         try {
           const res = await API.post('/api/channel/fetch_models', {
             base_url: inputs['base_url'],
+            type: inputs['type'],
             key: inputs['key']
           });
           
@@ -885,7 +886,7 @@ const EditChannel = (props) => {
             </Typography.Text>
           </div>
           <TextArea
-            placeholder={t('此项可选,用于复写返回的状态码,比如将claude渠道的400错误复写为500(用于重试),请勿滥用该功能,例如:') + 
+            placeholder={t('此项可选,用于复写返回的状态码,比如将claude渠道的400错误复写为500(用于重试),请勿滥用该功能,例如:') +
               '\n' + JSON.stringify(STATUS_CODE_MAPPING_EXAMPLE, null, 2)}
             name="status_code_mapping"
             onChange={(value) => {
@@ -984,7 +985,7 @@ const EditChannel = (props) => {
             />
             <Typography.Text
               style={{
-                color: 'rgba(var(--semi-blue-5), 1)', 
+                color: 'rgba(var(--semi-blue-5), 1)',
                 userSelect: 'none',
                 cursor: 'pointer'
               }}
@@ -1000,7 +1001,7 @@ const EditChannel = (props) => {
               {t('填入模板')}
               </Typography.Text>
             </>
-          )}  
+          )}
         </Spin>
       </SideSheet>
     </>