diff --git a/agent.py b/agent.py index 2b79b3f..6d402b8 100644 --- a/agent.py +++ b/agent.py @@ -210,52 +210,13 @@ class GenerateRemediationRequest(StrictModel): risk_summary: dict[str, Any] | None = None -class InventoryResult(StrictModel): - audit_id: str - target: AuditTarget - status: Literal["ok", "partial", "unreachable", "setup_required"] - output_dir: str - generated_files: list[OutputFile] = Field(default_factory=list) - inventory: dict[str, Any] - evidence: list[Evidence] = Field(default_factory=list) - warnings: list[str] = Field(default_factory=list) - mutations_performed: bool = False +AgentStatus = Literal["ok", "partial", "unreachable", "setup_required"] -class FindingResult(StrictModel): - audit_id: str - target: AuditTarget - status: Literal["ok", "partial", "setup_required"] - output_dir: str - generated_files: list[OutputFile] = Field(default_factory=list) - findings: list[Finding] - evidence: list[Evidence] = Field(default_factory=list) - warnings: list[str] = Field(default_factory=list) - mutations_performed: bool = False +class AgentComplianceAuditorResult(dict[str, Any]): + """Compact JSON-shaped result alias for a smaller Agent Card output schema.""" -class RiskSummaryResult(StrictModel): - audit_id: str - target: AuditTarget - status: Literal["ok", "partial"] - output_dir: str - generated_files: list[OutputFile] = Field(default_factory=list) - risk_summary: dict[str, Any] - findings: list[Finding] - warnings: list[str] = Field(default_factory=list) - mutations_performed: bool = False - - -class RemediationResult(StrictModel): - audit_id: str - target: AuditTarget - status: Literal["ok", "partial"] - output_dir: str - generated_files: list[OutputFile] = Field(default_factory=list) - remediation_markdown: str - sarif: dict[str, Any] - warnings: list[str] = Field(default_factory=list) - mutations_performed: bool = False class AgentComplianceAuditorConfig(StrictModel):