main.py 1.0 KB

123456789101112131415161718192021222324252627282930313233343536
  1. from fastapi import FastAPI
  2. from fastapi.middleware.cors import CORSMiddleware
  3. from app.config import settings
  4. from app.routers.plan import router as plan_router
  5. from app.routers.agent import router as agent_router
  6. app = FastAPI(title=settings.app_name)
  7. origins = [origin.strip() for origin in settings.cors_origins.split(",") if origin.strip()]
  8. if not origins:
  9. origins = ["*"]
  10. origin_regex = None
  11. app_env = (settings.app_env or "").strip().lower()
  12. if app_env not in {"prod", "production"}:
  13. # 非生产环境放宽到 localhost/127.0.0.1 任意端口,避免端口/主机名切换触发预检 400
  14. origin_regex = r"^https?://(localhost|127\.0\.0\.1)(:\d+)?$"
  15. app.add_middleware(
  16. CORSMiddleware,
  17. allow_origins=origins,
  18. allow_origin_regex=origin_regex,
  19. allow_credentials=True,
  20. allow_methods=["*"],
  21. allow_headers=["*"],
  22. )
  23. app.include_router(plan_router)
  24. app.include_router(agent_router)
  25. @app.get("/health")
  26. def health_check():
  27. return {"status": "ok", "service": settings.app_name}