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_test.go | |
| 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_test.go')
| -rw-r--r-- | pkg/aflow/flow_test.go | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/pkg/aflow/flow_test.go b/pkg/aflow/flow_test.go index 8ab8016f3..19abce6b1 100644 --- a/pkg/aflow/flow_test.go +++ b/pkg/aflow/flow_test.go @@ -76,7 +76,8 @@ func TestWorkflow(t *testing.T) { flows := make(map[string]*Flow) err := register[flowInputs, flowOutputs]("test", "description", flows, []*Flow{ { - Name: "flow", + Name: "flow", + Model: "model", Root: NewPipeline( NewFuncAction("func-action", func(ctx *Context, args firstFuncInputs) (firstFuncOutputs, error) { @@ -530,6 +531,7 @@ func TestNoInputs(t *testing.T) { flows := make(map[string]*Flow) err := register[flowInputs, flowOutputs]("test", "description", flows, []*Flow{ { + Model: "model", Root: NewFuncAction("func-action", func(ctx *Context, args flowInputs) (flowOutputs, error) { return flowOutputs{}, nil |