支持静态文件的定期清理

Author: shell-nlp

gpt_server/model_worker/base/model_worker_base.py

@@ -1,7 +1,9 @@
 import asyncio
+from datetime import datetime
 from typing import List
 import json
 import sys
+import shutil
 from abc import ABC, abstractmethod
 from fastapi import BackgroundTasks, Request, FastAPI
 from fastapi.responses import JSONResponse, StreamingResponse
@@ -43,6 +45,22 @@ def get_context_length_(config):
     return 2048
 
 
+async def cleanup_static_files():
+    """清理静态文件目录并重建"""
+    logger.info(f"{datetime.now()}  开始清理静态文件目录：{STATIC_DIR}")
+    shutil.rmtree(STATIC_DIR, ignore_errors=True)
+    os.makedirs(STATIC_DIR, exist_ok=True)
+    logger.info(f"{datetime.now()}  清理完成")
+    await asyncio.sleep(1)  # 60分钟 = 3600秒
+
+
+async def run_scheduler():
+    """每60分钟执行一定时任务"""
+    while True:
+        await cleanup_static_files()
+        await asyncio.sleep(60 * 60)  # 60分钟 = 3600秒
+
+
 class ModelWorkerBase(BaseModelWorker, ABC):
     def __init__(
         self,
@@ -272,7 +290,7 @@ def run(cls):
         @app.on_event("startup")
         async def startup():
             global worker
-
+            asyncio.create_task(run_scheduler())
             worker = cls.get_worker(
                 worker_addr=worker_addr,
                 model_path=args.model_name_or_path,

gpt_server/serving/openai_api_server.py

@@ -811,7 +811,7 @@ async def speech(request: OpenAISpeechRequest):
         )
 
 
-@app.post("/v1/audio/speech2", dependencies=[Depends(check_api_key)])
+@app.post("/v1/audio/speech2", dependencies=[Depends(check_api_key)], deprecated=True)
 async def speech(request: SpeechRequest):
     os.makedirs(OUTPUT_DIR, exist_ok=True)  # 即使存在也不会报错
     list_voices = await edge_tts.list_voices()

gpt_server/utils.py

@@ -16,7 +16,6 @@
 logger.add("logs/gpt_server.log", rotation="100 MB", level="INFO")
 root_dir = Path(__file__).parent
 STATIC_DIR = root_dir / "static"
-os.makedirs(STATIC_DIR, exist_ok=True)
 
 
 def kill_child_processes(parent_pid, including_parent=False):