New features: Route Stickness, Handoff, Clarification, Read-Only/Transactional, Long Term Memory

tests/unit/test_tool_policies.py

@@ -82,3 +82,17 @@ def test_requires_confirmation_alias_is_supported(tmp_path):
 
     assert policy.operation_type == "transactional"
     assert policy.require_confirmation is True
+
+
+def test_workflow_execution_policy_is_exposed(tmp_path):
+    path = tmp_path / "tool_policies.yaml"
+    path.write_text(
+        """defaults:\n  operation_type: read_only\ntool_policies:\n  refund:\n    operation_type: transactional\n    require_confirmation: true\n    execution:\n      mode: workflow\n      workflow: refund_order\n      version: 2\n""",
+        encoding="utf-8",
+    )
+    registry = ToolPolicyRegistry(str(path))
+    policy = registry.get("refund")
+    assert policy is not None
+    assert policy.execution.mode == "workflow"
+    assert policy.execution.workflow == "refund_order"
+    assert policy.execution.version == 2

tests/unit/test_transactional_workflows.py

@@ -0,0 +1,76 @@
+from pathlib import Path
+
+import pytest
+
+from agent_framework.workflows import (
+    FileWorkflowRepository,
+    WorkflowActionRegistry,
+    WorkflowRuntime,
+    WorkflowToolExecutor,
+)
+
+
+@pytest.mark.asyncio
+async def test_deterministic_workflow_routes_and_caches(tmp_path: Path):
+    (tmp_path / "refund.active.yaml").write_text("version: 1\n", encoding="utf-8")
+    (tmp_path / "refund.v1.yaml").write_text(
+        """name: refund
+version: 1
+start: validate
+nodes:
+  - id: validate
+    action: validate
+    input: {order_id: $.input.order_id}
+  - id: execute
+    action: execute
+    input: {order_id: $.input.order_id}
+edges:
+  - from: validate
+    to: execute
+    when: {path: $.nodes.validate.valid, equals: true}
+  - from: validate
+    to: END
+    when: {path: $.nodes.validate.valid, equals: false}
+  - from: execute
+    to: END
+""",
+        encoding="utf-8",
+    )
+    actions = WorkflowActionRegistry()
+    actions.register("validate", lambda params, state: {"valid": params["order_id"] == "123"})
+    actions.register("execute", lambda params, state: {"protocol": "P-1"})
+    runtime = WorkflowRuntime(FileWorkflowRepository(tmp_path), actions=actions)
+
+    ok = await runtime.arun("refund", {"order_id": "123"})
+    assert ok.status == "COMPLETED"
+    assert ok.output["execute"]["protocol"] == "P-1"
+    assert len(runtime._compiled) == 1
+
+    rejected = await runtime.arun("refund", {"order_id": "999"})
+    assert rejected.status == "COMPLETED"
+    assert "execute" not in rejected.output
+    assert len(runtime._compiled) == 1
+
+
+@pytest.mark.asyncio
+async def test_policy_adapter_runs_only_workflow_mode(tmp_path: Path):
+    (tmp_path / "job.active.yaml").write_text("version: 1\n", encoding="utf-8")
+    (tmp_path / "job.v1.yaml").write_text(
+        """name: job
+version: 1
+start: one
+nodes:
+  - id: one
+    action: one
+edges:
+  - from: one
+    to: END
+""",
+        encoding="utf-8",
+    )
+    actions = WorkflowActionRegistry()
+    actions.register("one", lambda params, state: {"ok": True})
+    adapter = WorkflowToolExecutor(WorkflowRuntime(FileWorkflowRepository(tmp_path), actions=actions))
+    assert await adapter.execute_from_policy(tool_name="x", arguments={}, policy={"execution": {"mode": "direct_tool"}}) is None
+    result = await adapter.execute_from_policy(tool_name="x", arguments={}, policy={"execution": {"mode": "workflow", "workflow": "job", "version": "active"}})
+    assert result["status"] == "COMPLETED"