diff --git a/wavesrv/pkg/cmdrunner/cmdrunner.go b/wavesrv/pkg/cmdrunner/cmdrunner.go index 09b2ac2b..9f5fbede 100644 --- a/wavesrv/pkg/cmdrunner/cmdrunner.go +++ b/wavesrv/pkg/cmdrunner/cmdrunner.go @@ -2683,7 +2683,7 @@ func doOpenAICmdInfoCompletion(cmd *sstore.CmdType, clientId string, opts *sstor }() var ch chan *packet.OpenAIPacketType var err error - if opts.APIToken == "" { + if opts.BaseURL == "" && opts.APIToken == "" { var conn *websocket.Conn ch, conn, err = openai.RunCloudCompletionStream(ctx, clientId, opts, prompt) if conn != nil { diff --git a/wavesrv/pkg/remote/openai/openai.go b/wavesrv/pkg/remote/openai/openai.go index 038dd579..4b060d2d 100644 --- a/wavesrv/pkg/remote/openai/openai.go +++ b/wavesrv/pkg/remote/openai/openai.go @@ -144,7 +144,7 @@ func RunCompletionStream(ctx context.Context, opts *sstore.OpenAIOptsType, promp if opts.Model == "" { return nil, fmt.Errorf("no openai model specified") } - if opts.APIToken == "" { + if opts.BaseURL == "" && opts.APIToken == "" { return nil, fmt.Errorf("no api token") } clientConfig := openaiapi.DefaultConfig(opts.APIToken)