Spaces:
Sleeping
Sleeping
File size: 2,651 Bytes
b09b8a3 e63c592 b09b8a3 e63c592 b09b8a3 e63c592 b09b8a3 e63c592 b09b8a3 e63c592 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | from fastapi import Depends, FastAPI
from fastapi.responses import ORJSONResponse
from app.core.auth import require_api_key, validate_api_key_configuration
from app.core.config import get_settings
from app.core.errors import PineconeIndexConfigError, setup_exception_handlers
from app.core.logging import configure_logging, get_logger
from app.core.metrics import setup_metrics
from app.core.rate_limit import setup_rate_limiter
from app.core.runtime import get_port
from app.core.security import configure_security
from app.routers.documents import router as documents_router
from app.routers.health import router as health_router
from app.routers.ingest import router as ingest_router
from app.routers.search import router as search_router
from app.routers.chat import router as chat_router
from app.routers.metrics import router as metrics_router
from app.services.pinecone_store import init_pinecone
settings = get_settings()
configure_logging(settings.LOG_LEVEL)
logger = get_logger(__name__)
# Validate API key configuration early so hosted deployments fail fast when misconfigured.
validate_api_key_configuration()
# Log runtime port / environment context at import time for easier diagnostics.
get_port()
app = FastAPI(
title="RAG Agent Workbench API",
version=settings.APP_VERSION,
default_response_class=ORJSONResponse,
docs_url="/docs",
redoc_url="/redoc",
openapi_url="/openapi.json",
)
# Core app configuration
configure_security(app)
setup_rate_limiter(app)
setup_metrics(app)
# Register routers with tags and ensure they are included in the schema.
# Health and docs remain public; all other routers are protected by API key dependency when configured.
app.include_router(health_router, tags=["health"])
app.include_router(ingest_router, tags=["ingest"], dependencies=[Depends(require_api_key)])
app.include_router(search_router, tags=["search"], dependencies=[Depends(require_api_key)])
app.include_router(documents_router, tags=["documents"], dependencies=[Depends(require_api_key)])
app.include_router(chat_router, tags=["chat"], dependencies=[Depends(require_api_key)])
app.include_router(metrics_router, tags=["metrics"], dependencies=[Depends(require_api_key)])
# Register exception handlers
setup_exception_handlers(app)
@app.on_event("startup")
async def startup_event() -> None:
"""Application startup hook."""
try:
init_pinecone(settings)
logger.info("Pinecone initialisation completed")
except PineconeIndexConfigError:
# Let the exception handler and FastAPI/uvicorn deal with the error.
# Re-raise to fail fast on misconfiguration.
raise |