Spaces:
Sleeping
Sleeping
| """ | |
| SQLAlchemy ORM models for tasks, episodes, and watch channels. | |
| """ | |
| from datetime import datetime | |
| from typing import List, Optional | |
| from sqlalchemy import Column, String, DateTime, Float, Boolean, JSON | |
| from sqlalchemy.sql import func | |
| from debug_env.server.database.db import Base | |
| class TaskRecord(Base): | |
| """ | |
| Task record — mirrors TASK_REGISTRY in database. | |
| Each row represents a debuggable task available in the benchmark. | |
| Seeded from TASK_REGISTRY at application startup. | |
| """ | |
| __tablename__ = "task_records" | |
| id = Column(String, primary_key=True, index=True) # e.g., "task1" | |
| title = Column(String, nullable=False) | |
| description = Column(String, nullable=False) | |
| difficulty = Column(String, nullable=False) # e.g., "easy", "medium", "hard" | |
| bug_type = Column(String, nullable=False) # e.g., "syntax", "logic", "type_error" | |
| files = Column(JSON, nullable=False) # List of filenames | |
| tools = Column(JSON, nullable=True) # List of available tools for this task | |
| scenario_type = Column(String, nullable=True) # e.g., "code_review", "cross_team_collaboration" | |
| participants = Column(JSON, nullable=True) # List of participant dicts for multi-user tasks | |
| etag = Column(String, nullable=False) # Cache validation tag | |
| created_at = Column(DateTime, default=datetime.utcnow, nullable=False) | |
| updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False) | |
| def __repr__(self) -> str: | |
| return f"<TaskRecord(id={self.id}, title={self.title}, difficulty={self.difficulty})>" | |
| class EpisodeRecord(Base): | |
| """ | |
| Episode record — tracks each task run (reset/step sequence). | |
| One row created per TaskLoader.load() call (when DebugEnvironment.reset() is called). | |
| Tracks workdir, status, and pass_rate for the episode lifecycle. | |
| """ | |
| __tablename__ = "episode_records" | |
| id = Column(String, primary_key=True, index=True) # UUID4 string | |
| task_id = Column(String, nullable=False, index=True) # e.g., "task1" (FK to task_records.id) | |
| user_id = Column(String, nullable=False, index=True) # e.g., "default_user" or from X-User-Id header | |
| workdir = Column(String, nullable=True) # Path to temp directory with task files | |
| status = Column( | |
| String, | |
| nullable=False, | |
| default="active", | |
| index=True, | |
| ) # "active", "passed", "failed", "abandoned" | |
| pass_rate = Column(Float, nullable=True) # 0.0 to 1.0, or None if not completed | |
| etag = Column(String, nullable=False) # Cache validation tag | |
| created_at = Column(DateTime, default=datetime.utcnow, nullable=False, index=True) | |
| updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False) | |
| def __repr__(self) -> str: | |
| return f"<EpisodeRecord(id={self.id}, task_id={self.task_id}, status={self.status})>" | |
| class WatchChannel(Base): | |
| """ | |
| Watch channel record — tracks webhook subscriptions for task/episode state changes. | |
| Allows clients to subscribe to notifications when tasks or episodes change. | |
| Lightweight structure for future expansion of notification features. | |
| """ | |
| __tablename__ = "watch_channels" | |
| id = Column(String, primary_key=True, index=True) # Client-provided ID | |
| task_id = Column(String, nullable=False, index=True) # Task being watched | |
| user_id = Column(String, nullable=False, index=True) # User who set up the watch | |
| webhook_address = Column(String, nullable=False) # URL to POST notifications to | |
| webhook_token = Column(String, nullable=True) # Optional auth token for webhook | |
| expires_at = Column(DateTime, nullable=True, index=True) # When this watch expires | |
| is_active = Column(Boolean, default=True, nullable=False) # Whether watch is currently active | |
| created_at = Column(DateTime, default=datetime.utcnow, nullable=False) | |
| updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False) | |
| def __repr__(self) -> str: | |
| return f"<WatchChannel(id={self.id}, task_id={self.task_id}, is_active={self.is_active})>" | |