diff options
| author | Dmitry Vyukov <dvyukov@google.com> | 2026-01-14 11:35:20 +0100 |
|---|---|---|
| committer | Dmitry Vyukov <dvyukov@google.com> | 2026-01-14 11:07:16 +0000 |
| commit | a9d6a79219801d2130df3b1a792c57f0e5428e9f (patch) | |
| tree | bc900771cf25374ed86011f4c0a85e7eb4647d2e /pkg/aflow/flow | |
| parent | 1b03c2cc6e672ed19398ca4a9ce22da45299e68a (diff) | |
pkg/aflow: allow to specify model per-flow
We may want to use a weaker model for some workflows.
Allow to use different models for different workflows.
Diffstat (limited to 'pkg/aflow/flow')
| -rw-r--r-- | pkg/aflow/flow/assessment/kcsan.go | 1 | ||||
| -rw-r--r-- | pkg/aflow/flow/assessment/moderation.go | 1 | ||||
| -rw-r--r-- | pkg/aflow/flow/patching/patching.go | 1 |
3 files changed, 3 insertions, 0 deletions
diff --git a/pkg/aflow/flow/assessment/kcsan.go b/pkg/aflow/flow/assessment/kcsan.go index 92033aff7..67d695eb9 100644 --- a/pkg/aflow/flow/assessment/kcsan.go +++ b/pkg/aflow/flow/assessment/kcsan.go @@ -23,6 +23,7 @@ func init() { ai.WorkflowAssessmentKCSAN, "assess if a KCSAN report is about a benign race that only needs annotations or not", &aflow.Flow{ + Model: aflow.GoodBalancedModel, Root: &aflow.Pipeline{ Actions: []aflow.Action{ kernel.Checkout, diff --git a/pkg/aflow/flow/assessment/moderation.go b/pkg/aflow/flow/assessment/moderation.go index 4b78f901c..8d9ac4a0b 100644 --- a/pkg/aflow/flow/assessment/moderation.go +++ b/pkg/aflow/flow/assessment/moderation.go @@ -33,6 +33,7 @@ func init() { ai.WorkflowModeration, "assess if a bug report is consistent and actionable or not", &aflow.Flow{ + Model: aflow.GoodBalancedModel, Root: &aflow.Pipeline{ Actions: []aflow.Action{ aflow.NewFuncAction("extract-crash-type", extractCrashType), diff --git a/pkg/aflow/flow/patching/patching.go b/pkg/aflow/flow/patching/patching.go index ef10f1a2e..766cf089f 100644 --- a/pkg/aflow/flow/patching/patching.go +++ b/pkg/aflow/flow/patching/patching.go @@ -43,6 +43,7 @@ func init() { ai.WorkflowPatching, "generate a kernel patch fixing a provided bug reproducer", &aflow.Flow{ + Model: aflow.BestExpensiveModel, Root: &aflow.Pipeline{ Actions: []aflow.Action{ baseCommitPicker, |