知识库和智能体同步

zhaoqingang

2024-12-18 c941b948f1fa8eff615985a83110a401ae426ab3

 main.py

@@ -1,5 +1,5 @@
from contextlib import asynccontextmanager
# from apscheduler.schedulers.background import BackgroundScheduler
from apscheduler.schedulers.background import BackgroundScheduler
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from starlette.staticfiles import StaticFiles
@@ -22,7 +22,8 @@
from app.api.user import user_router
from app.api.group import group_router
from app.api.role import role_router
from app.task.fetch_agent import sync_agents, initialize_agents, sync_agents_v2
# from app.models.base_model import init_db
from app.task.fetch_agent import sync_agents, initialize_agents, sync_agents_v2, sync_knowledge
from app.init_config.init_run_data import sync_default_data


@@ -36,6 +37,7 @@
    await sync_default_data()

    sync_agents_v2()
    sync_knowledge()
    yield
    # 在应用关闭时执行清理操作（如果需要）
    pass
@@ -57,9 +59,9 @@
#     allow_headers=["*"],  # 允许所有头部
# )
# 创建调度器
# scheduler = BackgroundScheduler()
# scheduler.add_job(sync_resource, 'interval', minutes=1, id="sync_resource_data")
# scheduler.start()
scheduler = BackgroundScheduler()
scheduler.add_job(sync_agents_v2, 'interval', minutes=60, id="sync_resource_data")
scheduler.start()


app.include_router(auth_router, prefix='/api/auth', tags=["auth"])