junaid0600 commited on
Commit
1a89fae
·
1 Parent(s): ef20791

Fix GraderResponse schema example score from 1 to 0.75

Browse files
Files changed (1) hide show
  1. 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: float = Field(..., gt=0.0, lt=1.0)
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(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"