Initial commit: add .gitignore and README
This commit is contained in:
35
fusionagi/core/blockers.py
Normal file
35
fusionagi/core/blockers.py
Normal file
@@ -0,0 +1,35 @@
|
||||
"""Blockers and checkpoints for AGI state machine."""
|
||||
|
||||
from typing import Any, Protocol
|
||||
|
||||
from fusionagi.schemas.goal import Blocker, Checkpoint
|
||||
from fusionagi._logger import logger
|
||||
|
||||
|
||||
class BlockersAndCheckpoints:
|
||||
"""Tracks blockers (why stuck) and checkpoints (resumable points)."""
|
||||
|
||||
def __init__(self) -> None:
|
||||
self._blockers: dict[str, list[Blocker]] = {}
|
||||
self._checkpoints: dict[str, list[Checkpoint]] = {}
|
||||
|
||||
def add_blocker(self, blocker: Blocker) -> None:
|
||||
self._blockers.setdefault(blocker.task_id, []).append(blocker)
|
||||
logger.info("Blocker added", extra={"task_id": blocker.task_id, "reason": blocker.reason[:80] if blocker.reason else ""})
|
||||
|
||||
def get_blockers(self, task_id: str) -> list[Blocker]:
|
||||
return list(self._blockers.get(task_id, []))
|
||||
|
||||
def clear_blockers(self, task_id: str) -> None:
|
||||
self._blockers.pop(task_id, None)
|
||||
|
||||
def add_checkpoint(self, checkpoint: Checkpoint) -> None:
|
||||
self._checkpoints.setdefault(checkpoint.task_id, []).append(checkpoint)
|
||||
logger.debug("Checkpoint added", extra={"task_id": checkpoint.task_id})
|
||||
|
||||
def get_latest_checkpoint(self, task_id: str) -> Checkpoint | None:
|
||||
lst = self._checkpoints.get(task_id, [])
|
||||
return lst[-1] if lst else None
|
||||
|
||||
def list_checkpoints(self, task_id: str) -> list[Checkpoint]:
|
||||
return list(self._checkpoints.get(task_id, []))
|
||||
Reference in New Issue
Block a user