agenttt.py

def normalize_ipo_calendar(ipo_df: pd.DataFrame) -> pd.DataFrame:
    df = ipo_df.copy()
    if df.empty:
        return df

    # Defensive: use `.get()` so missing columns don't raise KeyError.
    df["symbol"] = df.get("symbol")
    df["companyName"] = df.get("companyName")
    df["exchange"] = df.get("exchange")
    df["ipoDate"] = pd.to_datetime(df.get("date"), errors="coerce")

    # Keep only rows with a usable symbol.
    df = df.dropna(subset=["symbol"])
    df["symbol"] = df["symbol"].astype(str).str.strip()

    return df