feat: import 6 MCP services + common workspace

2026-04-27 17:34:14 +02:00
parent 9676f22a8e
commit 6fc3d1d94f
67 changed files with 10693 additions and 0 deletions
@@ -0,0 +1,136 @@
+from __future__ import annotations
+
+import os
+
+from fastapi import Depends, FastAPI, HTTPException
+from option_mcp_common.auth import Principal, TokenStore, require_principal
+from option_mcp_common.mcp_bridge import mount_mcp_endpoint
+from option_mcp_common.server import build_app
+from pydantic import BaseModel
+
+from mcp_macro.fetchers import (
+    fetch_asset_price,
+    fetch_economic_indicators,
+    fetch_equity_futures,
+    fetch_macro_calendar,
+    fetch_market_overview,
+    fetch_treasury_yields,
+)
+
+# --- Body models ---
+
+class GetEconomicIndicatorsReq(BaseModel):
+    indicators: list[str] | None = None
+
+
+class GetMacroCalendarReq(BaseModel):
+    days: int = 7
+    country_filter: list[str] | None = None
+    importance_min: str | None = None
+    start: str | None = None
+    end: str | None = None
+
+
+class GetMarketOverviewReq(BaseModel):
+    pass
+
+
+class GetAssetPriceReq(BaseModel):
+    ticker: str
+
+
+class GetTreasuryYieldsReq(BaseModel):
+    pass
+
+
+class GetEquityFuturesReq(BaseModel):
+    pass
+
+
+# --- ACL helper ---
+
+def _check(principal: Principal, *, core: bool = False, observer: bool = False) -> None:
+    allowed: set[str] = set()
+    if core:
+        allowed.add("core")
+    if observer:
+        allowed.add("observer")
+    if not (principal.capabilities & allowed):
+        raise HTTPException(403, f"capability required: {allowed}")
+
+
+# --- App factory ---
+
+def create_app(*, fred_api_key: str = "", finnhub_api_key: str = "", token_store: TokenStore) -> FastAPI:
+    app = build_app(name="mcp-macro", version="0.1.0", token_store=token_store)
+
+    @app.post("/tools/get_economic_indicators", tags=["reads"])
+    async def t_get_economic_indicators(
+        body: GetEconomicIndicatorsReq, principal: Principal = Depends(require_principal)
+    ):
+        _check(principal, core=True, observer=True)
+        return await fetch_economic_indicators(
+            fred_api_key=fred_api_key, indicators=body.indicators
+        )
+
+    @app.post("/tools/get_macro_calendar", tags=["reads"])
+    async def t_get_macro_calendar(
+        body: GetMacroCalendarReq, principal: Principal = Depends(require_principal)
+    ):
+        _check(principal, core=True, observer=True)
+        return await fetch_macro_calendar(
+            finnhub_api_key=finnhub_api_key,
+            days_ahead=body.days,
+            country_filter=body.country_filter,
+            importance_min=body.importance_min,
+            start=body.start,
+            end=body.end,
+        )
+
+    @app.post("/tools/get_market_overview", tags=["reads"])
+    async def t_get_market_overview(
+        body: GetMarketOverviewReq, principal: Principal = Depends(require_principal)
+    ):
+        _check(principal, core=True, observer=True)
+        return await fetch_market_overview()
+
+    @app.post("/tools/get_asset_price", tags=["reads"])
+    async def t_get_asset_price(
+        body: GetAssetPriceReq, principal: Principal = Depends(require_principal)
+    ):
+        _check(principal, core=True, observer=True)
+        return await fetch_asset_price(body.ticker)
+
+    @app.post("/tools/get_treasury_yields", tags=["reads"])
+    async def t_get_treasury_yields(
+        body: GetTreasuryYieldsReq, principal: Principal = Depends(require_principal)
+    ):
+        _check(principal, core=True, observer=True)
+        return await fetch_treasury_yields()
+
+    @app.post("/tools/get_equity_futures", tags=["reads"])
+    async def t_get_equity_futures(
+        body: GetEquityFuturesReq, principal: Principal = Depends(require_principal)
+    ):
+        _check(principal, core=True, observer=True)
+        return await fetch_equity_futures()
+
+    # ───── MCP endpoint (/mcp) — bridge verso /tools/* ─────
+    port = int(os.environ.get("PORT", "9013"))
+    mount_mcp_endpoint(
+        app,
+        name="cerbero-macro",
+        version="0.1.0",
+        token_store=token_store,
+        internal_base_url=f"http://localhost:{port}",
+        tools=[
+            {"name": "get_economic_indicators", "description": "FRED economic indicators (Fed rate, CPI, ecc)."},
+            {"name": "get_macro_calendar", "description": "Eventi macro con filtri country/importance/date range."},
+            {"name": "get_market_overview", "description": "Snapshot overview mercato macro."},
+            {"name": "get_asset_price", "description": "Prezzo cross-asset: WTI, DXY, SPX, VIX, yields, FX, ecc."},
+            {"name": "get_treasury_yields", "description": "Curva US Treasury 2y/5y/10y/30y + shape detection."},
+            {"name": "get_equity_futures", "description": "Futures ES/NQ/YM/RTY con session status."},
+        ],
+    )
+
+    return app