From 345f2822f5072c34d9f969ae077830af968043b9 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 27 二月 2025 11:44:31 +0800
Subject: [PATCH] 优化输出

---
 app/init_config/init_run_data.py |   26 +++++++++++++++++---------
 1 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/app/init_config/init_run_data.py b/app/init_config/init_run_data.py
index 8bcff03..db095fb 100644
--- a/app/init_config/init_run_data.py
+++ b/app/init_config/init_run_data.py
@@ -1,17 +1,25 @@
 from app.models.base_model import SessionLocal
-from app.service.v2.initialize_data import dialog_menu_sync, default_group_sync, default_role_sync, app_register_sync, \
-    basic_agent_sync, admin_account_sync
-from app.task.fetch_agent import sync_resources_from_json
+from app.service.v2.initialize_data import dialog_menu_sync, default_group_sync, default_role_sync, \
+    basic_agent_sync, admin_account_sync, sync_rg_api_token
+from app.task.sync_account_token import sync_token
 
 
 async def sync_default_data():
     db = SessionLocal()
-    await dialog_menu_sync(db)  # 灏忔暟
-    await default_group_sync(db) # 榛樿缁�
-    await default_role_sync(db) # 榛樿瑙掕壊
-    await app_register_sync(db) # 娉ㄥ唽鐨勫簲鐢�
-    await basic_agent_sync(db) # 寮�鍙戠殑agent
-    await admin_account_sync(db) # 瓒呯璐﹀彿
+    try:
+        await dialog_menu_sync(db)  # 灏忔暟
+        await default_group_sync(db)  # 榛樿缁�
+        await default_role_sync(db)  # 榛樿瑙掕壊
+        # await app_register_sync(db)  # 娉ㄥ唽鐨勫簲鐢�
+        await basic_agent_sync(db)  # 寮�鍙戠殑agent
+        await admin_account_sync(db)  #
+        await sync_rg_api_token(db)  #
+        await sync_token()  # 鍚姩鍚屾token浠诲姟
+
+    except Exception as e:
+        print(e)
+    finally:
+        db.close()
     # await default_role_sync(db)  # 椤甸潰璧勬簮閰嶇疆淇℃伅
     # await default_role_sync(db)  # 榛樿鐨勮鑹茶祫婧�
 

--
Gitblit v1.8.0