rag-gateway
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
zhaoqingang
2025-01-10 c878490140aa4e9d7b71c49214fb47f8618d403b 
 app/service/v2/initialize_data.py


@@ -2,7 +2,7 @@


import time


import os


from Log import logger


from app.config.const import DIFY


from app.config.const import DIFY, ENV_CONF_PATH


from app.models import MenuCapacityModel, WebMenuModel, GroupModel, RoleModel, DialogModel, UserModel, UserAppModel, \


    cipher_suite


from app.service.auth import UserAppDao


@@ -17,7 +17,7 @@




async def dialog_menu_sync(db):


    menu_list = []


    with open("env_conf/menu_conf.json", 'r', encoding='utf-8') as file:


    with open(os.path.join(ENV_CONF_PATH, "menu_conf.json") , 'r', encoding='utf-8') as file:


        # 加载JSON数据


        data = json.load(file)


        menu_list = data.get("data", [])


@@ -40,7 +40,8 @@






async def create_menu_sync(db):


    json_file_path = "env_conf/menu_conf.json.template"


    # json_file_path = "env_conf/menu_conf.json.template"


    json_file_path = os.path.join(ENV_CONF_PATH, "menu_conf.json.template")


    with open(json_file_path, 'r', encoding='utf-8') as file:


        json_data = json.load(file).get("data", [])


        # for menu in json_data:


@@ -107,7 +108,7 @@




async def app_register_sync(db):


    app_dict = {}


    with open("env_conf/app_register_conf.json", 'r', encoding='utf-8') as file:


    with open(os.path.join(ENV_CONF_PATH, "app_register_conf.json"), 'r', encoding='utf-8') as file:


        # 加载JSON数据


        app_dict = json.load(file)


        try:


@@ -119,7 +120,7 @@




async def basic_agent_sync(db):


    agent_list = []


    with open("env_conf/default_agent_conf.json", 'r', encoding='utf-8') as file:


    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", [])


@@ -137,8 +138,7 @@


        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"])


                                     icon=agent["icon"], tenant_id=user.id if user else "", dialog_type=agent["dialogType"], mode=agent["mode"])


                db.add(dialog)


                db.commit()


                db.refresh(dialog)