From 36c00e400801786ee0f06a50e6046f49777ed188 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 14 一月 2025 15:47:31 +0800
Subject: [PATCH] tmp
---
app/init_config/init_run_data.py | 1 +
app/config/env_conf/account.yaml | 7 +++++++
app/service/v2/initialize_data.py | 30 ++++++++++++++++++++++++++++++
3 files changed, 38 insertions(+), 0 deletions(-)
diff --git a/app/config/env_conf/account.yaml b/app/config/env_conf/account.yaml
new file mode 100644
index 0000000..b2a6095
--- /dev/null
+++ b/app/config/env_conf/account.yaml
@@ -0,0 +1,7 @@
+chat_server:
+ account: user@example.com
+ password: secret
+
+workflow_server:
+ account: basic@mail.com
+ password: Basic2024
\ No newline at end of file
diff --git a/app/init_config/init_run_data.py b/app/init_config/init_run_data.py
index 76ef2db..668a8bc 100644
--- a/app/init_config/init_run_data.py
+++ b/app/init_config/init_run_data.py
@@ -10,6 +10,7 @@
await default_role_sync(db) # 榛樿瑙掕壊
await app_register_sync(db) # 娉ㄥ唽鐨勫簲鐢�
await basic_agent_sync(db) # 寮�鍙戠殑agent
+ await admin_account_sync(db) # 瓒呯璐﹀彿
# await default_role_sync(db) # 椤甸潰璧勬簮閰嶇疆淇℃伅
# await default_role_sync(db) # 榛樿鐨勮鑹茶祫婧�
diff --git a/app/service/v2/initialize_data.py b/app/service/v2/initialize_data.py
index 5d2043a..6491b2a 100644
--- a/app/service/v2/initialize_data.py
+++ b/app/service/v2/initialize_data.py
@@ -220,3 +220,33 @@
if not user_id:
raise Exception("Failed to register with app")
print({"msg": "User registered successfully", "userFlag": user_id})
+
+
+async def admin_account_sync(db):
+ agent_list = []
+ with open(os.path.join(ENV_CONF_PATH, "default_agent_conf.json"), 'r', encoding='utf-8') as file:
+ # 鍔犺浇JSON鏁版嵁
+ agent_dict = json.load(file)
+ agent_list = agent_dict.get("basic", [])
+ user = db.query(UserModel).filter_by(permission="admin").first()
+ for agent in agent_list:
+ dialog = db.query(DialogModel).filter(DialogModel.id == agent["id"]).first()
+ if dialog:
+ try:
+ dialog.name = agent["name"]
+ dialog.description = agent["description"]
+ dialog.icon = agent["icon"]
+ db.commit()
+ except Exception as e:
+ logger.error(e)
+ else:
+ try:
+ dialog = DialogModel(id=agent["id"], name=agent["name"], description=agent["description"],
+ icon=agent["icon"], tenant_id=user.id if user else "", dialog_type="3",
+ agent_id=agent["id"])
+ db.add(dialog)
+ db.commit()
+ db.refresh(dialog)
+ except Exception as e:
+ print(e)
+ db.rollback()
\ No newline at end of file
--
Gitblit v1.8.0