add provider and local to wave ai telemetry

Author: sawka

pkg/aiusechat/uctypes/uctypes.go

@@ -332,6 +332,8 @@ type AIMetrics struct {
 	WidgetAccess      bool           `json:"widgetaccess"`
 	ThinkingLevel     string         `json:"thinkinglevel,omitempty"`
 	AIMode            string         `json:"aimode,omitempty"`
+	AIProvider        string         `json:"aiprovider,omitempty"`
+	IsLocal           bool           `json:"islocal,omitempty"`
 }
 
 type AIFunctionCallInput struct {

pkg/aiusechat/usechat.go

@@ -60,6 +60,14 @@ func getSystemPrompt(apiType string, model string, isBuilder bool) []string {
 	return []string{basePrompt}
 }
 
+func isLocalEndpoint(endpoint string) bool {
+	if endpoint == "" {
+		return false
+	}
+	endpointLower := strings.ToLower(endpoint)
+	return strings.Contains(endpointLower, "localhost") || strings.Contains(endpointLower, "127.0.0.1")
+}
+
 func getWaveAISettings(premium bool, builderMode bool, rtInfo waveobj.ObjRTInfo) (*uctypes.AIOptsType, error) {
 	maxTokens := DefaultMaxTokens
 	if builderMode {
@@ -353,6 +361,11 @@ func RunAIChat(ctx context.Context, sseHandler *sse.SSEHandlerCh, backend UseCha
 	defer activeChats.Delete(chatOpts.ChatId)
 
 	stepNum := chatstore.DefaultChatStore.CountUserMessages(chatOpts.ChatId)
+	aiProvider := chatOpts.Config.Provider
+	if aiProvider == "" {
+		aiProvider = uctypes.AIProvider_Custom
+	}
+	isLocal := isLocalEndpoint(chatOpts.Config.Endpoint)
 	metrics := &uctypes.AIMetrics{
 		ChatId:  chatOpts.ChatId,
 		StepNum: stepNum,
@@ -364,6 +377,8 @@ func RunAIChat(ctx context.Context, sseHandler *sse.SSEHandlerCh, backend UseCha
 		ToolDetail:    make(map[string]int),
 		ThinkingLevel: chatOpts.Config.ThinkingLevel,
 		AIMode:        chatOpts.Config.AIMode,
+		AIProvider:    aiProvider,
+		IsLocal:       isLocal,
 	}
 	firstStep := true
 	var cont *uctypes.WaveContinueResponse
@@ -563,6 +578,8 @@ func sendAIMetricsTelemetry(ctx context.Context, metrics *uctypes.AIMetrics) {
 		WaveAIWidgetAccess:         metrics.WidgetAccess,
 		WaveAIThinkingLevel:        metrics.ThinkingLevel,
 		WaveAIMode:                 metrics.AIMode,
+		WaveAIProvider:             metrics.AIProvider,
+		WaveAIIsLocal:              metrics.IsLocal,
 	})
 	_ = telemetry.RecordTEvent(ctx, event)
 }

pkg/telemetry/telemetrydata/telemetrydata.go

@@ -148,6 +148,8 @@ type TEventProps struct {
 	WaveAIWidgetAccess         bool           `json:"waveai:widgetaccess,omitempty"`
 	WaveAIThinkingLevel        string         `json:"waveai:thinkinglevel,omitempty"`
 	WaveAIMode                 string         `json:"waveai:mode,omitempty"`
+	WaveAIProvider             string         `json:"waveai:provider,omitempty"`
+	WaveAIIsLocal              bool           `json:"waveai:islocal,omitempty"`
 	WaveAIFeedback             string         `json:"waveai:feedback,omitempty" tstype:"\"good\" | \"bad\""`
 	WaveAIAction               string         `json:"waveai:action,omitempty"`