소스 검색

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))