forge-guardrails 0.1.2

Foundation types for an LLM-agent workflow framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

"""Unit tests for forge.core.steps.StepTracker."""

from forge.core.steps import StepTracker


class TestStepTracker:
    def test_empty_required_steps_is_satisfied(self):
        tracker = StepTracker(required_steps=[])
        assert tracker.is_satisfied() is True

    def test_partial_completion_not_satisfied(self):
        tracker = StepTracker(required_steps=["get_pricing", "get_history"])
        tracker.record("get_pricing")
        assert tracker.is_satisfied() is False

    def test_full_completion_is_satisfied(self):
        tracker = StepTracker(required_steps=["get_pricing", "get_history"])
        tracker.record("get_pricing")
        tracker.record("get_history")
        assert tracker.is_satisfied() is True

    def test_pending_returns_uncompleted_in_order(self):
        tracker = StepTracker(
            required_steps=["step_a", "step_b", "step_c"]
        )
        tracker.record("step_b")
        assert tracker.pending() == ["step_a", "step_c"]

    def test_pending_preserves_original_order(self):
        tracker = StepTracker(
            required_steps=["z_step", "a_step", "m_step"]
        )
        tracker.record("a_step")
        assert tracker.pending() == ["z_step", "m_step"]

    def test_record_is_idempotent(self):
        tracker = StepTracker(required_steps=["step_a"])
        tracker.record("step_a")
        tracker.record("step_a")
        assert tracker.is_satisfied() is True
        assert len(tracker.completed_steps) == 1

    def test_summary_hint_no_completions(self):
        tracker = StepTracker(required_steps=["step_a", "step_b"])
        assert tracker.summary_hint() == "[No steps completed yet]"

    def test_summary_hint_with_completions(self):
        tracker = StepTracker(required_steps=["get_pricing", "get_history"])
        tracker.record("get_pricing")
        tracker.record("get_history")
        hint = tracker.summary_hint()
        assert "get_history" in hint
        assert "get_pricing" in hint
        assert hint.startswith("[Steps completed:")

    def test_summary_hint_preserves_execution_order(self):
        tracker = StepTracker(required_steps=["z_tool", "a_tool"])
        tracker.record("z_tool")
        tracker.record("a_tool")
        assert tracker.summary_hint() == "[Steps completed: z_tool, a_tool]"

    def test_summary_hint_execution_order_not_alphabetical(self):
        tracker = StepTracker(required_steps=["c_step", "a_step", "b_step"])
        tracker.record("b_step")
        tracker.record("c_step")
        tracker.record("a_step")
        assert tracker.summary_hint() == "[Steps completed: b_step, c_step, a_step]"

    def test_recording_non_required_step(self):
        tracker = StepTracker(required_steps=["step_a"])
        tracker.record("unrelated_tool")
        assert tracker.is_satisfied() is False
        assert "unrelated_tool" in tracker.completed_steps