Fix openai regression (#1758)

this impacts custom openai providers, not wave proxy
wavetermdev · Jan 17, 2025 · f35375e · f35375e
1 parent 1cd1744
commit f35375e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/pkg/waveai/openaibackend.go b/pkg/waveai/openaibackend.go
@@ -29,7 +29,7 @@ func defaultAzureMapperFn(model string) string {
 
 func setApiType(opts *wshrpc.WaveAIOptsType, clientConfig *openaiapi.ClientConfig) error {
 	ourApiType := strings.ToLower(opts.APIType)
-	if ourApiType == "" || ourApiType == strings.ToLower(string(openaiapi.APITypeOpenAI)) {
+	if ourApiType == "" || ourApiType == APIType_OpenAI || ourApiType == strings.ToLower(string(openaiapi.APITypeOpenAI)) {
 		clientConfig.APIType = openaiapi.APITypeOpenAI
 		return nil
 	} else if ourApiType == strings.ToLower(string(openaiapi.APITypeAzure)) {

diff --git a/pkg/waveai/waveai.go b/pkg/waveai/waveai.go
@@ -75,7 +75,6 @@ func RunAICommand(ctx context.Context, request wshrpc.WaveAIStreamRequest) chan
 		request.Opts.Model = "default"
 		backend = WaveAICloudBackend{}
 	} else {
-		request.Opts.APIType = APIType_OpenAI
 		backend = OpenAIBackend{}
 	}
 	if backend == nil {