from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from retrieval import router as retrieval_router
from fastapi_mcp import FastApiMCP

app = FastAPI(title="Insurance RAG API", version="1.0.0")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(retrieval_router)

@app.get("/")
async def root():
    return {"message": "Insurance RAG API is running", "docs": "/docs"}

# MCP integration
mcp = FastApiMCP(app, name="Insurance MCP Connector")
mcp.mount()