feat(coderd/telemetry): track AI task usage

johnstcn · johnstcn · commit 99fcbee0fee4 · 2025-08-19T17:32:45.000+01:00
diff --git a/coderd/telemetry/telemetry.go b/coderd/telemetry/telemetry.go
@@ -776,6 +776,7 @@ func ConvertWorkspaceBuild(build database.WorkspaceBuild) WorkspaceBuild {
 		TemplateVersionID: build.TemplateVersionID,
 		// #nosec G115 - Safe conversion as build numbers are expected to be positive and within uint32 range
 		BuildNumber: uint32(build.BuildNumber),
+		HasAITask:   build.HasAITask.Valid && build.HasAITask.Bool,
 	}
 }
 
@@ -1105,6 +1106,7 @@ func ConvertTemplateVersion(version database.TemplateVersion) TemplateVersion {
 	if version.SourceExampleID.Valid {
 		snapVersion.SourceExampleID = &version.SourceExampleID.String
 	}
+	snapVersion.HasAITask = version.HasAITask.Valid && version.HasAITask.Bool
 	return snapVersion
 }
 
@@ -1357,6 +1359,7 @@ type WorkspaceBuild struct {
 	TemplateVersionID uuid.UUID `json:"template_version_id"`
 	JobID             uuid.UUID `json:"job_id"`
 	BuildNumber       uint32    `json:"build_number"`
+	HasAITask         bool      `json:"has_ai_task"`
 }
 
 type Workspace struct {
@@ -1404,6 +1407,7 @@ type TemplateVersion struct {
 	OrganizationID  uuid.UUID  `json:"organization_id"`
 	JobID           uuid.UUID  `json:"job_id"`
 	SourceExampleID *string    `json:"source_example_id,omitempty"`
+	HasAITask       bool       `json:"has_ai_task"`
 }
 
 type ProvisionerJob struct {
diff --git a/coderd/telemetry/telemetry_test.go b/coderd/telemetry/telemetry_test.go
@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"net/url"
+	"slices"
 	"sort"
 	"testing"
 	"time"
@@ -105,6 +106,52 @@ func TestTelemetry(t *testing.T) {
 			OpenIn:       database.WorkspaceAppOpenInSlimWindow,
 			AgentID:      wsagent.ID,
 		})
+
+		taskJob := dbgen.ProvisionerJob(t, db, nil, database.ProvisionerJob{
+			Provisioner:    database.ProvisionerTypeTerraform,
+			StorageMethod:  database.ProvisionerStorageMethodFile,
+			Type:           database.ProvisionerJobTypeTemplateVersionDryRun,
+			OrganizationID: org.ID,
+		})
+		taskTpl := dbgen.Template(t, db, database.Template{
+			Provisioner:    database.ProvisionerTypeTerraform,
+			OrganizationID: org.ID,
+			CreatedBy:      user.ID,
+		})
+		_ = dbgen.TemplateVersion(t, db, database.TemplateVersion{
+			OrganizationID: org.ID,
+			TemplateID:     uuid.NullUUID{UUID: taskTpl.ID, Valid: true},
+			CreatedBy:      user.ID,
+			JobID:          taskJob.ID,
+			HasAITask:      sql.NullBool{Bool: true, Valid: true},
+		})
+		taskWs := dbgen.Workspace(t, db, database.WorkspaceTable{
+			OwnerID:        user.ID,
+			OrganizationID: org.ID,
+			TemplateID:     taskTpl.ID,
+		})
+		taskWsResource := dbgen.WorkspaceResource(t, db, database.WorkspaceResource{
+			JobID: taskJob.ID,
+		})
+		taskWsAgent := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{
+			ResourceID: taskWsResource.ID,
+		})
+		taskWsApp := dbgen.WorkspaceApp(t, db, database.WorkspaceApp{
+			SharingLevel: database.AppSharingLevelOwner,
+			Health:       database.WorkspaceAppHealthDisabled,
+			OpenIn:       database.WorkspaceAppOpenInSlimWindow,
+			AgentID:      taskWsAgent.ID,
+		})
+		_ = dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{
+			Transition:         database.WorkspaceTransitionStart,
+			Reason:             database.BuildReasonAutostart,
+			WorkspaceID:        taskWs.ID,
+			TemplateVersionID:  tv.ID,
+			JobID:              taskJob.ID,
+			HasAITask:          sql.NullBool{Valid: true, Bool: true},
+			AITaskSidebarAppID: uuid.NullUUID{Valid: true, UUID: taskWsApp.ID},
+		})
+
 		group := dbgen.Group(t, db, database.Group{
 			OrganizationID: org.ID,
 		})
@@ -148,19 +195,19 @@ func TestTelemetry(t *testing.T) {
 		})
 
 		_, snapshot := collectSnapshot(ctx, t, db, nil)
-		require.Len(t, snapshot.ProvisionerJobs, 1)
+		require.Len(t, snapshot.ProvisionerJobs, 2)
 		require.Len(t, snapshot.Licenses, 1)
