a2a-source-edit: write agent.py

agent.py

@@ -210,52 +210,13 @@ class GenerateRemediationRequest(StrictModel):
     risk_summary: dict[str, Any] | None = None
 
 
-class InventoryResult(StrictModel):
+AgentStatus = Literal["ok", "partial", "unreachable", "setup_required"]
-    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
 
 
-class FindingResult(StrictModel):
+class AgentComplianceAuditorResult(dict[str, Any]):
-    audit_id: str
+    """Compact JSON-shaped result alias for a smaller Agent Card output schema."""
-    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 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):