瀏覽代碼

fix: get upstream models

CalciumIon 1 年之前
父節點
當前提交
3003d12a20
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      controller/channel.go

+ 2 - 2
controller/channel.go

@@ -115,8 +115,8 @@ func FetchUpstreamModels(c *gin.Context) {
 	//	return
 	//}
 	baseURL := common.ChannelBaseURLs[channel.Type]
-	if channel.GetBaseURL() == "" {
-		channel.BaseURL = &baseURL
+	if channel.GetBaseURL() != "" {
+		baseURL = channel.GetBaseURL()
 	}
 	url := fmt.Sprintf("%s/v1/models", baseURL)
 	body, err := GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key))