rag-gateway
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
zhaoqingang
2025-02-11 0078ac6ffa486c481c26b7c0b033ff9664096a7f 
 app/service/v2/initialize_data.py


@@ -1,8 +1,13 @@


import json


import time


import os




import yaml


from passlib.context import CryptContext




from Log import logger


from app.config.const import DIFY


# from app.api import pwd_context


from app.config.const import DIFY, ENV_CONF_PATH, RAGFLOW


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


    cipher_suite


from app.service.auth import UserAppDao


@@ -14,10 +19,12 @@


from app.config.config import settings


from app.utils.password_handle import generate_password




pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")






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,29 +47,35 @@






async def create_menu_sync(db):


    json_file_path = "env_conf/menu_conf.json"


    # 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:


            menu['dialog'].clear()


        # for menu in json_data:


        #     menu['dialog'].clear()


    dialogs = db.query(DialogModel).all()




    dialog_dict = {}


    for dialog in dialogs:


        if dialog.name not in dialog_dict:


            dialog_dict[dialog.name] = []


        dialog_dict[dialog.name].append(dialog)




    for menu in json_data:


        if menu['title'] in dialog_dict:


            for dialog in dialog_dict[menu['title']]:


                new_dialog_item = {


        dialog_dict[dialog.name].append({


                    'id': dialog.id,


                    'chat_id': dialog.id,


                    'chat_type': '',


                    'agentType': dialog.dialog_type


                }


                menu['dialog'].append(new_dialog_item)


                })




    for menu in json_data:


        # if menu['title'] in dialog_dict:


        #     for dialog in dialog_dict[menu['title']]:


        #         new_dialog_item = {


        #             'id': dialog.id,


        #             'chat_id': dialog.id,


        #             'chat_type': '',


        #             'agentType': dialog.dialog_type


        #         }


        menu['dialog']= dialog_dict.get(menu['title'], [])


    json_data = {"data": json_data}


    new_file_name = f"menu_conf.json.template"


    new_file_path = os.path.join(os.path.dirname(json_file_path), new_file_name)


@@ -102,7 +115,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:


@@ -114,7 +127,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", [])


@@ -132,8 +145,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)


@@ -214,3 +226,37 @@


    if not user_id:


        raise Exception("Failed to register with app")


    print({"msg": "User registered successfully", "userFlag": user_id})






async def admin_user_sync(db):


    try:


        config = {}


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


            # 加载JSON数据


            config = yaml.safe_load(file)


        # print(config)


        db_user = db.query(UserModel).filter(UserModel.username == config["smart_server"]["account"]).first()


        if db_user:


            print("admin_user_sync: 用户已经存在！")


            return


        register_dict = {}




        for app in [RAGFLOW, DIFY]:


            register_dict[app] = {"id": config[app].get("id", "123"), "name": config[app]["account"],


                                        "pwd":config[app]["password"],


                                        "email": config[app]["account"]}




        # 存储用户信息


        hashed_password = pwd_context.hash(config["smart_server"]["password"])


        user_model = UserModel(username=config["smart_server"]["account"], hashed_password=hashed_password, email="",


                               phone="", login_name="", sync_flag="", creator=0, permission="admin")


        db.add(user_model)


        db.commit()


        db.refresh(user_model)


        u_id = user_model.id


        user_app_dao = UserAppDao(db)


        for k, v in register_dict.items():


            await user_app_dao.update_and_insert_data(v.get("name"), user_model.encrypted_password(v.get("pwd")), v.get("email"), u_id, str(v.get("id")), k)


    except Exception as e:


        print(e)


        db.rollback()