feat: causal lab

This commit is contained in:
OpenSquared
2026-06-28 18:08:09 +02:00
parent c286c7c000
commit 6ebbf4326e
5 changed files with 481 additions and 133 deletions

View File

@@ -2,6 +2,9 @@
Instrument Dashboard Router.
Exposes per-instrument snapshot (price, indicators, regime, trend, events) and AI narrative.
"""
import json
import math
from datetime import datetime, timedelta
from fastapi import APIRouter, HTTPException, Query
from pydantic import BaseModel
from typing import List, Dict, Any, Optional
@@ -85,3 +88,142 @@ def update_drivers(instrument_id: str, body: DriverUpdate) -> Dict[str, Any]:
raise HTTPException(status_code=500, detail=str(e))
return {"ok": True, "instrument_id": instrument_id.upper(), "drivers_count": len(body.drivers)}
# ── Instrument mult (pips → price conversion) ─────────────────────────────────
_INST_MULT: Dict[str, int] = {"EURUSD": 10000, "GBPUSD": 10000, "USDJPY": 100, "AUDUSD": 10000}
def _get_mult(inst: str) -> int:
return _INST_MULT.get(inst.upper(), 10)
def _decay(days_after: int, absorption_days: int, decay_type: str) -> float:
"""Decay factor ∈ [0,1] for a given number of days after the event."""
if days_after < 0:
return 0.0
if decay_type == "step":
return 1.0 if days_after <= absorption_days else 0.0
elif decay_type == "linear":
return max(0.0, 1.0 - days_after / max(absorption_days, 1))
else: # exp — 3 time-constants reach ~5% at absorption_days
lam = 3.0 / max(absorption_days, 1)
return math.exp(-lam * days_after)
@router.get("/{instrument_id}/theoretical-curve")
def get_theoretical_curve(
instrument_id: str,
period: str = Query("1y"),
) -> List[Dict[str, Any]]:
"""
Courbe théorique composite : pour chaque jour calendaire de la période,
somme des impacts décroissants des analyses causales stockées.
Retourne [{date, cumulative_pips, contributions: [{template_name, event_name, event_date, pips, decay_factor}]}]
"""
from services.database import get_conn
period_lookback: Dict[str, int] = {
"5d": 7, "1mo": 35, "3mo": 95, "6mo": 190,
"1y": 370, "2y": 740, "5y": 1830,
}
lookback = period_lookback.get(period, 370)
date_to = datetime.utcnow().date()
date_from = date_to - timedelta(days=lookback)
# Fetch events that started before date_from too — they may still be decaying into the window
extended_from = date_from - timedelta(days=90)
inst_upper = instrument_id.upper()
conn = get_conn()
try:
rows = conn.execute("""
SELECT a.id,
a.prediction_json,
a.activation_score,
e.start_date AS event_date,
e.name AS event_name,
t.name AS template_name,
t.calibration_json
FROM causal_event_analyses a
JOIN market_events e ON e.id = a.market_event_id
JOIN causal_graph_templates t ON t.id = a.template_id
WHERE a.instrument = ?
AND e.start_date >= ?
AND e.start_date <= ?
ORDER BY e.start_date
""", (inst_upper, str(extended_from), str(date_to))).fetchall()
finally:
conn.close()
# ── Build calendar-day series ─────────────────────────────────────────────
all_dates: List[str] = []
cur = date_from
while cur <= date_to:
all_dates.append(str(cur))
cur += timedelta(days=1)
curve: Dict[str, Dict] = {
d: {"cumulative_pips": 0.0, "contributions": []} for d in all_dates
}
for row in rows:
r = dict(row)
try:
predictions = json.loads(r["prediction_json"] or "{}")
calib = json.loads(r["calibration_json"] or "{}")
except Exception:
continue
# Extract predicted pips for this instrument from node_values dict
inst_lower = inst_upper.lower()
predicted_pips: Optional[float] = None
if inst_lower in predictions:
predicted_pips = float(predictions[inst_lower])
else:
for k, v in predictions.items():
if inst_lower in k.lower():
try:
predicted_pips = float(v)
break
except (TypeError, ValueError):
pass
if predicted_pips is None or predicted_pips == 0:
continue
absorption_days: int = max(1, int(calib.get("absorption_days", 7)))
dtype: str = str(calib.get("decay_type", "exp"))
event_date_str: str = r["event_date"][:10]
try:
event_date = datetime.strptime(event_date_str, "%Y-%m-%d").date()
except ValueError:
continue
for d in all_dates:
cal_date = datetime.strptime(d, "%Y-%m-%d").date()
days_after = (cal_date - event_date).days
df = _decay(days_after, absorption_days, dtype)
if df < 0.01:
continue
contribution = round(predicted_pips * df, 2)
curve[d]["cumulative_pips"] += contribution
curve[d]["contributions"].append({
"template_name": r["template_name"],
"event_name": r["event_name"],
"event_date": event_date_str,
"pips": contribution,
"decay_factor": round(df, 3),
})
# Round totals and strip empty-contribution days at the edges
result = []
for d in all_dates:
entry = curve[d]
entry["cumulative_pips"] = round(entry["cumulative_pips"], 1)
result.append({"date": d, **entry})
return result