From 03f8b01b3bb31fb46ae0233af8ff19da93cee1be Mon Sep 17 00:00:00 2001 From: KevinHuSh Date: Tue, 16 Apr 2024 19:12:12 +0800 Subject: [PATCH] fix bug for fasetembed (#392) ### What problem does this PR solve? Issue link:#325 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/document_app.py | 2 +- api/db/services/llm_service.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/api/apps/document_app.py b/api/apps/document_app.py index c01f73a36..8402d121f 100644 --- a/api/apps/document_app.py +++ b/api/apps/document_app.py @@ -58,7 +58,7 @@ def upload(): if not e: return get_data_error_result( retmsg="Can't find this knowledgebase!") - if DocumentService.get_doc_count(kb.tenant_id) >= os.environ.get('MAX_FILE_NUM_PER_USER', 8192): + if DocumentService.get_doc_count(kb.tenant_id) >= int(os.environ.get('MAX_FILE_NUM_PER_USER', 8192)): return get_data_error_result( retmsg="Exceed the maximum file number of a free user!") diff --git a/api/db/services/llm_service.py b/api/db/services/llm_service.py index a94701b0d..f565da132 100644 --- a/api/db/services/llm_service.py +++ b/api/db/services/llm_service.py @@ -83,7 +83,12 @@ class TenantLLMService(CommonService): llm = LLMService.query(llm_name=llm_name) if llm and llm[0].fid in ["QAnything", "FastEmbed"]: model_config = {"llm_factory": llm[0].fid, "api_key":"", "llm_name": llm_name, "api_base": ""} - if not model_config: raise LookupError("Model({}) not authorized".format(mdlnm)) + if not model_config: + if llm_name == "flag-embedding": + model_config = {"llm_factory": "Tongyi-Qianwen", "api_key": "", + "llm_name": llm_name, "api_base": ""} + else: + raise LookupError("Model({}) not authorized".format(mdlnm)) if llm_type == LLMType.EMBEDDING.value: if model_config["llm_factory"] not in EmbeddingModel: