85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
|
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
|