158 lines
5.1 KiB
Python
158 lines
5.1 KiB
Python
"""FastAPI application factory."""
|
||
|
||
from loguru import logger
|
||
from contextlib import asynccontextmanager
|
||
from fastapi import FastAPI
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from fastapi.middleware.trustedhost import TrustedHostMiddleware
|
||
from fastapi.responses import JSONResponse
|
||
from fastapi.exceptions import RequestValidationError
|
||
from starlette.exceptions import HTTPException as StarletteHTTPException
|
||
|
||
from .config import Settings
|
||
from .middleware import UserContextMiddleware
|
||
from ..api.routes import router
|
||
from ..api.endpoints import table_metadata
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""Application lifespan manager."""
|
||
logger.info("Starting up TH Agenter application...")
|
||
yield
|
||
logger.info("Shutting down TH Agenter application...")
|
||
|
||
|
||
# def create_app(settings: Settings = None) -> FastAPI:
|
||
# """Create and configure FastAPI application."""
|
||
# if settings is None:
|
||
# from .config import get_settings
|
||
# settings = get_settings()
|
||
|
||
# # Create FastAPI app
|
||
# app = FastAPI(
|
||
# title=settings.app_name,
|
||
# version=settings.app_version,
|
||
# description="基于Vue的第一个聊天智能体应用，使用FastAPI后端，由DrGraph修改",
|
||
# debug=settings.debug,
|
||
# lifespan=lifespan,
|
||
# )
|
||
|
||
# # Add middleware
|
||
# setup_middleware(app, settings)
|
||
|
||
# # Add exception handlers
|
||
# setup_exception_handlers(app)
|
||
|
||
# # Include routers
|
||
# app.include_router(router, prefix="/api")
|
||
|
||
# app.include_router(table_metadata.router)
|
||
# # 在现有导入中添加
|
||
# from ..api.endpoints import database_config
|
||
|
||
# # 在路由注册部分添加
|
||
# app.include_router(database_config.router)
|
||
# # Health check endpoint
|
||
# @app.get("/health")
|
||
# async def health_check():
|
||
# return {"status": "healthy", "version": settings.app_version}
|
||
|
||
# # Root endpoint
|
||
# @app.get("/")
|
||
# async def root():
|
||
# return {"message": "Chat Agent API is running"}
|
||
# return app
|
||
|
||
|
||
def setup_middleware(app: FastAPI, settings: Settings) -> None:
|
||
"""Setup application middleware."""
|
||
|
||
# User context middleware (should be first to set context for all requests)
|
||
app.add_middleware(UserContextMiddleware)
|
||
|
||
# CORS middleware
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=settings.cors.allowed_origins,
|
||
allow_credentials=True,
|
||
allow_methods=settings.cors.allowed_methods,
|
||
allow_headers=settings.cors.allowed_headers,
|
||
)
|
||
|
||
# Trusted host middleware (for production)
|
||
if settings.environment == "production":
|
||
app.add_middleware(
|
||
TrustedHostMiddleware,
|
||
allowed_hosts=["*"] # Configure this properly in production
|
||
)
|
||
|
||
|
||
def setup_exception_handlers(app: FastAPI) -> None:
|
||
"""Setup global exception handlers."""
|
||
|
||
@app.exception_handler(StarletteHTTPException)
|
||
async def http_exception_handler(request, exc):
|
||
return JSONResponse(
|
||
status_code=exc.status_code,
|
||
content={
|
||
"error": {
|
||
"type": "http_error",
|
||
"message": exc.detail,
|
||
"status_code": exc.status_code
|
||
}
|
||
}
|
||
)
|
||
|
||
def make_json_serializable(obj):
|
||
"""递归地将对象转换为JSON可序列化的格式"""
|
||
if obj is None or isinstance(obj, (str, int, float, bool)):
|
||
return obj
|
||
elif isinstance(obj, bytes):
|
||
return obj.decode('utf-8')
|
||
elif isinstance(obj, (ValueError, Exception)):
|
||
return str(obj)
|
||
elif isinstance(obj, dict):
|
||
return {k: make_json_serializable(v) for k, v in obj.items()}
|
||
elif isinstance(obj, (list, tuple)):
|
||
return [make_json_serializable(item) for item in obj]
|
||
else:
|
||
# For any other object, convert to string
|
||
return str(obj)
|
||
|
||
@app.exception_handler(RequestValidationError)
|
||
async def validation_exception_handler(request, exc):
|
||
# Convert any non-serializable objects to strings in error details
|
||
try:
|
||
errors = make_json_serializable(exc.errors())
|
||
except Exception as e:
|
||
# Fallback: if even our conversion fails, use a simple error message
|
||
errors = [{"type": "serialization_error", "msg": f"Error processing validation details: {str(e)}"}]
|
||
|
||
return JSONResponse(
|
||
status_code=422,
|
||
content={
|
||
"error": {
|
||
"type": "validation_error",
|
||
"message": "Request validation failed",
|
||
"details": errors
|
||
}
|
||
}
|
||
)
|
||
|
||
@app.exception_handler(Exception)
|
||
async def general_exception_handler(request, exc):
|
||
logger.error(f"Unhandled exception: {exc}", exc_info=True)
|
||
return JSONResponse(
|
||
status_code=500,
|
||
content={
|
||
"error": {
|
||
"type": "internal_error",
|
||
"message": "Internal server error"
|
||
}
|
||
}
|
||
)
|
||
|
||
|
||
# Create the app instance
|
||
# app = create_app() |