address PR comment

coderd/aitasks.go

@@ -327,7 +327,7 @@ func taskFromDBTaskAndWorkspace(dbTask database.Task, ws codersdk.Workspace) cod
 		if ws.LatestBuild.Transition != codersdk.WorkspaceTransitionStart || ws.LatestAppStatus.CreatedAt.After(ws.LatestBuild.CreatedAt) {
 			currentState = &codersdk.TaskStateEntry{
 				Timestamp: ws.LatestAppStatus.CreatedAt,
-				State:     codersdk.TaskState(ws.LatestAppStatus.State),
+				State:     codersdk.TaskStateFromWorkspaceAppStatus(ws.LatestAppStatus.State),
 				Message:   ws.LatestAppStatus.Message,
 				URI:       ws.LatestAppStatus.URI,
 			}

codersdk/aitasks.go

@@ -146,8 +146,25 @@ const (
 	// TaskStateFailed indicates the AI agent reported a failure state.
 	// Reported via the workspace app status.
 	TaskStateFailed TaskState = "failed"
+	// TaskStateUnknown is the zero value for TaskState.
+	TaskStateUnknown TaskState = ""
 )
 
+func TaskStateFromWorkspaceAppStatus(was WorkspaceAppStatusState) TaskState {
+	switch was {
+	case WorkspaceAppStatusStateWorking:
+		return TaskStateWorking
+	case WorkspaceAppStatusStateIdle:
+		return TaskStateIdle
+	case WorkspaceAppStatusStateComplete:
+		return TaskStateComplete
+	case WorkspaceAppStatusStateFailure:
+		return TaskStateFailed
+	default:
+		return TaskStateUnknown
+	}
+}
+
 // Task represents a task.
 //
 // Experimental: This type is experimental and may change in the future.