From c43bf1dcf5495300d905c6d77a5a32085c2f64a9 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Wed, 19 Nov 2025 19:10:45 +0800 Subject: [PATCH] Fix: refine error msg. (#11380) ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- admin/server/auth.py | 4 ++-- api/apps/kb_app.py | 5 +++-- common/connection_utils.py | 20 ++++++++++++++++++++ conf/llm_factories.json | 2 +- 4 files changed, 26 insertions(+), 5 deletions(-) diff --git a/admin/server/auth.py b/admin/server/auth.py index 58a900afd..6c8bc2cb8 100644 --- a/admin/server/auth.py +++ b/admin/server/auth.py @@ -31,7 +31,7 @@ from common.constants import ActiveEnum, StatusEnum from api.utils.crypt import decrypt from common.misc_utils import get_uuid from common.time_utils import current_timestamp, datetime_format, get_format_time -from common.connection_utils import construct_response +from common.connection_utils import sync_construct_response from common import settings @@ -130,7 +130,7 @@ def login_admin(email: str, password: str): user.last_login_time = get_format_time() user.save() msg = "Welcome back!" - return construct_response(data=resp, auth=user.get_id(), message=msg) + return sync_construct_response(data=resp, auth=user.get_id(), message=msg) def check_admin(username: str, password: str): diff --git a/api/apps/kb_app.py b/api/apps/kb_app.py index 94cb8cf5b..4e8015d7f 100644 --- a/api/apps/kb_app.py +++ b/api/apps/kb_app.py @@ -886,6 +886,7 @@ async def check_embedding(): try: v, _ = emb_mdl.encode([title, txt_in]) + assert len(v[1]) == len(ck["vector"]), f"The dimension ({len(v[1])}) of given embedding model is different from the original ({len(ck['vector'])})" sim_content = _cos_sim(v[1], ck["vector"]) title_w = 0.1 qv_mix = title_w * v[0] + (1 - title_w) * v[1] @@ -895,8 +896,8 @@ async def check_embedding(): if sim_mix > sim: sim = sim_mix mode = "title+content" - except Exception: - return get_error_data_result(message="embedding failure") + except Exception as e: + return get_error_data_result(message=f"Embedding failure. {e}") eff_sims.append(sim) results.append({ diff --git a/common/connection_utils.py b/common/connection_utils.py index 0f765d57e..5b8154f0c 100644 --- a/common/connection_utils.py +++ b/common/connection_utils.py @@ -120,3 +120,23 @@ async def construct_response(code=RetCode.SUCCESS, message="success", data=None, response.headers["Access-Control-Allow-Headers"] = "*" response.headers["Access-Control-Expose-Headers"] = "Authorization" return response + + +def sync_construct_response(code=RetCode.SUCCESS, message="success", data=None, auth=None): + import flask + result_dict = {"code": code, "message": message, "data": data} + response_dict = {} + for key, value in result_dict.items(): + if value is None and key != "code": + continue + else: + response_dict[key] = value + response = flask.make_response(flask.jsonify(response_dict)) + if auth: + response.headers["Authorization"] = auth + response.headers["Access-Control-Allow-Origin"] = "*" + response.headers["Access-Control-Allow-Method"] = "*" + response.headers["Access-Control-Allow-Headers"] = "*" + response.headers["Access-Control-Allow-Headers"] = "*" + response.headers["Access-Control-Expose-Headers"] = "Authorization" + return response diff --git a/conf/llm_factories.json b/conf/llm_factories.json index 0eefa7a14..71f47be2e 100644 --- a/conf/llm_factories.json +++ b/conf/llm_factories.json @@ -4848,7 +4848,7 @@ ] }, { - "name": "JieKou.AI", + "name": "Jiekou.AI", "logo": "", "tags": "LLM,TEXT EMBEDDING,TEXT RE-RANK", "status": "1",