From 7f5908e77ae0e7fef4b7901341b8c2c4bbb74b28 Mon Sep 17 00:00:00 2001 From: Dmitry Vyukov Date: Thu, 15 Jan 2026 20:53:57 +0100 Subject: 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. --- dashboard/app/ai_test.go | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-) (limited to 'dashboard/app/ai_test.go') 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) -- cgit mrf-deployment