Sep 2, 2025 · Sep 2, 2025 · Sep 2, 2025 · Sep 2, 2025
diff --git a/coderd/aitasks.go b/coderd/aitasks.go
 "github.com/coder/coder/v2/coderd/rbac/policy"
 "github.com/coder/coder/v2/coderd/searchquery"
 "github.com/coder/coder/v2/coderd/taskname"
 "github.com/coder/coder/v2/coderd/util/slice"
 "github.com/coder/coder/v2/codersdk"
 )

 return
 }
 if data.builds[0].HasAITask == nil || !*data.builds[0].HasAITask {
 httpapi.ResourceNotFound(rw)
 return
 // TODO(DanielleMaywood):
 // This is a temporary workaround. When a task has just been created, but
 // not yet provisioned, the workspace build will not have `HasAITask` set.
 //
 // When we reach this code flow, it is _either_ because the workspace is
 // not a task, or it is a task that has not yet been provisioned. This
 // endpoint should rarely be called with a non-task workspace so we
 // should be fine with this extra database call to check if it has the
 // special "AI Task" parameter.
 parameters, err := api.Database.GetWorkspaceBuildParameters(ctx, data.builds[0].ID)
 if err != nil {
 httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
 Message: "Internal error fetching workspace build parameters.",
 Detail:  err.Error(),
 })
 return
 }

 _, hasAITask := slice.Find(parameters, func(t database.WorkspaceBuildParameter) bool {
 return t.Name == codersdk.AITaskPromptParameterName
 })

 if !hasAITask {
 httpapi.ResourceNotFound(rw)
 return
 }
 }

 appStatus := codersdk.WorkspaceAppStatus{}
diff --git a/coderd/aitasks_test.go b/coderd/aitasks_test.go
 {Name: codersdk.AITaskPromptParameterName, Value: wantPrompt},
 }
 })
 coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID)

 // Fetch the task by ID via experimental API and verify fields.
 exp := codersdk.NewExperimentalClient(client)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -23,6 +23,7 @@ import (
		"github.com/coder/coder/v2/coderd/rbac/policy"
		"github.com/coder/coder/v2/coderd/searchquery"
		"github.com/coder/coder/v2/coderd/taskname"
		"github.com/coder/coder/v2/coderd/util/slice"
		"github.com/coder/coder/v2/codersdk"
		)

Expand DownExpand Up		@@ -440,8 +441,32 @@ func (api API) taskGet(rw http.ResponseWriter, r http.Request) {
		return
		}
		if data.builds[0].HasAITask == nil \|\| !*data.builds[0].HasAITask {
		httpapi.ResourceNotFound(rw)
		return
		// TODO(DanielleMaywood):
		// This is a temporary workaround. When a task has just been created, but
		// not yet provisioned, the workspace build will not have `HasAITask` set.
		//
		// When we reach this code flow, it is _either_ because the workspace is
		// not a task, or it is a task that has not yet been provisioned. This
		// endpoint should rarely be called with a non-task workspace so we
		// should be fine with this extra database call to check if it has the
		// special "AI Task" parameter.
		parameters, err := api.Database.GetWorkspaceBuildParameters(ctx, data.builds[0].ID)
		if err != nil {
		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
		Message: "Internal error fetching workspace build parameters.",
		Detail: err.Error(),
		})
		return
		}

		_, hasAITask := slice.Find(parameters, func(t database.WorkspaceBuildParameter) bool {
		return t.Name == codersdk.AITaskPromptParameterName
		})

		if !hasAITask {
		httpapi.ResourceNotFound(rw)
		return
		}
		}

		appStatus := codersdk.WorkspaceAppStatus{}
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -253,7 +253,6 @@ func TestTasks(t *testing.T) {
		{Name: codersdk.AITaskPromptParameterName, Value: wantPrompt},
		}
		})
		coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID)
Copy link ContributorAuthor DanielleMaywoodSep 2, 2025 Choose a reason for hiding this comment The reason will be displayed to describe this comment to others.Learn more. Removing this call effectively tests that we can fetch a taskprior to the build job being finished mafredri reacted with thumbs up emoji

		// Fetch the task by ID via experimental API and verify fields.
		exp := codersdk.NewExperimentalClient(client)
Expand Down