From 48e33c7fbd6969d05b2affda0321595d2d6b370a Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 10 二月 2025 14:51:06 +0800
Subject: [PATCH] 增加ping機制

---
 main.py |   13 +++++++++----
 1 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 16f7a3b..bf04686 100644
--- a/main.py
+++ b/main.py
@@ -15,7 +15,7 @@
 from app.api.label import label_router
 from app.api.llm import llm_router
 from app.api.organization import dept_router
-from app.api.v2.chat import chat1_router
+from app.api.v2.chat import chat_router_v2
 from app.api.v2.public_api import public_api
 from app.api.report import router as report_router
 from app.api.resource import menu_router
@@ -24,10 +24,12 @@
 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.delete_execl_file import delete_file_after_delay
 # from app.models.base_model import init_db
 from app.task.fetch_agent import sync_agents, initialize_agents, sync_agents_v2, sync_knowledge, \
     sync_resources_from_json
 from app.init_config.init_run_data import sync_default_data
+from app.task.sync_account_token import sync_token, start_sync_token_task
 
 init_db()
 
@@ -39,8 +41,9 @@
     # initialize_agents()
     # # 鍦ㄥ簲鐢ㄥ惎鍔ㄦ椂鍚屾浠g悊
     # sync_agents()
-    await sync_default_data()
 
+
+    await sync_default_data()
     sync_agents_v2()
     sync_knowledge()
     sync_resources_from_json()
@@ -59,7 +62,7 @@
 # 璁剧疆 CORS 涓棿浠�
 # app.add_middleware(
 #     CORSMiddleware,
-#     allow_origins=["http://192.168.20.119:9303", "http://192.168.20.119:9301", "http://smartai.com:8293","http://localhost:9303",  "http://127.0.0.1:9303","http://localhost:5173","http://192.168.20.158:5173"],
+#     allow_origins=["*"],
 #     allow_credentials=True,
 #     allow_methods=["*"],  # 鍏佽鎵�鏈夋柟娉�
 #     allow_headers=["*"],  # 鍏佽鎵�鏈夊ご閮�
@@ -67,6 +70,8 @@
 # 鍒涘缓璋冨害鍣�
 scheduler = BackgroundScheduler()
 scheduler.add_job(sync_agents_v2, 'interval', minutes=60, id="sync_resource_data")
+scheduler.add_job(start_sync_token_task, 'interval', minutes=5, id="sync_token_1")
+# scheduler.add_job(delete_file_after_delay, 'interval', minutes=10, id="delete_file_after_delay")
 scheduler.start()
 
 app.include_router(auth_router, prefix='/api/auth', tags=["auth"])
@@ -86,7 +91,7 @@
 app.include_router(canvas_router, prefix='/api/canvas', tags=["canvas"])
 app.include_router(label_router, prefix='/api/label', tags=["label"])
 app.include_router(public_api, prefix='/v1/api', tags=["public_api"])
-app.include_router(chat1_router, prefix='/v1/chat', tags=["chat1"])
+app.include_router(chat_router_v2, prefix='/v1/chat', tags=["chat1"])
 app.mount("/static", StaticFiles(directory="app/images"), name="static")
 
 if __name__ == "__main__":

--
Gitblit v1.8.0