Explorar el Código

fix: support AI Gateway, close #52

CaIon hace 2 años
padre
commit
0004c1022d
Se han modificado 1 ficheros con 2 adiciones y 6 borrados
  1. 2 6
      controller/channel-test.go

+ 2 - 6
controller/channel-test.go

@@ -43,14 +43,10 @@ func testChannel(channel *model.Channel, request ChatRequest) (err error, openai
 	default:
 		request.Model = "gpt-3.5-turbo"
 	}
-	requestURL := common.ChannelBaseURLs[channel.Type]
+	requestURL := getFullRequestURL(channel.GetBaseURL(), "/v1/chat/completions", channel.Type)
+
 	if channel.Type == common.ChannelTypeAzure {
 		requestURL = getFullRequestURL(channel.GetBaseURL(), fmt.Sprintf("/openai/deployments/%s/chat/completions?api-version=2023-03-15-preview", request.Model), channel.Type)
-	} else {
-		if channel.GetBaseURL() != "" {
-			requestURL = channel.GetBaseURL()
-		}
-		requestURL += "/v1/chat/completions"
 	}
 
 	jsonData, err := json.Marshal(request)