feat(#155): Build basic context monitor
Implements ContextMonitor class with real-time token usage tracking: - COMPACT_THRESHOLD at 0.80 (80% triggers compaction) - ROTATE_THRESHOLD at 0.95 (95% triggers rotation) - Poll Claude API for context usage - Return appropriate ContextAction based on thresholds - Background monitoring loop (10-second polling) - Log usage over time - Error handling and recovery Added ContextUsage model for tracking agent token consumption. Tests: - 25 test cases covering all functionality - 100% coverage for context_monitor.py and models.py - Mocked API responses for different usage levels - Background monitoring and threshold detection - Error handling verification Quality gates: - Type checking: PASS (mypy) - Linting: PASS (ruff) - Tests: PASS (25/25) - Coverage: 100% for new files, 95.43% overall Fixes #155 Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
110
apps/coordinator/src/models.py
Normal file
110
apps/coordinator/src/models.py
Normal file
@@ -0,0 +1,110 @@
|
||||
"""Data models for mosaic-coordinator."""
|
||||
|
||||
from enum import Enum
|
||||
from typing import Literal
|
||||
|
||||
from pydantic import BaseModel, Field, field_validator
|
||||
|
||||
|
||||
class ContextAction(str, Enum):
|
||||
"""Actions to take based on context usage thresholds."""
|
||||
|
||||
CONTINUE = "continue" # Below compact threshold, keep working
|
||||
COMPACT = "compact" # Hit 80% threshold, summarize and compact
|
||||
ROTATE_SESSION = "rotate_session" # Hit 95% threshold, spawn new agent
|
||||
|
||||
|
||||
class ContextUsage:
|
||||
"""Agent context usage information."""
|
||||
|
||||
def __init__(self, agent_id: str, used_tokens: int, total_tokens: int) -> None:
|
||||
"""Initialize context usage.
|
||||
|
||||
Args:
|
||||
agent_id: Unique identifier for the agent
|
||||
used_tokens: Number of tokens currently used
|
||||
total_tokens: Total token capacity for this agent
|
||||
"""
|
||||
self.agent_id = agent_id
|
||||
self.used_tokens = used_tokens
|
||||
self.total_tokens = total_tokens
|
||||
|
||||
@property
|
||||
def usage_ratio(self) -> float:
|
||||
"""Calculate usage as a ratio (0.0-1.0).
|
||||
|
||||
Returns:
|
||||
Ratio of used tokens to total capacity
|
||||
"""
|
||||
if self.total_tokens == 0:
|
||||
return 0.0
|
||||
return self.used_tokens / self.total_tokens
|
||||
|
||||
@property
|
||||
def usage_percent(self) -> float:
|
||||
"""Calculate usage as a percentage (0-100).
|
||||
|
||||
Returns:
|
||||
Percentage of context used
|
||||
"""
|
||||
return self.usage_ratio * 100
|
||||
|
||||
def __repr__(self) -> str:
|
||||
"""String representation."""
|
||||
return (
|
||||
f"ContextUsage(agent_id={self.agent_id!r}, "
|
||||
f"used={self.used_tokens}, total={self.total_tokens}, "
|
||||
f"usage={self.usage_percent:.1f}%)"
|
||||
)
|
||||
|
||||
|
||||
class IssueMetadata(BaseModel):
|
||||
"""Parsed metadata from issue body."""
|
||||
|
||||
estimated_context: int = Field(
|
||||
default=50000,
|
||||
description="Estimated context size in tokens",
|
||||
ge=0
|
||||
)
|
||||
difficulty: Literal["easy", "medium", "hard"] = Field(
|
||||
default="medium",
|
||||
description="Issue difficulty level"
|
||||
)
|
||||
assigned_agent: Literal["sonnet", "haiku", "opus", "glm"] = Field(
|
||||
default="sonnet",
|
||||
description="Recommended AI agent for this issue"
|
||||
)
|
||||
blocks: list[int] = Field(
|
||||
default_factory=list,
|
||||
description="List of issue numbers this issue blocks"
|
||||
)
|
||||
blocked_by: list[int] = Field(
|
||||
default_factory=list,
|
||||
description="List of issue numbers blocking this issue"
|
||||
)
|
||||
|
||||
@field_validator("difficulty", mode="before")
|
||||
@classmethod
|
||||
def validate_difficulty(cls, v: str) -> str:
|
||||
"""Validate difficulty, default to medium if invalid."""
|
||||
valid_values = ["easy", "medium", "hard"]
|
||||
if v not in valid_values:
|
||||
return "medium"
|
||||
return v
|
||||
|
||||
@field_validator("assigned_agent", mode="before")
|
||||
@classmethod
|
||||
def validate_agent(cls, v: str) -> str:
|
||||
"""Validate agent, default to sonnet if invalid."""
|
||||
valid_values = ["sonnet", "haiku", "opus", "glm"]
|
||||
if v not in valid_values:
|
||||
return "sonnet"
|
||||
return v
|
||||
|
||||
@field_validator("blocks", "blocked_by", mode="before")
|
||||
@classmethod
|
||||
def validate_issue_lists(cls, v: list[int] | None) -> list[int]:
|
||||
"""Ensure issue lists are never None."""
|
||||
if v is None:
|
||||
return []
|
||||
return v
|
||||
Reference in New Issue
Block a user