app.py

from __future__ import annotations

import os
import sys
from pathlib import Path
from typing import Any

from fastapi import FastAPI

CURRENT_DIR = Path(__file__).resolve().parent
PLUGIN_DIR = CURRENT_DIR / "causalml" / "mcp_output" / "mcp_plugin"
if str(PLUGIN_DIR) not in sys.path:
    sys.path.insert(0, str(PLUGIN_DIR))

app = FastAPI(title="causalml MCP Info API", version="1.0.0")


@app.get("/")
def root() -> dict[str, Any]:
    return {
        "service": "causalml-mcp",
        "description": "Supplementary FastAPI info service for local development.",
        "mcp_http_endpoint": "/mcp",
        "note": "This app is not the MCP server entrypoint.",
    }


@app.get("/health")
def health() -> dict[str, str]:
    return {"status": "healthy"}


@app.get("/tools")
def tools() -> dict[str, Any]:
    from mcp_service import create_app

    mcp = create_app()
    discovered: list[dict[str, Any]] = []
    for tool in getattr(mcp, "tools", []):
        discovered.append(
            {
                "name": getattr(tool, "name", getattr(tool, "__name__", "unknown")),
                "description": getattr(tool, "description", ""),
            }
        )
    return {"count": len(discovered), "tools": discovered}


PORT = int(os.getenv("PORT", "7860"))


if __name__ == "__main__":
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=PORT)