telemetry panel for waveai

sawka · sawka · commit 6d8befb1fa98 · 2025-09-29T13:09:49.000-07:00
diff --git a/cmd/testai/main-testai.go b/cmd/testai/main-testai.go
@@ -23,6 +23,11 @@ import (
 //go:embed testschema.json
 var testSchemaJSON string
 
+const (
+	DefaultAnthropicModel = "claude-sonnet-4-5"
+	DefaultOpenAIModel    = "gpt-5"
+)
+
 // TestResponseWriter implements http.ResponseWriter and additional interfaces for testing
 type TestResponseWriter struct {
 	header http.Header
@@ -203,13 +208,13 @@ func testAnthropic(ctx context.Context, model, message string, tools []uctypes.T
 func testT1(ctx context.Context) {
 	tool := aiusechat.GetAdderToolDefinition()
 	tools := []uctypes.ToolDefinition{tool}
-	testAnthropic(ctx, "claude-sonnet-4-20250514", "what is 2+2, use the provider adder tool", tools)
+	testAnthropic(ctx, DefaultAnthropicModel, "what is 2+2, use the provider adder tool", tools)
 }
 
 func testT2(ctx context.Context) {
 	tool := aiusechat.GetAdderToolDefinition()
 	tools := []uctypes.ToolDefinition{tool}
-	testOpenAI(ctx, "gpt-5", "what is 2+2+8, use the provider adder tool", tools)
+	testOpenAI(ctx, DefaultOpenAIModel, "what is 2+2+8, use the provider adder tool", tools)
 }
 
 func printUsage() {
@@ -222,8 +227,8 @@ func printUsage() {
 	fmt.Println("  go run main-testai.go --tools 'Help me configure GitHub Actions monitoring'")
 	fmt.Println("")
 	fmt.Println("Default models:")
-	fmt.Println("  OpenAI: gpt-5")
-	fmt.Println("  Anthropic: claude-sonnet-4-20250514")
+	fmt.Printf("  OpenAI: %s\n", DefaultOpenAIModel)
+	fmt.Printf("  Anthropic: %s\n", DefaultAnthropicModel)
 	fmt.Println("")
 	fmt.Println("Environment variables:")
 	fmt.Println("  OPENAI_APIKEY (for OpenAI models)")
@@ -235,10 +240,10 @@ func main() {
 	var model string
 	flag.BoolVar(&anthropic, "anthropic", false, "Use Anthropic API instead of OpenAI")
 	flag.BoolVar(&tools, "tools", false, "Enable GitHub Actions Monitor tools for testing")
-	flag.StringVar(&model, "model", "", "AI model to use (defaults: gpt-5 for OpenAI, claude-sonnet-4-20250514 for Anthropic)")
+	flag.StringVar(&model, "model", "", fmt.Sprintf("AI model to use (defaults: %s for OpenAI, %s for Anthropic)", DefaultOpenAIModel, DefaultAnthropicModel))
 	flag.BoolVar(&help, "help", false, "Show usage information")
-	flag.BoolVar(&t1, "t1", false, "Run preset T1 test (claude-sonnet-4-20250514 with 'what is 2+2')")
-	flag.BoolVar(&t2, "t2", false, "Run preset T2 test (gpt-5 with 'what is 2+2')")
+	flag.BoolVar(&t1, "t1", false, fmt.Sprintf("Run preset T1 test (%s with 'what is 2+2')", DefaultAnthropicModel))
+	flag.BoolVar(&t2, "t2", false, fmt.Sprintf("Run preset T2 test (%s with 'what is 2+2')", DefaultOpenAIModel))
 	flag.Parse()
 
 	if help {
@@ -261,9 +266,9 @@ func main() {
 	// Set default model based on API type if not provided
 	if model == "" {
 		if anthropic {
-			model = "claude-sonnet-4-20250514"
+			model = DefaultAnthropicModel
 		} else {
-			model = "gpt-5"
+			model = DefaultOpenAIModel
 		}
 	}
 
diff --git a/frontend/app/aipanel/aipanel.tsx b/frontend/app/aipanel/aipanel.tsx
@@ -16,6 +16,7 @@ import { AIDroppedFiles } from "./aidroppedfiles";
 import { AIPanelHeader } from "./aipanelheader";
 import { AIPanelInput, type AIPanelInputRef } from "./aipanelinput";
 import { AIPanelMessages } from "./aipanelmessages";
+import { TelemetryRequiredMessage } from "./telemetryrequired";
 import { WaveAIModel, type DroppedFile } from "./waveai-model";
 
 interface AIPanelProps {
@@ -33,6 +34,7 @@ const AIPanelComponent = memo(({ className, onClose }: AIPanelProps) => {
     const isLayoutMode = jotai.useAtomValue(atoms.controlShiftDelayAtom);
     const showOverlayBlockNums = jotai.useAtomValue(getSettingsKeyAtom("app:showoverlayblocknums")) ?? true;
     const isInputFocused = jotai.useAtomValue(atoms.waveAIFocusedAtom);
+    const telemetryEnabled = jotai.useAtomValue(getSettingsKeyAtom("telemetry:enabled")) ?? false;
 
     const { messages, sendMessage, status, setMessages, error } = useChat({
         transport: new DefaultChatTransport({
@@ -283,20 +285,26 @@ const AIPanelComponent = memo(({ className, onClose }: AIPanelProps) => {
             <AIPanelHeader onClose={onClose} model={model} />
 
             <div key="main-content" className="flex-1 flex flex-col min-h-0">
-                <AIPanelMessages messages={messages} status={status} />
-                {errorMessage && (
-                    <div className="px-4 py-2 text-red-400 bg-red-900/20 border-l-4 border-red-500 mx-2 mb-2">
-                        <div className="text-sm">{errorMessage}</div>
-                    </div>
+                {!telemetryEnabled ? (
+                    <TelemetryRequiredMessage />
+                ) : (
+                    <>
+                        <AIPanelMessages messages={messages} status={status} />
+                        {errorMessage && (
+                            <div className="px-4 py-2 text-red-400 bg-red-900/20 border-l-4 border-red-500 mx-2 mb-2">
+                                <div className="text-sm">{errorMessage}</div>
+                            </div>
+                        )}
+                        <AIDroppedFiles model={model} />
+                        <AIPanelInput
+                            ref={inputRef}
+                            input={input}
+                            setInput={setInput}
+                            onSubmit={handleSubmit}
+                            status={status}
+                        />
+                    </>
                 )}
-                <AIDroppedFiles model={model} />
-                <AIPanelInput
-                    ref={inputRef}
-                    input={input}
-                    setInput={setInput}
-                    onSubmit={handleSubmit}
-                    status={status}
-                />
             </div>
         </div>
     );
diff --git a/frontend/app/aipanel/telemetryrequired.tsx b/frontend/app/aipanel/telemetryrequired.tsx
@@ -0,0 +1,89 @@
+// Copyright 2025, Command Line Inc.
+// SPDX-License-Identifier: Apache-2.0
+
+import { RpcApi } from "@/app/store/wshclientapi";
+import { TabRpcClient } from "@/app/store/wshrpcutil";
+import { cn } from "@/util/util";
+import { useState } from "react";
+
+interface TelemetryRequiredMessageProps {
+    className?: string;
+}
+
+const TelemetryRequiredMessage = ({ className }: TelemetryRequiredMessageProps) => {
+    const [isEnabling, setIsEnabling] = useState(false);
+
+    const handleEnableTelemetry = async () => {
+        setIsEnabling(true);
+        try {
+            await RpcApi.WaveAIEnableTelemetryCommand(TabRpcClient);
+        } catch (error) {
+            console.error("Failed to enable telemetry:", error);
+            setIsEnabling(false);
+        }
+    };
+
+    return (
+        <div className={cn("flex flex-col h-full", className)}>
+            <div className="flex-grow"></div>
+            <div className="flex items-center justify-center p-8 text-center">
+                <div className="max-w-md space-y-6">
+                    <div className="space-y-4">
+                        <i className="fa fa-sparkles text-accent text-5xl"></i>
+                        <h2 className="text-2xl font-semibold text-foreground">Wave AI</h2>
+                        <p className="text-secondary leading-relaxed">
+                            Wave AI is free to use and provides integrated AI chat that can interact with your widgets,
+                            help you with code, analyze files, and assist with your terminal workflows.
+                        </p>
+                    </div>
+
+                    <div className="bg-blue-900/20 border border-blue-500 rounded-lg p-4">
+                        <div className="flex items-start gap-3">
+                            <i className="fa fa-info-circle text-blue-400 text-lg mt-0.5"></i>
+                            <div className="text-left">
+                                <div className="text-blue-400 font-medium mb-1">Telemetry keeps Wave AI free</div>
+                                <div className="text-secondary text-sm mb-3">
+                                    <p className="mb-2">
+                                        To keep Wave AI free for everyone, we require a small amount of <i>anonymous</i>{" "}
+                                        usage data (app version, feature usage, system info).
+                                    </p>
+                                    <p className="mb-2">
+                                        This helps us block abuse by automated systems and ensure it's used by real
+                                        people like you.
+                                    </p>
+                                    <p>
+                                        We never collect your files, prompts, keystrokes, hostnames, or personally
+                                        identifying information.
+                                    </p>
+                                </div>
+                                <button
+                                    onClick={handleEnableTelemetry}
+                                    disabled={isEnabling}
+                                    className="bg-accent/80 hover:bg-accent disabled:bg-accent/50 text-background px-4 py-2 rounded-lg font-medium cursor-pointer disabled:cursor-not-allowed"
+                                >
+                                    {isEnabling ? "Enabling..." : "Enable Telemetry and Continue"}
+                                </button>
+                            </div>
+                        </div>
+                    </div>
+
+                    <div className="text-xs text-secondary">
+                        <a
+                            href="https://waveterm.dev/privacy"
+                            target="_blank"
+                            rel="noopener noreferrer"
+                            className="!text-secondary hover:!text-accent/80 cursor-pointer"
+                        >
+                            Privacy Policy
+                        </a>
+                    </div>
+                </div>
+            </div>
+            <div className="flex-grow-[2]"></div>
+        </div>
+    );
+};
+
+TelemetryRequiredMessage.displayName = "TelemetryRequiredMessage";
+
+export { TelemetryRequiredMessage };
diff --git a/frontend/app/store/wshclientapi.ts b/frontend/app/store/wshclientapi.ts
@@ -462,6 +462,11 @@ class RpcApiType {
         return client.wshRpcCall("waitforroute", data, opts);
     }
 
+    // command "waveaienabletelemetry" [call]
+    WaveAIEnableTelemetryCommand(client: WshClient, opts?: RpcOpts): Promise<void> {
+        return client.wshRpcCall("waveaienabletelemetry", null, opts);
+    }
+
     // command "waveinfo" [call]
     WaveInfoCommand(client: WshClient, opts?: RpcOpts): Promise<WaveInfoData> {
         return client.wshRpcCall("waveinfo", null, opts);
diff --git a/pkg/aiusechat/usechat.go b/pkg/aiusechat/usechat.go
@@ -28,7 +28,7 @@ const (
 )
 
 const DefaultAPI = APIType_OpenAI
-const DefaultAnthropicModel = "claude-sonnet-4-20250514"
+const DefaultAnthropicModel = "claude-sonnet-4-5"
 const DefaultAIEndpoint = "https://cfapi.waveterm.dev/api/waveai"
 const DefaultMaxTokens = 4 * 1024
 const DefaultOpenAIModel = "gpt-5-mini"
diff --git a/pkg/telemetry/telemetrydata/telemetrydata.go b/pkg/telemetry/telemetrydata/telemetrydata.go
@@ -14,20 +14,21 @@ import (
 )
 
 var ValidEventNames = map[string]bool{
-	"app:startup":        true,
-	"app:shutdown":       true,
-	"app:activity":       true,
-	"app:display":        true,
-	"app:counts":         true,
-	"action:magnify":     true,
-	"action:settabtheme": true,
-	"action:runaicmd":    true,
-	"action:createtab":   true,
-	"action:createblock": true,
-	"wsh:run":            true,
-	"debug:panic":        true,
-	"conn:connect":       true,
-	"conn:connecterror":  true,
+	"app:startup":            true,
+	"app:shutdown":           true,
+	"app:activity":           true,
+	"app:display":            true,
+	"app:counts":             true,
+	"action:magnify":         true,
+	"action:settabtheme":     true,
+	"action:runaicmd":        true,
+	"action:createtab":       true,
+	"action:createblock":     true,
+	"wsh:run":                true,
+	"debug:panic":            true,
+	"conn:connect":           true,
+	"conn:connecterror":      true,
+	"waveai:enabletelemetry": true,
 }
 
 type TEvent struct {
diff --git a/pkg/wshrpc/wshclient/wshclient.go b/pkg/wshrpc/wshclient/wshclient.go
@@ -550,6 +550,12 @@ func WaitForRouteCommand(w *wshutil.WshRpc, data wshrpc.CommandWaitForRouteData,
 	return resp, err
 }
 
+// command "waveaienabletelemetry", wshserver.WaveAIEnableTelemetryCommand
+func WaveAIEnableTelemetryCommand(w *wshutil.WshRpc, opts *wshrpc.RpcOpts) error {
+	_, err := sendRpcRequestCallHelper[any](w, "waveaienabletelemetry", nil, opts)
+	return err
+}
+
 // command "waveinfo", wshserver.WaveInfoCommand
 func WaveInfoCommand(w *wshutil.WshRpc, opts *wshrpc.RpcOpts) (*wshrpc.WaveInfoData, error) {
 	resp, err := sendRpcRequestCallHelper[*wshrpc.WaveInfoData](w, "waveinfo", nil, opts)
diff --git a/pkg/wshrpc/wshrpctypes.go b/pkg/wshrpc/wshrpctypes.go
@@ -136,7 +136,8 @@ const (
 	Command_VDomRender          = "vdomrender"
 	Command_VDomUrlRequest      = "vdomurlrequest"
 
-	Command_AiSendMessage = "aisendmessage"
+	Command_AiSendMessage           = "aisendmessage"
+	Command_WaveAIEnableTelemetry   = "waveaienabletelemetry"
 
 	Command_GetRTInfo = "getrtinfo"
 	Command_SetRTInfo = "setrtinfo"
@@ -258,6 +259,7 @@ type WshRpcInterface interface {
 
 	// ai
 	AiSendMessageCommand(ctx context.Context, data AiMessageData) error
+	WaveAIEnableTelemetryCommand(ctx context.Context) error
 
 	// rtinfo
 	GetRTInfoCommand(ctx context.Context, data CommandGetRTInfoData) (*waveobj.ObjRTInfo, error)
diff --git a/pkg/wshrpc/wshserver/wshserver.go b/pkg/wshrpc/wshserver/wshserver.go
@@ -168,7 +168,6 @@ func (ws *WshServer) SetRTInfoCommand(ctx context.Context, data wshrpc.CommandSe
 	return nil
 }
 
-
 func (ws *WshServer) ResolveIdsCommand(ctx context.Context, data wshrpc.CommandResolveIdsData) (wshrpc.CommandResolveIdsRtnData, error) {
 	rtn := wshrpc.CommandResolveIdsRtnData{}
 	rtn.ResolvedIds = make(map[string]waveobj.ORef)
@@ -829,6 +828,48 @@ func (ws WshServer) SendTelemetryCommand(ctx context.Context) error {
 	return wcloud.SendAllTelemetry(ctx, client.OID)
 }
 
+func (ws *WshServer) WaveAIEnableTelemetryCommand(ctx context.Context) error {
+	// Enable telemetry in config
+	meta := waveobj.MetaMapType{
+		wconfig.ConfigKey_TelemetryEnabled: true,
+	}
+	err := wconfig.SetBaseConfigValue(meta)
+	if err != nil {
+		return fmt.Errorf("error setting telemetry enabled: %w", err)
+	}
+
+	// Get client for telemetry operations
+	client, err := wstore.DBGetSingleton[*waveobj.Client](ctx)
+	if err != nil {
+		return fmt.Errorf("getting client data for telemetry: %v", err)
+	}
+
+	// Send no-telemetry update to cloud (async)
+	go func() {
+		ctx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancelFn()
+		err := wcloud.SendNoTelemetryUpdate(ctx, client.OID, false) // false means telemetry is enabled
+		if err != nil {
+			log.Printf("error sending no-telemetry update: %v", err)
+		}
+	}()
+
+	// Record the telemetry event
+	event := telemetrydata.MakeTEvent("waveai:enabletelemetry", telemetrydata.TEventProps{})
+	err = telemetry.RecordTEvent(ctx, event)
+	if err != nil {
+		log.Printf("error recording waveai:enabletelemetry event: %v", err)
+	}
+
+	// Immediately send telemetry to cloud
+	err = wcloud.SendAllTelemetry(ctx, client.OID)
+	if err != nil {
+		log.Printf("error sending telemetry after enabling: %v", err)
+	}
+
+	return nil
+}
+
 var wshActivityRe = regexp.MustCompile(`^[a-z:#]+$`)
 
 func (ws *WshServer) WshActivityCommand(ctx context.Context, data map[string]int) error {

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ const (`
`28`	`28`	`)`
`29`	`29`
`30`	`30`	`const DefaultAPI = APIType_OpenAI`
`31`		`-const DefaultAnthropicModel = "claude-sonnet-4-20250514"`
	`31`	`+const DefaultAnthropicModel = "claude-sonnet-4-5"`
`32`	`32`	`const DefaultAIEndpoint = "https://cfapi.waveterm.dev/api/waveai"`
`33`	`33`	`const DefaultMaxTokens = 4 * 1024`
`34`	`34`	`const DefaultOpenAIModel = "gpt-5-mini"`