Spaces:
Running
Running
File size: 769 Bytes
4ec75cf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | # server/models/security_models.py
from pydantic import BaseModel, Field
from typing import Optional
class IdentifyVulnerability(BaseModel):
action_type: str = 'identify_vulnerability'
vuln_type: str = Field(..., description='Type of vulnerability detected')
cvss_score: float = Field(..., ge=0.0, le=10.0)
severity: str = Field(..., description='critical|high|medium|low')
affected_line: int = Field(..., ge=1)
class ProposeFix(BaseModel):
action_type: str = 'propose_fix'
fix_code: str = Field(..., max_length=500)
explanation: str = Field(..., max_length=200)
class ReviseFix(BaseModel):
action_type: str = 'revise_fix'
fix_code: str = Field(..., max_length=500)
addressed_feedback: str = Field(..., max_length=200)
|