From 3f3469130bb1d47d68d11ee3858fab71b19440e8 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Wed, 4 Dec 2024 11:53:23 +0800 Subject: [PATCH] Fix preview issue in file manager. (#3846) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/file_app.py | 9 +++++++-- rag/llm/rerank_model.py | 1 - 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/api/apps/file_app.py b/api/apps/file_app.py index 630a12295..6542c54e9 100644 --- a/api/apps/file_app.py +++ b/api/apps/file_app.py @@ -329,8 +329,13 @@ def get(file_id): e, file = FileService.get_by_id(file_id) if not e: return get_data_error_result(message="Document not found!") - b, n = File2DocumentService.get_storage_address(file_id=file_id) - response = flask.make_response(STORAGE_IMPL.get(b, n)) + + blob = STORAGE_IMPL.get(file.parent_id, file.location) + if not blob: + b, n = File2DocumentService.get_storage_address(file_id=file_id) + blob = STORAGE_IMPL.get(b, n) + + response = flask.make_response(blob) ext = re.search(r"\.([^.]+)$", file.name) if ext: if file.type == FileType.VISUAL.value: diff --git a/rag/llm/rerank_model.py b/rag/llm/rerank_model.py index 61752aec0..2aa439269 100644 --- a/rag/llm/rerank_model.py +++ b/rag/llm/rerank_model.py @@ -126,7 +126,6 @@ class YoudaoRerank(DefaultRerank): with YoudaoRerank._model_lock: if not YoudaoRerank._model: try: - logging.info("LOADING BCE...") YoudaoRerank._model = RerankerModel(model_name_or_path=os.path.join( get_home_cache_dir(), re.sub(r"^[a-zA-Z0-9]+/", "", model_name)))