chef_agent
This commit is contained in:
8
main.py
8
main.py
@@ -5,11 +5,14 @@ from contextlib import asynccontextmanager
|
||||
from fastapi import FastAPI
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from sqlmodel import SQLModel
|
||||
|
||||
from api.agent_api import agent_router
|
||||
from core.db import engine
|
||||
from core.config.settings import settings
|
||||
from api.scheduler_api import scheduler_router
|
||||
from api.lottery_api import lottery_router
|
||||
from core.task.dlt_scheduler import scheduler
|
||||
from lc.chef_agent import chef_agent
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(
|
||||
@@ -30,14 +33,19 @@ async def lifespan(app: FastAPI):
|
||||
# 启动定时调度器
|
||||
scheduler.start()
|
||||
|
||||
# 初始化chef_agent
|
||||
chef_agent.initialize()
|
||||
|
||||
yield # 此处交出控制权,服务开始运行
|
||||
# ========== 关闭时执行 ==========
|
||||
logger.info("服务关闭,释放资源")
|
||||
chef_agent.close()
|
||||
scheduler.shutdown()
|
||||
|
||||
app = FastAPI(lifespan=lifespan)
|
||||
app.include_router(scheduler_router)
|
||||
app.include_router(lottery_router)
|
||||
app.include_router(agent_router)
|
||||
|
||||
app.add_middleware(
|
||||
CORSMiddleware, # type: ignore[arg-type]
|
||||
|
||||
Reference in New Issue
Block a user