初始化

app/main.py

@@ -0,0 +1,68 @@
+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"}