1.添加定时任务(每天凌晨四点处理redis记录的删除,存储入mysql持久化操作)

2.修复了prompt中没有用户输入信息的bug
3.优化了更新类接口代码(删除了db.add()代码)
This commit is contained in:
killua committed 2024-05-07 11:26:59 +08:00
1 parent de370f77ac
commit 065528f509
7 files changed
+65 -16

No files matched your search

+18 -1
View File
@@ -1,12 +1,16 @@
from fastapi import FastAPI, Depends
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
from sqlalchemy import create_engine
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.cron import CronTrigger
from .models import Base
from .routes.character_route import router as character_router
from .routes.user_route import router as user_router
from .routes.session_route import router as session_router
from .routes.chat_route import router as chat_router
from .dependencies.logger import get_logger
from .background_tasks import updating_redis_cache
from config import get_config
@@ -28,8 +32,20 @@ logger.info("数据库初始化完成")
#------------------------------------------------------
#--------------------设置定时任务-----------------------
scheduler = AsyncIOScheduler()
scheduler.add_job(updating_redis_cache, CronTrigger.from_crontab("0 4 * * *"))
@asynccontextmanager
async def lifespan(app:FastAPI):
scheduler.start() #启动定时任务
yield
scheduler.shutdown() #关闭定时任务
logger.info("定时任务初始化完成")
#------------------------------------------------------
#--------------------创建FastAPI实例--------------------
app = FastAPI()
app = FastAPI(lifespan=lifespan)
logger.info("FastAPI实例创建完成")
#------------------------------------------------------
@@ -42,6 +58,7 @@ app.include_router(chat_router)
logger.info("路由初始化完成")
#-------------------------------------------------------
#-------------------设置跨域中间件-----------------------
app.add_middleware(
CORSMiddleware,