diff --git a/admin/server/auth.py b/admin/server/auth.py index 30d3bd4dd7..9fb4f70ca6 100644 --- a/admin/server/auth.py +++ b/admin/server/auth.py @@ -110,7 +110,8 @@ def add_tenant_for_admin(user_info: dict, role: str): "embd_id": settings.EMBEDDING_MDL, "asr_id": settings.ASR_MDL, "parser_ids": settings.PARSERS, - "img2txt_id": settings.IMAGE2TEXT_MDL + "img2txt_id": settings.IMAGE2TEXT_MDL, + "rerank_id": settings.RERANK_MDL, } usr_tenant = { "tenant_id": user_info["id"], diff --git a/api/db/init_data.py b/api/db/init_data.py index 49a094eb32..291f39fe5a 100644 --- a/api/db/init_data.py +++ b/api/db/init_data.py @@ -58,7 +58,8 @@ def init_superuser(nickname=DEFAULT_SUPERUSER_NICKNAME, email=DEFAULT_SUPERUSER_ "embd_id": settings.EMBEDDING_MDL, "asr_id": settings.ASR_MDL, "parser_ids": settings.PARSERS, - "img2txt_id": settings.IMAGE2TEXT_MDL + "img2txt_id": settings.IMAGE2TEXT_MDL, + "rerank_id": settings.RERANK_MDL, } usr_tenant = { "tenant_id": user_info["id"],