Skip to content

Commit c50bf86

Browse files
committed
backend update to not allow waveai could without telemetry
1 parent d497c93 commit c50bf86

2 files changed

Lines changed: 11 additions & 6 deletions

File tree

pkg/aiusechat/openai/openai-backend.go

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,17 +24,19 @@ import (
2424
"github.com/wavetermdev/waveterm/pkg/web/sse"
2525
)
2626

27-
// sanitizeHostnameInError removes the specific hostname from error messages
28-
func sanitizeHostnameInError(err error, baseURL string) error {
27+
// sanitizeHostnameInError removes the Wave cloud hostname from error messages
28+
func sanitizeHostnameInError(err error) error {
2929
if err == nil {
3030
return nil
3131
}
3232

3333
errStr := err.Error()
34-
parsedURL, parseErr := url.Parse(baseURL)
34+
parsedURL, parseErr := url.Parse(uctypes.DefaultAIEndpoint)
3535
if parseErr == nil && parsedURL.Host != "" {
36-
errStr = strings.ReplaceAll(errStr, baseURL, "AI service")
37-
errStr = strings.ReplaceAll(errStr, parsedURL.Host, "host")
36+
if strings.Contains(errStr, parsedURL.Host) {
37+
errStr = strings.ReplaceAll(errStr, uctypes.DefaultAIEndpoint, "AI service")
38+
errStr = strings.ReplaceAll(errStr, parsedURL.Host, "host")
39+
}
3840
}
3941

4042
return fmt.Errorf("%s", errStr)
@@ -520,7 +522,7 @@ func RunOpenAIChatStep(
520522

521523
resp, err := httpClient.Do(req)
522524
if err != nil {
523-
return nil, nil, nil, sanitizeHostnameInError(err, chatOpts.Config.Endpoint)
525+
return nil, nil, nil, sanitizeHostnameInError(err)
524526
}
525527
defer resp.Body.Close()
526528

pkg/aiusechat/usechat.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,9 @@ func getWaveAISettings(premium bool, builderMode bool, rtInfo waveobj.ObjRTInfo)
8484
if err != nil {
8585
return nil, err
8686
}
87+
if config.WaveAICloud && !telemetry.IsTelemetryEnabled() {
88+
return nil, fmt.Errorf("Wave AI cloud modes require telemetry to be enabled")
89+
}
8790
apiToken := config.APIToken
8891
if apiToken == "" && config.APITokenSecretName != "" {
8992
secret, exists, err := secretstore.GetSecret(config.APITokenSecretName)

0 commit comments

Comments
 (0)