Spaces:
Sleeping
Sleeping
Commit ·
1a89fae
1
Parent(s): ef20791
Fix GraderResponse schema example score from 1 to 0.75
Browse files- env/models.py +11 -3
env/models.py
CHANGED
|
@@ -185,14 +185,22 @@ class GraderRequest(BaseModel):
|
|
| 185 |
episode: Optional[dict] = None
|
| 186 |
|
| 187 |
class GraderResponse(BaseModel):
|
| 188 |
-
score:
|
| 189 |
-
feedback:
|
| 190 |
breakdown: dict
|
| 191 |
|
| 192 |
@field_validator("score")
|
| 193 |
@classmethod
|
| 194 |
def clamp_score(cls, v):
|
| 195 |
-
return max(0.001, min(0.999, round(v, 4)))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 196 |
|
| 197 |
class HealthResponse(BaseModel):
|
| 198 |
status: str = "ok"
|
|
|
|
| 185 |
episode: Optional[dict] = None
|
| 186 |
|
| 187 |
class GraderResponse(BaseModel):
|
| 188 |
+
score: float = Field(..., description="Score strictly between 0 and 1 exclusive")
|
| 189 |
+
feedback: str
|
| 190 |
breakdown: dict
|
| 191 |
|
| 192 |
@field_validator("score")
|
| 193 |
@classmethod
|
| 194 |
def clamp_score(cls, v):
|
| 195 |
+
return max(0.001, min(0.999, round(float(v), 4)))
|
| 196 |
+
|
| 197 |
+
model_config = {"json_schema_extra": {
|
| 198 |
+
"example": {
|
| 199 |
+
"score": 0.75,
|
| 200 |
+
"feedback": "Correct fix applied.",
|
| 201 |
+
"breakdown": {"fix_correctness": 0.5, "explanation": 0.15, "confidence": 0.05}
|
| 202 |
+
}
|
| 203 |
+
}}
|
| 204 |
|
| 205 |
class HealthResponse(BaseModel):
|
| 206 |
status: str = "ok"
|