Gitrew / models.py
at0day's picture
Upload 17 files
1c78613 verified
Raw History Blame Contribute Delete
1.22 kB
"""
Data models for the Git Commit Reviewer Environment.
"""
from typing import List, Optional
from openenv.core.env_server.types import Action, Observation
from pydantic import Field
class GitCommitReviewerAction(Action):
"""Action for the Git Commit Reviewer environment."""
score: float = Field(..., description="Quality score from 0.0 to 1.0")
type_valid: bool = Field(..., description="Does it follow Conventional Commits type rules?")
summary_length: int = Field(..., description="Length of the summary line")
issues: List[str] = Field(..., description="List of specific problems identified")
suggestion: str = Field(..., description="Rewritten commit message")
class GitCommitReviewerObservation(Observation):
"""Observation from the Git Commit Reviewer environment."""
commit_message: str = Field(default="", description="The git commit message to review")
commit_idx: int = Field(default=0, description="Current commit index")
total_commits: int = Field(default=0, description="Total commits in this task")
done: bool = Field(default=False, description="Is the task complete?")
reward: float = Field(default=0.0, description="Reward from the latest action")