68 lines
1.7 KiB
Python
68 lines
1.7 KiB
Python
|
from fastapi import FastAPI
|
|||
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|||
|
|
from app.api.routes import api_router
|
|||
|
|
from app.core.config import settings
|
|||
|
|
from app.utils.logger import logger
|
|||
|
|
from app.services import milvus_service
|
|||
|
|
|
|||
|
|
# 创建应用
|
|||
|
|
app = FastAPI(
|
|||
|
|
title="车辆法规智能检索系统",
|
|||
|
|
description="基于RAG技术的法规检索与合规分析后端API",
|
|||
|
|
version="1.0.0",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# CORS配置
|
|||
|
|
app.add_middleware(
|
|||
|
|
CORSMiddleware,
|
|||
|
|
allow_origins=["*"],
|
|||
|
|
allow_credentials=True,
|
|||
|
|
allow_methods=["*"],
|
|||
|
|
allow_headers=["*"],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 注册路由
|
|||
|
|
app.include_router(api_router, prefix="/api")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.on_event("startup")
|
|||
|
|
async def startup_event():
|
|||
|
|
"""启动时初始化"""
|
|||
|
|
logger.info("Starting application...")
|
|||
|
|
|
|||
|
|
# 初始化Milvus集合（仅在服务可用时）
|
|||
|
|
try:
|
|||
|
|
if milvus_service is not None:
|
|||
|
|
milvus_service.create_regulations_collection()
|
|||
|
|
logger.info("Milvus collection initialized")
|
|||
|
|
else:
|
|||
|
|
logger.warning("Milvus service not available, using mock data")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"Milvus initialization failed: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.on_event("shutdown")
|
|||
|
|
async def shutdown_event():
|
|||
|
|
"""关闭时清理"""
|
|||
|
|
logger.info("Shutting down application...")
|
|||
|
|
try:
|
|||
|
|
if milvus_service is not None:
|
|||
|
|
milvus_service.disconnect()
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"Shutdown cleanup error: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.get("/")
|
|||
|
|
async def root():
|
|||
|
|
"""根路径"""
|
|||
|
|
return {
|
|||
|
|
"message": "车辆法规智能检索系统 API",
|
|||
|
|
"version": "1.0.0",
|
|||
|
|
"docs": "/docs",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.get("/health")
|
|||
|
|
async def health():
|
|||
|
|
"""健康检查"""
|
|||
|
|
return {"status": "healthy"}
|