From 9c275b214f9619a64cd2998596ce696610185eb4 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期二, 19 十一月 2024 10:05:56 +0800 Subject: [PATCH] 增加删除 --- main.py | 14 ++++++++++++++ 1 files changed, 14 insertions(+), 0 deletions(-) diff --git a/main.py b/main.py index 90683f5..b86b6f7 100644 --- a/main.py +++ b/main.py @@ -1,19 +1,24 @@ from contextlib import asynccontextmanager +from apscheduler.schedulers.background import BackgroundScheduler from fastapi import FastAPI from app.api.auth import router as auth_router +from app.api.canvas import canvas_router from app.api.chat import router as chat_router from app.api.agent import router as agent_router +from app.api.dialog import dialog_router from app.api.excel import router as excel_router from app.api.files import router as files_router from app.api.knowledge import knowledge_router from app.api.organization import dept_router from app.api.report import router as report_router from app.api.resource import menu_router +from app.api.sync_data import sync_router from app.api.user import user_router from app.api.group import group_router from app.api.role import role_router from app.models.base_model import init_db from app.task.fetch_agent import sync_agents, initialize_agents +from app.task.sync_resources import sync_resource # 浣跨敤 Lifespan 浜嬩欢澶勭悊绋嬪簭 @@ -36,6 +41,12 @@ ) +# 鍒涘缓璋冨害鍣� +scheduler = BackgroundScheduler() +scheduler.add_job(sync_resource, 'interval', minutes=1, id="sync_resource_data") +scheduler.start() + + app.include_router(auth_router, prefix='/api/auth', tags=["auth"]) app.include_router(chat_router, prefix='/api/chat', tags=["chat"]) app.include_router(agent_router, prefix='/api/agent', tags=["agent"]) @@ -48,6 +59,9 @@ app.include_router(dept_router, prefix='/api/dept', tags=["dept"]) app.include_router(menu_router, prefix='/api/menu', tags=["menu"]) app.include_router(knowledge_router, prefix='/api/knowledge', tags=["knowledge"]) +app.include_router(dialog_router, prefix='/api/dialog', tags=["dialog"]) +app.include_router(canvas_router, prefix='/api/canvas', tags=["canvas"]) +app.include_router(sync_router, prefix='/api/sync', tags=["sync"]) if __name__ == "__main__": import uvicorn -- Gitblit v1.8.0