group_fqcd_jr/app/main.py

from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse

from app.api.controllers.admin import router as admin_router
from app.api.controllers.agent_runs import router as agent_runs_router
from app.api.controllers.conversations import router as conversations_router
from app.api.controllers.health import router as health_router
from app.api.controllers.investment_goals import router as investment_goals_router
from app.api.controllers.knowledge import router as knowledge_router
from app.api.controllers.local_test import router as local_test_router
from app.api.controllers.market_data import router as market_data_router
from app.api.controllers.offsite_fund import operation_router as offsite_operation_router
from app.api.controllers.offsite_fund import router as offsite_fund_router
from app.api.controllers.onboarding import router as onboarding_router
from app.api.controllers.product_governance import router as product_governance_router
from app.api.controllers.public_platform import router as public_platform_router
from app.api.controllers.recommendation_plans import (
    admin_router as recommendation_plan_admin_router,
)
from app.api.controllers.recommendation_plans import (
    router as recommendation_plan_router,
)
from app.api.middleware import attach_trace_id
from app.core.config import get_settings
from app.core.errors import AgentError


def create_app() -> FastAPI:
    settings = get_settings()
    application = FastAPI(title=settings.app_name, version="0.1.0")
    application.middleware("http")(attach_trace_id)
    @application.exception_handler(AgentError)
    async def agent_error_handler(request: Request, exc: AgentError) -> JSONResponse:
        context = getattr(request.state, "request_context", None)
        trace_id = (
            getattr(context, "trace_id", None)
            or getattr(request.state, "trace_id", None)
            or request.headers.get("X-Trace-ID")
            or ""
        )
        headers: dict[str, str] = {}
        retry_after = getattr(exc, "retry_after_seconds", None)
        if isinstance(retry_after, int):
            headers["Retry-After"] = str(retry_after)
        return JSONResponse(status_code=exc.status_code, content={
            "error": {"code": exc.code, "message": exc.message,
                      "retryable": exc.is_retryable, "field_errors": []},
            "meta": {"trace_id": trace_id},
        }, headers=headers or None)
    application.include_router(agent_runs_router)
    application.include_router(conversations_router)
    application.include_router(public_platform_router)
    application.include_router(onboarding_router)
    application.include_router(offsite_fund_router)
    application.include_router(offsite_operation_router)
    application.include_router(knowledge_router)
    application.include_router(health_router)
    application.include_router(investment_goals_router)
    application.include_router(admin_router)
    application.include_router(product_governance_router)
    application.include_router(recommendation_plan_router)
    application.include_router(recommendation_plan_admin_router)
    application.include_router(market_data_router)
    if settings.app_env.casefold() == "local":
        application.include_router(local_test_router)
    return application


app = create_app()