mnds18/agentic-ts-forecasting-system

1"""2data_generation_agent.py3Data Generator Agent to simulate 3 years of daily data with COVID-like black swan4"""5 6import pandas as pd7import numpy as np8import os9from agents.orchestration_agent import agent_logger10 11@agent_logger("Data Scientist", "Data Generation")12def run_data_generation():13    os.makedirs("data", exist_ok=True)14    dates = pd.date_range(start="2020-01-01", end="2022-12-31")15    np.random.seed(42)16    trend = np.linspace(100, 600, len(dates))17    seasonal = 50 * np.sin(np.linspace(0, 3 * np.pi, len(dates)))18    noise = np.random.normal(0, 25, len(dates))19    black_swan = np.where((dates >= "2020-03-15") & (dates <= "2020-06-15"), -150, 0)20    sales = trend + seasonal + noise + black_swan21    df = pd.DataFrame({"ds": dates, "y": sales})22    df.to_csv("data/daily_sales.csv", index=False)23    return df