Upload folder using huggingface_hub

models.py

@@ -184,19 +184,55 @@ class RewardSummary(BaseModel):
 class PythonReviewAction(Action):
     """Structured review action emitted by a model or trainer."""
 
-    # Primary UI Fields (7 benchmark + 3 template = 10 total)
     operation: str = Field(default="ADD_COMMENT", description="The operation to perform.")
     findings: List[ReviewFinding] = Field(default_factory=list, description="The findings list.")
     patched_code: Optional[str] = Field(default=None, description="The fixed source code.")
     
-    action_type: ActionType = ActionType.ADD_COMMENT
-    line_number: Optional[int] = Field(default=None, ge=1)
-    issue_type: Optional[IssueType] = None
-    severity: Optional[Severity] = None
-    comment: Optional[str] = None
+    action_type: ActionType = Field(
+        default=ActionType.ADD_COMMENT,
+        description="Choose the review action: comment on a line, skip a clean line, ask for context, approve, or request changes.",
+    )
+    line_number: Optional[int] = Field(
+        default=None,
+        ge=1,
+        description="Required for ADD_COMMENT and SKIP_LINE. Enter the code line number you are acting on.",
+    )
+    issue_type: Optional[IssueType] = Field(
+        default=None,
+        description="Required for ADD_COMMENT. Pick the issue category for the selected line.",
+    )
+    severity: Optional[Severity] = Field(
+        default=None,
+        description="Required for ADD_COMMENT. Pick how serious the issue is.",
+    )
+    comment: Optional[str] = Field(
+        default=None,
+        description="Required for ADD_COMMENT. Also used for ASK_CONTEXT if you want to ask a question in plain text.",
+    )
     suggestion: Optional[str] = None
     question: Optional[str] = None
 
+    @classmethod
+    def model_json_schema(cls, *args, **kwargs):
+        """Trim legacy fields from the generated UI schema."""
+
+        schema = super().model_json_schema(*args, **kwargs)
+        properties = schema.get("properties", {})
+        visible_fields = {
+            "action_type",
+            "line_number",
+            "issue_type",
+            "severity",
+            "comment",
+        }
+        schema["properties"] = {
+            name: value for name, value in properties.items() if name in visible_fields
+        }
+        schema["required"] = [
+            name for name in schema.get("required", []) if name in visible_fields
+        ]
+        return schema
+
     @model_validator(mode="after")
     def validate_action_shape(self) -> "PythonReviewAction":
         """Require the right fields for each action type."""