-		require.Len(t, snapshot.Templates, 1)
-		require.Len(t, snapshot.TemplateVersions, 2)
+		require.Len(t, snapshot.Templates, 2)
+		require.Len(t, snapshot.TemplateVersions, 3)
 		require.Len(t, snapshot.Users, 1)
 		require.Len(t, snapshot.Groups, 2)
 		// 1 member in the everyone group + 1 member in the custom group
 		require.Len(t, snapshot.GroupMembers, 2)
-		require.Len(t, snapshot.Workspaces, 1)
-		require.Len(t, snapshot.WorkspaceApps, 1)
-		require.Len(t, snapshot.WorkspaceAgents, 1)
-		require.Len(t, snapshot.WorkspaceBuilds, 1)
-		require.Len(t, snapshot.WorkspaceResources, 1)
+		require.Len(t, snapshot.Workspaces, 2)
+		require.Len(t, snapshot.WorkspaceApps, 2)
+		require.Len(t, snapshot.WorkspaceAgents, 2)
+		require.Len(t, snapshot.WorkspaceBuilds, 2)
+		require.Len(t, snapshot.WorkspaceResources, 2)
 		require.Len(t, snapshot.WorkspaceAgentStats, 1)
 		require.Len(t, snapshot.WorkspaceProxies, 1)
 		require.Len(t, snapshot.WorkspaceModules, 1)
@@ -169,11 +216,18 @@ func TestTelemetry(t *testing.T) {
 		require.Len(t, snapshot.TelemetryItems, 2)
 		require.Len(t, snapshot.WorkspaceAgentMemoryResourceMonitors, 1)
 		require.Len(t, snapshot.WorkspaceAgentVolumeResourceMonitors, 1)
-		wsa := snapshot.WorkspaceAgents[0]
+		wsa := snapshot.WorkspaceAgents[1]
 		require.Len(t, wsa.Subsystems, 2)
 		require.Equal(t, string(database.WorkspaceAgentSubsystemEnvbox), wsa.Subsystems[0])
 		require.Equal(t, string(database.WorkspaceAgentSubsystemExectrace), wsa.Subsystems[1])
 
+		require.True(t, slices.ContainsFunc(snapshot.TemplateVersions, func(ttv telemetry.TemplateVersion) bool {
+			return ttv.HasAITask
+		}))
+		require.True(t, slices.ContainsFunc(snapshot.WorkspaceBuilds, func(twb telemetry.WorkspaceBuild) bool {
+			return twb.HasAITask
+		}))
+
 		tvs := snapshot.TemplateVersions
 		sort.Slice(tvs, func(i, j int) bool {
 			// Sort by SourceExampleID presence (non-nil comes before nil)

Original file line number	Diff line number	Diff line change
`@@ -776,6 +776,7 @@ func ConvertWorkspaceBuild(build database.WorkspaceBuild) WorkspaceBuild {`
`776`	`776`	`TemplateVersionID: build.TemplateVersionID,`
`777`	`777`	`// #nosec G115 - Safe conversion as build numbers are expected to be positive and within uint32 range`
`778`	`778`	`BuildNumber: uint32(build.BuildNumber),`
	`779`	`+ HasAITask: build.HasAITask.Valid && build.HasAITask.Bool,`
`779`	`780`	`}`
`780`	`781`	`}`
`781`	`782`
`@@ -1105,6 +1106,7 @@ func ConvertTemplateVersion(version database.TemplateVersion) TemplateVersion {`
`1105`	`1106`	`if version.SourceExampleID.Valid {`
`1106`	`1107`	`snapVersion.SourceExampleID = &version.SourceExampleID.String`
`1107`	`1108`	`}`
	`1109`	`+ snapVersion.HasAITask = version.HasAITask.Valid && version.HasAITask.Bool`
`1108`	`1110`	`return snapVersion`
`1109`	`1111`	`}`
`1110`	`1112`
`@@ -1357,6 +1359,7 @@ type WorkspaceBuild struct {`
`1357`	`1359`	TemplateVersionID uuid.UUID `json:"template_version_id"`
`1358`	`1360`	JobID uuid.UUID `json:"job_id"`
`1359`	`1361`	BuildNumber uint32 `json:"build_number"`
	`1362`	+ HasAITask bool `json:"has_ai_task"`
`1360`	`1363`	`}`
`1361`	`1364`
`1362`	`1365`	`type Workspace struct {`
`@@ -1404,6 +1407,7 @@ type TemplateVersion struct {`
`1404`	`1407`	OrganizationID uuid.UUID `json:"organization_id"`
`1405`	`1408`	JobID uuid.UUID `json:"job_id"`
`1406`	`1409`	SourceExampleID *string `json:"source_example_id,omitempty"`
	`1410`	+ HasAITask bool `json:"has_ai_task"`
`1407`	`1411`	`}`
`1408`	`1412`
`1409`	`1413`	`type ProvisionerJob struct {`