main.py 711 B

1234567891011121314151617181920212223242526272829303132
  1. from contextlib import asynccontextmanager
  2. from fastapi import FastAPI
  3. from fastapi.middleware.cors import CORSMiddleware
  4. from api.routes.diet import router as diet_router
  5. from api.routes.health import router as health_router
  6. from memory.store import init_db
  7. @asynccontextmanager
  8. async def lifespan(_app: FastAPI):
  9. init_db()
  10. yield
  11. app = FastAPI(
  12. title="HealthRecordAgent API",
  13. version="1.0.0",
  14. lifespan=lifespan,
  15. )
  16. app.include_router(health_router, prefix="/api")
  17. app.include_router(diet_router, prefix="/api")
  18. app.add_middleware(
  19. CORSMiddleware,
  20. allow_origins=["*"], # 开发阶段允许全部
  21. allow_credentials=True,
  22. allow_methods=["*"],
  23. allow_headers=["*"],
  24. )