| """ | |
| Coordinator Agent module for vulnerability intelligence. | |
| This agent is responsible for coordinating the other agents and generating the final report. | |
| """ | |
| import json | |
| import time | |
| import argparse | |
| from typing import Dict, List, Any, Optional, Union | |
| from smolagents import tool | |
| from ..tools import utils | |
| from ..tools.parsers import CWEParser | |
| from . import cve_agent, nvd_agent, cisa_agent, cwe_agent | |
| logger = utils.setup_logger("coordinator_agent") | |
| # Enhance with CWE information | |
| for vuln in merged_results.get("vulnerabilities", []): | |
| if "description" in vuln: | |
| # Try to extract CWEs from the description | |
| cwe_ids = CWEParser.extract_cwe_from_cve(vuln["description"]) | |
| if cwe_ids: | |
| # ... rest of the code ... |