diff options
| author | Dmitry Vyukov <dvyukov@google.com> | 2026-01-15 20:53:57 +0100 |
|---|---|---|
| committer | Dmitry Vyukov <dvyukov@google.com> | 2026-01-20 21:12:57 +0000 |
| commit | 7f5908e77ae0e7fef4b7901341b8c2c4bbb74b28 (patch) | |
| tree | 2ccbc85132a170d046837de6bdd8be3317f94060 /dashboard/app/ai_test.go | |
| parent | 2494e18d5ced59fc7f0522749041e499d3082a9e (diff) | |
pkg/aflow: make LLM model per-agent rather than per-flow
Having LLM model per-agent is even more flexible than per-flow.
We can have some more complex tasks during patch generation with the most elaborate model,
but also some simpler ones with less elaborate models.
Diffstat (limited to 'dashboard/app/ai_test.go')
| -rw-r--r-- | dashboard/app/ai_test.go | 30 |
1 files changed, 15 insertions, 15 deletions
diff --git a/dashboard/app/ai_test.go b/dashboard/app/ai_test.go index b775b2a89..addf71f5a 100644 --- a/dashboard/app/ai_test.go +++ b/dashboard/app/ai_test.go @@ -64,9 +64,9 @@ func TestAIBugWorkflows(t *testing.T) { _, err := c.aiClient.AIJobPoll(&dashapi.AIJobPollReq{ CodeRevision: prog.GitRevision, Workflows: []dashapi.AIWorkflow{ - {Type: "patching", Name: "patching", LLMModel: "smarty"}, - {Type: "patching", Name: "patching-foo", LLMModel: "smarty"}, - {Type: "patching", Name: "patching-bar", LLMModel: "smarty"}, + {Type: "patching", Name: "patching"}, + {Type: "patching", Name: "patching-foo"}, + {Type: "patching", Name: "patching-bar"}, }, }) require.NoError(t, err) @@ -77,10 +77,10 @@ func TestAIBugWorkflows(t *testing.T) { _, err = c.aiClient.AIJobPoll(&dashapi.AIJobPollReq{ CodeRevision: prog.GitRevision, Workflows: []dashapi.AIWorkflow{ - {Type: "patching", Name: "patching", LLMModel: "smarty"}, - {Type: "patching", Name: "patching-bar", LLMModel: "smarty"}, - {Type: "patching", Name: "patching-baz", LLMModel: "smarty"}, - {Type: "assessment-kcsan", Name: "assessment-kcsan", LLMModel: "smarty"}, + {Type: "patching", Name: "patching"}, + {Type: "patching", Name: "patching-bar"}, + {Type: "patching", Name: "patching-baz"}, + {Type: "assessment-kcsan", Name: "assessment-kcsan"}, }, }) require.NoError(t, err) @@ -88,11 +88,11 @@ func TestAIBugWorkflows(t *testing.T) { _, err = c.aiClient.AIJobPoll(&dashapi.AIJobPollReq{ CodeRevision: prog.GitRevision, Workflows: []dashapi.AIWorkflow{ - {Type: "patching", Name: "patching", LLMModel: "smarty"}, - {Type: "patching", Name: "patching-bar", LLMModel: "smarty"}, - {Type: "patching", Name: "patching-qux", LLMModel: "smarty"}, - {Type: "assessment-kcsan", Name: "assessment-kcsan", LLMModel: "smarty"}, - {Type: "assessment-kcsan", Name: "assessment-kcsan-foo", LLMModel: "smarty"}, + {Type: "patching", Name: "patching"}, + {Type: "patching", Name: "patching-bar"}, + {Type: "patching", Name: "patching-qux"}, + {Type: "assessment-kcsan", Name: "assessment-kcsan"}, + {Type: "assessment-kcsan", Name: "assessment-kcsan-foo"}, }, }) require.NoError(t, err) @@ -115,7 +115,7 @@ func TestAIJob(t *testing.T) { resp, err := c.aiClient.AIJobPoll(&dashapi.AIJobPollReq{ CodeRevision: prog.GitRevision, Workflows: []dashapi.AIWorkflow{ - {Type: "assessment-kcsan", Name: "assessment-kcsan", LLMModel: "smarty"}, + {Type: "assessment-kcsan", Name: "assessment-kcsan"}, }, }) require.NoError(t, err) @@ -134,7 +134,7 @@ func TestAIJob(t *testing.T) { resp2, err2 := c.aiClient.AIJobPoll(&dashapi.AIJobPollReq{ CodeRevision: prog.GitRevision, Workflows: []dashapi.AIWorkflow{ - {Type: "assessment-kcsan", Name: "assessment-kcsan", LLMModel: "smarty"}, + {Type: "assessment-kcsan", Name: "assessment-kcsan"}, }, }) require.NoError(t, err2) @@ -210,7 +210,7 @@ func TestAIAssessmentKCSAN(t *testing.T) { resp, err := c.aiClient.AIJobPoll(&dashapi.AIJobPollReq{ CodeRevision: prog.GitRevision, Workflows: []dashapi.AIWorkflow{ - {Type: ai.WorkflowAssessmentKCSAN, Name: string(ai.WorkflowAssessmentKCSAN), LLMModel: "smarty"}, + {Type: ai.WorkflowAssessmentKCSAN, Name: string(ai.WorkflowAssessmentKCSAN)}, }, }) require.NoError(t, err) |