Merge branch 'feat/0.3.4'

dataelement · Aug 20, 2024 · 5644dbe · 5644dbe
2 parents f4c782b + 106a533
commit 5644dbe
Show file tree

Hide file tree

Showing 8 changed files with 37 additions and 25 deletions.
diff --git a/src/backend/bisheng/api/services/assistant.py b/src/backend/bisheng/api/services/assistant.py
@@ -150,7 +150,10 @@ async def create_assistant(cls, request: Request, login_user: UserPayload, assis
         # 自动补充默认的模型配置
         assistant_llm = LLMService.get_assistant_llm()
         if assistant_llm.llm_list:
-            assistant.model_name = assistant_llm.llm_list[0].model_id
+            for one in assistant_llm.llm_list:
+                if one.default:
+                    assistant.model_id = one.model_id
+                    break
 
         # 自动生成描述
         assistant, _, _ = await cls.get_auto_info(assistant)

diff --git a/src/backend/bisheng/api/services/finetune.py b/src/backend/bisheng/api/services/finetune.py
@@ -20,7 +20,7 @@
 from bisheng.cache import InMemoryCache
 from bisheng.database.models.finetune import (Finetune, FinetuneChangeModelName, FinetuneDao,
                                               FinetuneExtraParams, FinetuneList, FinetuneStatus)
-from bisheng.database.models.model_deploy import ModelDeploy, ModelDeployDao
+from bisheng.database.models.model_deploy import ModelDeploy, ModelDeployDao, ModelDeployInfo
 from bisheng.database.models.server import Server, ServerDao
 from bisheng.database.models.sft_model import SftModelDao
 from bisheng.utils.logger import logger
@@ -538,7 +538,14 @@ def get_model_list(cls, login_user: UserPayload, server_id: int) -> List[ModelDe
             ret.append(ModelDeployDao.insert_one(ModelDeploy(server=str(server_id),
                                                              model=one,
                                                              endpoint=f'http://{server_info.endpoint}/v2.1/models')))
-        return ret
+
+        # 过滤可用来进行微调的模型列表
+        all_sft_model = SftModelDao.get_all_sft_model()
+        sft_model_dict = {one.model_name: True for one in all_sft_model}
+        res = []
+        for one in ret:
+            res.append(ModelDeployInfo(**one.dict(), sft_support=sft_model_dict.get(one.model, False)))
+        return res
 
     @classmethod
     def get_gpu_info(cls) -> UnifiedResponseModel:

diff --git a/src/backend/bisheng/api/v1/user.py b/src/backend/bisheng/api/v1/user.py
@@ -240,6 +240,10 @@ async def list_user(*,
             groups = list(set(groups) & set(group_id))
             if not groups:
                 raise HTTPException(status_code=500, detail='无查看权限')
+        # 查询用户组下的角色, 和角色筛选条件做交集，得到真正去查询的角色ID
+        group_roles = RoleDao.get_role_by_groups(groups, None, 0, 0)
+        if role_id:
+            roles = list(set(role_id) & set([one.id for one in group_roles]))
     # 通过用户组和角色过滤出来的用户id
     user_ids = []
     if groups:
@@ -248,10 +252,7 @@ async def list_user(*,
         if not groups_user_ids:
             return resp_200({'data': [], 'total': 0})
         user_ids = list(set([one.user_id for one in groups_user_ids]))
-        # 查询用户组下的角色, 和角色筛选条件做交集，得到真正去查询的角色ID
-        group_roles = RoleDao.get_role_by_groups(groups, None, 0, 0)
-        if role_id:
-            roles = list(set(role_id) & set([one.id for one in group_roles]))
+
     if roles:
         roles_user_ids = UserRoleDao.get_roles_user(roles)
         if not roles_user_ids:

diff --git a/src/backend/bisheng/script/convert_sys_embeddings.py b/src/backend/bisheng/script/convert_sys_embeddings.py
@@ -106,19 +106,6 @@ def convert_sys_embeddings_to_mysql():
             if one.name == name:
                 need_add_server[name] = one
 
-    if not need_add_server[LLMServerType.BISHENG_RT.value]:
-        # 添加一个默认的RT服务提供方
-        server = LLMServer(
-            name=f"RT_OLD",
-            description='系统升级自动添加',
-            type=LLMServerType.BISHENG_RT.value,
-            config={
-                "host_base_url": 'http://xxxx:8000',
-            },
-            user_id=1,
-        )
-        llm_server = LLMDao.insert_server_with_models(server, [])
-
     # 重新设置知识库的模型配置
     update_knowledge = []
     for one in all_knowledge:
@@ -130,6 +117,19 @@ def convert_sys_embeddings_to_mysql():
     if update_knowledge:
         KnowledgeDao.update_knowledge_list(update_knowledge)
 
+    if not need_add_server_index.get(LLMServerType.BISHENG_RT.value):
+        # 添加一个默认的RT服务提供方
+        server = LLMServer(
+            name=f"RT_OLD",
+            description='系统升级自动添加，后续不建议使用',
+            type=LLMServerType.BISHENG_RT.value,
+            config={
+                "host_base_url": 'http://xxxx:8000',
+            },
+            user_id=1,
+        )
+        llm_server = LLMDao.insert_server_with_models(server, [])
+
 
 if __name__ == '__main__':
     convert_sys_embeddings_to_mysql()
diff --git a/src/frontend/src/pages/SkillPage/components/ToolItem.tsx b/src/frontend/src/pages/SkillPage/components/ToolItem.tsx
@@ -33,12 +33,12 @@ export default function ToolItem({
                                 onClick={(e) => onEdit(data.id)}
                             ><SettingIcon /></div>
                         }
-                        {/* {
+                        {
                             onSetClick && <div
                                 className="group-hover:opacity-100 opacity-0 hover:bg-[#EAEDF3] rounded cursor-pointer"
                                 onClick={onSetClick}
                             ><SettingIcon /></div>
-                        } */}
+                        }
                     </div>
                     <p className="text-sm text-muted-foreground mt-2">{data.description}</p>
                 </div>

diff --git a/src/frontend/src/pages/SkillPage/components/editAssistant/ModelSelect.tsx b/src/frontend/src/pages/SkillPage/components/editAssistant/ModelSelect.tsx
@@ -33,12 +33,13 @@ export default function ModelSelect({ type = 'assistant', value, onChange }) {
         setModelValue(_value)
         setOptions(_options)
         setLoading(false)
+
         if (!_value.length) onChange(null)
     }
 
     useEffect(() => {
         loadModels()
-    }, [])
+    }, [value])
     if (loading) return null
     return <Cascader
         selectPlaceholder="选择一个模型"

diff --git a/src/frontend/src/pages/SkillPage/tabTools.tsx b/src/frontend/src/pages/SkillPage/tabTools.tsx
@@ -80,7 +80,7 @@ export default function tabTools({ select = null, onSelect }) {
             </div>
           </div>
         </div>
-        <div className="h-full w-full flex-1 overflow-auto bg-background-login p-5 pt-12 scrollbar-hide">
+        <div className="h-full w-full flex-1 overflow-auto bg-background-login p-5 pb-20 pt-12 scrollbar-hide">
           <Accordion type="single" collapsible className="w-full">
             {options.length ? (
               options.map((el) => (

diff --git a/src/frontend/vite.config.ts b/src/frontend/vite.config.ts
@@ -6,7 +6,7 @@ import { viteStaticCopy } from 'vite-plugin-static-copy';
 import svgr from "vite-plugin-svgr";
 
 // Use environment variable to determine the target.
-const target = process.env.VITE_PROXY_TARGET || "http://192.168.106.120:2002";
+const target = process.env.VITE_PROXY_TARGET || "https://bisheng.dataelem.com/";
 const apiRoutes = ["^/api/", "/health"];
 
 const proxyTargets = apiRoutes.reduce((proxyObj, route) => {