first commit

services/compliance-backend/app/main.py

@@ -0,0 +1,84 @@
+import logging
+import time
+
+import structlog
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+from prometheus_fastapi_instrumentator import Instrumentator
+
+from .api import kb, compliance, regulation
+from .core.config import settings
+
+# 结构化日志配置
+structlog.configure(
+    wrapper_class=structlog.make_filtering_bound_logger(
+        getattr(logging, settings.log_level.upper(), logging.INFO)
+    )
+)
+logger = structlog.get_logger()
+
+app = FastAPI(
+    title="AI合规智能中枢 API",
+    description="面向车企与工厂的全链路合规智能平台",
+    version="0.1.0",
+    docs_url="/docs",
+    redoc_url="/redoc",
+)
+
+# CORS（开发环境）
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"] if settings.app_env == "development" else [],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Prometheus 指标
+Instrumentator().instrument(app).expose(app)
+
+# 注册路由
+app.include_router(kb.router)
+app.include_router(compliance.router)
+app.include_router(regulation.router)
+
+
+@app.middleware("http")
+async def log_requests(request: Request, call_next):
+    start = time.time()
+    response = await call_next(request)
+    duration_ms = int((time.time() - start) * 1000)
+    logger.info(
+        "request",
+        method=request.method,
+        path=request.url.path,
+        status=response.status_code,
+        duration_ms=duration_ms,
+    )
+    return response
+
+
+@app.get("/health")
+async def health():
+    """健康检查（含依赖服务检测）"""
+    import httpx
+    from .core.config import settings
+
+    checks = {"status": "ok", "services": {}}
+
+    # 检查嵌入服务
+    try:
+        async with httpx.AsyncClient(timeout=5) as client:
+            r = await client.get(f"{settings.embedding_service_url}/health")
+            checks["services"]["embedding"] = "ok" if r.status_code == 200 else "degraded"
+    except Exception:
+        checks["services"]["embedding"] = "unavailable"
+
+    # 检查 MCP Server
+    try:
+        async with httpx.AsyncClient(timeout=5) as client:
+            r = await client.get(f"{settings.mcp_server_url}/health")
+            checks["services"]["mcp"] = "ok" if r.status_code == 200 else "degraded"
+    except Exception:
+        checks["services"]["mcp"] = "unavailable"
+
+    return checks