feat: add Nailong Fund advisor capabilities
This commit is contained in:
+34
-3
@@ -5,10 +5,22 @@ 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
|
||||
|
||||
@@ -16,22 +28,41 @@ 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.status_code >= 500, "field_errors": []},
|
||||
"meta": {"trace_id": context.trace_id if context else ""},
|
||||
})
|
||||
"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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user