445865e429
Some checks failed
- Integrate GPU scoring inline into reasoning/multi_path.py (auto-uses GPU when available) - Integrate GPU deduplication into multi_agent/consensus_engine.py - Add semantic_search() method to memory/semantic_graph.py with GPU acceleration - Integrate GPU training into self_improvement/training.py AutoTrainer - Fix all 758 ruff lint issues (whitespace, import sorting, unused imports, ambiguous vars, undefined names) - Fix all 40 mypy type errors across the codebase (no-any-return, union-attr, arg-type, etc.) - Fix deprecated ruff config keys (select/ignore -> [tool.ruff.lint]) - Add .dockerignore to exclude .venv/, tests/, docs/ from Docker builds - Add type hints and docstrings to verification/outcome.py - Fix E402 import ordering in witness_agent.py - Fix F821 undefined names in vector_pgvector.py and native.py - Fix E741 ambiguous variable names in reflective.py and recommender.py All 276 tests pass. 0 ruff errors. 0 mypy errors. Co-Authored-By: Nakamoto, S <defi@defi-oracle.io>
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
"""Blockers and checkpoints for AGI state machine."""
|
|
|
|
|
|
from fusionagi._logger import logger
|
|
from fusionagi.schemas.goal import Blocker, Checkpoint
|
|
|
|
|
|
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, []))
|