From 77d4d9195f8bbb2a19d56ce0fdad2d751d94dadb Mon Sep 17 00:00:00 2001 From: Dmitry Vyukov Date: Wed, 18 Feb 2026 14:35:35 +0100 Subject: pkg/aflow: export Context.Close This will be needed to an MCP server. --- pkg/aflow/execute.go | 4 ++-- pkg/aflow/test_action.go | 2 +- pkg/aflow/test_tool.go | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) (limited to 'pkg') diff --git a/pkg/aflow/execute.go b/pkg/aflow/execute.go index 719001933..9fcfc940b 100644 --- a/pkg/aflow/execute.go +++ b/pkg/aflow/execute.go @@ -39,7 +39,7 @@ func (flow *Flow) Execute(ctx context.Context, model, workdir string, inputs map onEvent: onEvent, } - defer c.close() + defer c.Close() if s := ctx.Value(stubContextKey); s != nil { c.stubContext = *s.(*stubContext) } @@ -292,7 +292,7 @@ func (ctx *Context) TempDir() (string, error) { return dir, nil } -func (ctx *Context) close() { +func (ctx *Context) Close() { for _, dir := range ctx.cachedDirs { ctx.cache.Release(dir) } diff --git a/pkg/aflow/test_action.go b/pkg/aflow/test_action.go index a34410cf0..bcb633789 100644 --- a/pkg/aflow/test_action.go +++ b/pkg/aflow/test_action.go @@ -28,7 +28,7 @@ func TestAction(t *testing.T, a Action, workdir string, initArgs, wantResults an timeNow: time.Now, }, } - defer ctx.close() + defer ctx.Close() err := a.execute(ctx) gotResults := map[string]any{} gotError := "" diff --git a/pkg/aflow/test_tool.go b/pkg/aflow/test_tool.go index 9167e65c5..b2f47be74 100644 --- a/pkg/aflow/test_tool.go +++ b/pkg/aflow/test_tool.go @@ -28,7 +28,7 @@ func TestTool(t *testing.T, tool Tool, initState, initArgs, wantResults any, wan ctx := &Context{ state: state, } - defer ctx.close() + defer ctx.Close() gotResults, err := tool.execute(ctx, args) gotError := "" if err != nil { @@ -46,6 +46,6 @@ func FuzzTool(t *testing.T, tool Tool, initState, initArgs any) (map[string]any, ctx := &Context{ state: state, } - defer ctx.close() + defer ctx.Close() return tool.execute(ctx, args) } -- cgit mrf-deployment