@@ -145,16 +145,6 @@ func TestAITasksPrompts(t *testing.T) {
145145func TestAITasksCreate (t * testing.T ) {
146146t .Parallel ()
147147
148- makeEchoResponses := func (parameters []* proto.RichParameter )* echo.Responses {
149- return & echo.Responses {
150- Parse :echo .ParseComplete ,
151- ProvisionApply :echo .ApplyComplete ,
152- ProvisionPlan : []* proto.Response {
153- {Type :& proto.Response_Plan {Plan :& proto.PlanComplete {Parameters :parameters }}},
154- },
155- }
156- }
157-
158148t .Run ("OK" ,func (t * testing.T ) {
159149t .Parallel ()
160150
@@ -169,9 +159,16 @@ func TestAITasksCreate(t *testing.T) {
169159user := coderdtest .CreateFirstUser (t ,client )
170160
171161// Given: A template with an "AI Prompt" parameter
172- version := coderdtest .CreateTemplateVersion (t ,client ,user .OrganizationID ,makeEchoResponses ([]* proto.RichParameter {
173- {Name :"AI Prompt" ,Type :"string" },
174- }))
162+ version := coderdtest .CreateTemplateVersion (t ,client ,user .OrganizationID ,& echo.Responses {
163+ Parse :echo .ParseComplete ,
164+ ProvisionApply :echo .ApplyComplete ,
165+ ProvisionPlan : []* proto.Response {
166+ {Type :& proto.Response_Plan {Plan :& proto.PlanComplete {
167+ Parameters : []* proto.RichParameter {{Name :"AI Prompt" ,Type :"string" }},
168+ HasAiTasks :true ,
169+ }}},
170+ },
171+ })
175172coderdtest .AwaitTemplateVersionJobCompleted (t ,client ,version .ID )
176173template := coderdtest .CreateTemplate (t ,client ,user .OrganizationID ,version .ID )
177174