55 lines
2.9 KiB
Python
55 lines
2.9 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from graph.spec import ExecutionGraphSpec, GraphEdgeSpec, GraphNodeSpec
|
||
|
|
|
||
|
|
|
||
|
|
TASK_EXECUTION_GRAPH_NAME = "task_execution"
|
||
|
|
TASK_EXECUTION_GRAPH_VERSION = 1
|
||
|
|
|
||
|
|
|
||
|
|
def task_execution_graph_v1() -> ExecutionGraphSpec:
|
||
|
|
nodes = {
|
||
|
|
"claim_task": GraphNodeSpec("claim_task", "claim_task", {"idempotent": False, "replay_safe": False}),
|
||
|
|
"prepare_worktree": GraphNodeSpec("prepare_worktree", "prepare_worktree", {"idempotent": True, "replay_safe": True}),
|
||
|
|
"build_context": GraphNodeSpec("build_context", "build_context", {"idempotent": True, "replay_safe": True}),
|
||
|
|
"coder": GraphNodeSpec("coder", "coder", {"idempotent": False, "replay_safe": "checkpointed", "contains_internal_tool_loop": True}),
|
||
|
|
"run_tests": GraphNodeSpec("run_tests", "run_tests", {"idempotent": True, "replay_safe": "checkpointed"}),
|
||
|
|
"review": GraphNodeSpec("review", "review", {"idempotent": True, "replay_safe": "checkpointed"}),
|
||
|
|
"judge": GraphNodeSpec("judge", "judge", {"idempotent": True, "replay_safe": "checkpointed"}),
|
||
|
|
"commit": GraphNodeSpec("commit", "commit", {"idempotent": False, "replay_safe": "guarded", "destructive": True}),
|
||
|
|
"retry_or_fail": GraphNodeSpec("retry_or_fail", "retry_or_fail", {"idempotent": False, "replay_safe": "checkpointed"}),
|
||
|
|
"cleanup": GraphNodeSpec("cleanup", "cleanup", {"idempotent": True, "replay_safe": True}),
|
||
|
|
"complete": GraphNodeSpec("complete", "complete", {"terminal": True}),
|
||
|
|
"fail": GraphNodeSpec("fail", "fail", {"terminal": True}),
|
||
|
|
}
|
||
|
|
edges = [
|
||
|
|
GraphEdgeSpec("claim_task", "prepare_worktree", "success"),
|
||
|
|
GraphEdgeSpec("prepare_worktree", "build_context", "success"),
|
||
|
|
GraphEdgeSpec("build_context", "coder", "success"),
|
||
|
|
GraphEdgeSpec("coder", "run_tests", "success"),
|
||
|
|
GraphEdgeSpec("coder", "retry_or_fail", "failure"),
|
||
|
|
GraphEdgeSpec("run_tests", "review", "complete"),
|
||
|
|
GraphEdgeSpec("review", "judge", "PASS"),
|
||
|
|
GraphEdgeSpec("review", "retry_or_fail", "REWORK_REQUIRED"),
|
||
|
|
GraphEdgeSpec("review", "retry_or_fail", "REJECTED"),
|
||
|
|
GraphEdgeSpec("judge", "commit", "PASS"),
|
||
|
|
GraphEdgeSpec("judge", "retry_or_fail", "FAIL"),
|
||
|
|
GraphEdgeSpec("commit", "cleanup", "success"),
|
||
|
|
GraphEdgeSpec("cleanup", "complete", "success"),
|
||
|
|
GraphEdgeSpec("retry_or_fail", "build_context", "retry_available"),
|
||
|
|
GraphEdgeSpec("retry_or_fail", "cleanup", "retry_exhausted"),
|
||
|
|
GraphEdgeSpec("cleanup", "fail", "failed"),
|
||
|
|
]
|
||
|
|
spec = ExecutionGraphSpec(
|
||
|
|
name=TASK_EXECUTION_GRAPH_NAME,
|
||
|
|
version=TASK_EXECUTION_GRAPH_VERSION,
|
||
|
|
graph_type="TASK_EXECUTION",
|
||
|
|
entry="claim_task",
|
||
|
|
nodes=nodes,
|
||
|
|
edges=edges,
|
||
|
|
terminal_nodes=["complete", "fail"],
|
||
|
|
metadata={"description": "Task execution graph v1 mirrors AutonomousTaskLoop semantics."},
|
||
|
|
)
|
||
|
|
spec.validate()
|
||
|
|
return spec
|