Function TestMultiStepContextEvolution

pkg/context/engine_test.go:355–413

(t *testing.T)

}

func TestMultiStepContextEvolution(t *testing.T) {

	mgr := NewManager(DefaultBudget(200000))
	registerTestTools(mgr)

	engine := NewEngine(mgr, DefaultEngineConfig())

	plan1 := engine.PrepareContext("multi-step", "get latest commits from github")
	t.Logf("Step 1 (GitHub): %d tools loaded", len(plan1.ToolsLoaded))

	engine.AdaptContext(plan1, ExecutionResult{
		Success:  true,
		ToolUsed: plan1.ToolsLoaded[0],
		Latency:  30 * time.Millisecond,
	})

	for _, id := range plan1.ToolsLoaded {
		mgr.Evict(id)
	}
	plan2 := engine.PrepareContext("multi-step-2", "search jira issues assigned to me")
	t.Logf("Step 2 (Jira): %d tools loaded", len(plan2.ToolsLoaded))

	jiraCount := 0
	githubCount := 0
	for _, id := range plan2.ToolsLoaded {
		if strings.HasPrefix(id, "jira") {
			jiraCount++
		}
		if strings.HasPrefix(id, "github") {
			githubCount++
		}
	}

	if jiraCount == 0 {
		t.Error("step 2 should have loaded Jira tools")
	}
	if githubCount > jiraCount {
		t.Errorf("step 2: Jira tools should dominate (jira=%d, github=%d)", jiraCount, githubCount)
	}

	for _, id := range plan2.ToolsLoaded {
		mgr.Evict(id)
	}
	plan3 := engine.PrepareContext("multi-step-3", "send slack message about the new jira issue")
	t.Logf("Step 3 (Slack): %d tools loaded", len(plan3.ToolsLoaded))

	hasSlack := false
	for _, id := range plan3.ToolsLoaded {
		if strings.HasPrefix(id, "slack") {
			hasSlack = true
		}
	}
	if !hasSlack {
		t.Error("step 3 should have loaded Slack tools")
	}

	t.Logf("Multi-step context evolution: GitHub(%d) → Jira(%d) → Slack(%d)",
		len(plan1.ToolsLoaded), len(plan2.ToolsLoaded), len(plan3.ToolsLoaded))