mirror of
https://github.com/mblanke/GooseStrike.git
synced 2026-03-01 05:50:22 -05:00
29 lines
945 B
Python
29 lines
945 B
Python
"""CVE triage agent."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Dict
|
|
|
|
from .base_agent import AgentResult, BaseAgent
|
|
|
|
|
|
class CVEAgent(BaseAgent):
|
|
name = "cve"
|
|
|
|
def build_prompt(self, context: Dict[str, Any]) -> str:
|
|
cves = context.get("cves", [])
|
|
lines = ["You are prioritizing CVEs for a legal assessment."]
|
|
for cve in cves:
|
|
lines.append(
|
|
f"{cve.get('cve_id')}: severity={cve.get('severity')} score={cve.get('score')} desc={cve.get('description','')[:120]}"
|
|
)
|
|
lines.append("Provide prioritized actions and validation steps. No exploit code.")
|
|
return "\n".join(lines)
|
|
|
|
def parse(self, raw: str) -> Dict[str, Any]:
|
|
recommendations = [line.strip() for line in raw.split('\n') if line.strip()]
|
|
return {"cve_actions": recommendations}
|
|
|
|
|
|
def run(context: Dict[str, Any]) -> AgentResult:
|
|
return CVEAgent().run(context)
|