From afc68af3a67b2f936429feb3a68c1f09e03ce930 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 11 四月 2025 10:07:24 +0800
Subject: [PATCH] 智能体中心不显示自研智能体

---
 app/service/v2/initialize_data.py |    5 +++--
 1 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/app/service/v2/initialize_data.py b/app/service/v2/initialize_data.py
index 28a9961..270978c 100644
--- a/app/service/v2/initialize_data.py
+++ b/app/service/v2/initialize_data.py
@@ -181,7 +181,7 @@
         else:
             try:
                 dialog = ComplexChatModel(id=agent["id"], name=agent["name"], description=agent["description"],
-                                     icon=agent["icon"], tenant_id=user.id if user else "", dialog_type=agent["dialogType"], mode=agent["mode"],chat_mode = agent["chat_mode"])
+                                     icon=agent["icon"], tenant_id=user.id if user else "", dialog_type=agent["dialogType"], mode=agent["mode"],chat_mode = agent["chat_mode"],chat_model = agent.get("chat_model"),chat_model_ds = agent.get("chat_model_ds"),chat_provider = agent.get("chat_provider"))
                 db.add(dialog)
                 db.commit()
                 db.refresh(dialog)
@@ -434,11 +434,12 @@
 async def system_license_sync(db):
     with open(os.path.join(ENV_CONF_PATH, "system.yaml") , 'r', encoding='utf-8') as file:
         # 鍔犺浇JSON鏁版嵁
-        config = json.load(file)
+        config = yaml.safe_load(file)
         try:
             system = db.query(SystemDataModel).filter_by(id=SYSTEM_ID).first()
             if system:
                 system.version = config["smart_system"].get("version")
+                system.machine_id = get_machine_id()
             else:
                 system = SystemDataModel(id=SYSTEM_ID, version=config["smart_system"].get("version"), title=config["smart_system"].get("title"), desc=config["smart_system"].get("desc"), machine_id=get_machine_id())
                 db.add(system)

--
Gitblit v1.8.0