replace startup event by lifespan

app.py

@@ -12,6 +12,7 @@ import time
 import traceback
 import uuid
 from collections import deque
+from contextlib import asynccontextmanager
 from functools import partial
 from typing import Dict
 
@@ -246,7 +247,14 @@ class LLMAPIHandler:
  return JSONResponse({"valid": False, "message": str(e)})
 
 
-app = FastAPI()
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+ loop = asyncio.get_running_loop()
+ loop.create_task(clear_storage())
+ yield
+
+
+app = FastAPI(lifespan=lifespan)
 
 app.mount(
  "/storage",
@@ -303,13 +311,6 @@ async def clear_storage(ttl: float = 1800):
  await asyncio.sleep(60)
 
 
-@app.on_event("startup")
-async def startup_event():
- # 运行在服务启动后立即执行的任务
- loop = asyncio.get_running_loop()
- loop.create_task(clear_storage())
-
-
 def main():
  server_config = CONFIG.get("SERVER_UVICORN", {})
  uvicorn.run(app="__main__:app", **server_config)