follow recommendation

Author: 6ba3i

internal/agent/component/begin.go

@@ -105,19 +105,19 @@ func (b *BeginComponent) Invoke(ctx context.Context, db *gorm.DB, inputs map[str
 	for _, field := range b.inputFields {
 		if _, direct := inputs[field]; !direct {
 			if query, scalar := inputs["query"].(string); scalar {
-				if value, initialized := state.Globals["begin@"+field]; initialized {
+				if value, initialized := state.GetGlobal("begin@" + field); initialized {
 					out[field] = value
 					continue
 				}
 				out[field] = query
-				state.Globals["begin@"+field] = query
+				state.SetGlobal("begin@"+field, query)
 				continue
 			}
 		}
 		if value, ok := beginInputValue(inputs, field); ok {
 			out[field] = value
-			state.Globals["begin@"+field] = value
-		} else if value, ok := state.Globals["begin@"+field]; ok {
+			state.SetGlobal("begin@"+field, value)
+		} else if value, ok := state.GetGlobal("begin@" + field); ok {
 			out[field] = value
 		}
 	}

internal/agent/component/begin_test.go

@@ -18,6 +18,7 @@ package component
 
 import (
 	"context"
+	"encoding/json"
 	"reflect"
 	"testing"
 
@@ -140,6 +141,27 @@ func TestBegin_PreservesStarterInputAcrossQueries(t *testing.T) {
 	}
 }
 
+func TestBegin_InitializesNilGlobalsAfterRestore(t *testing.T) {
+	c, _ := NewBeginComponent(map[string]any{
+		"inputs": map[string]any{"name": map[string]any{}},
+	})
+	var state canvas.CanvasState
+	if err := json.Unmarshal([]byte(`{"sys":{}}`), &state); err != nil {
+		t.Fatalf("Unmarshal: %v", err)
+	}
+
+	out, err := c.Invoke(canvas.WithState(t.Context(), &state), nil, map[string]any{"query": "Alice"})
+	if err != nil {
+		t.Fatalf("Invoke: %v", err)
+	}
+	if out["name"] != "Alice" {
+		t.Fatalf("name = %v, want Alice", out["name"])
+	}
+	if value, ok := state.GetGlobal("begin@name"); !ok || value != "Alice" {
+		t.Fatalf("begin@name = %v, %v; want Alice, true", value, ok)
+	}
+}
+
 // TestBegin_PassesThroughInputs asserts the full inputs map — including
 // arbitrary keys beyond query / user_id — is returned unchanged as
 // outputs. This is the contract downstream components rely on to access