"""
Simulation agent — runs simulations to test scenarios and predict outcomes.
"""

from __future__ import annotations
from typing import Any, Dict, List, Optional
from .base import Agent, AgentRole


class SimulationAgent(Agent):
    """Agent that models and runs simulations of complex systems."""

    def __init__(self, name: str = "simulator") -> None:
        super().__init__(name, AgentRole.SIMULATOR)
        self._simulations: List[Dict[str, Any]] = []
        self.register_capability("scenario_simulation")
        self.register_capability("monte_carlo_forecasting")

    def act(self, context: Dict[str, Any]) -> Dict[str, Any]:
        scenario = context.get("scenario", "default")
        result = self.simulate(scenario)
        return {"result": result, "iterations": 1000}

    def simulate(self, scenario: str, iterations: int = 100) -> Dict[str, Any]:
        result = {
            "scenario": scenario,
            "success_probability": 0.72,
            "mean_outcome": "positive",
            "variance": 0.15,
            "confidence_interval": (0.65, 0.85),
        }
        self._simulations.append(result)
        return result