Kevin Hu
commited on
Commit
·
ccbf14d
1
Parent(s):
c020852
Fix preview issue in file manager. (#3846)
Browse files### What problem does this PR solve?
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- api/apps/file_app.py +7 -2
- rag/llm/rerank_model.py +0 -1
api/apps/file_app.py
CHANGED
|
@@ -329,8 +329,13 @@ def get(file_id):
|
|
| 329 |
e, file = FileService.get_by_id(file_id)
|
| 330 |
if not e:
|
| 331 |
return get_data_error_result(message="Document not found!")
|
| 332 |
-
|
| 333 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 334 |
ext = re.search(r"\.([^.]+)$", file.name)
|
| 335 |
if ext:
|
| 336 |
if file.type == FileType.VISUAL.value:
|
|
|
|
| 329 |
e, file = FileService.get_by_id(file_id)
|
| 330 |
if not e:
|
| 331 |
return get_data_error_result(message="Document not found!")
|
| 332 |
+
|
| 333 |
+
blob = STORAGE_IMPL.get(file.parent_id, file.location)
|
| 334 |
+
if not blob:
|
| 335 |
+
b, n = File2DocumentService.get_storage_address(file_id=file_id)
|
| 336 |
+
blob = STORAGE_IMPL.get(b, n)
|
| 337 |
+
|
| 338 |
+
response = flask.make_response(blob)
|
| 339 |
ext = re.search(r"\.([^.]+)$", file.name)
|
| 340 |
if ext:
|
| 341 |
if file.type == FileType.VISUAL.value:
|
rag/llm/rerank_model.py
CHANGED
|
@@ -126,7 +126,6 @@ class YoudaoRerank(DefaultRerank):
|
|
| 126 |
with YoudaoRerank._model_lock:
|
| 127 |
if not YoudaoRerank._model:
|
| 128 |
try:
|
| 129 |
-
logging.info("LOADING BCE...")
|
| 130 |
YoudaoRerank._model = RerankerModel(model_name_or_path=os.path.join(
|
| 131 |
get_home_cache_dir(),
|
| 132 |
re.sub(r"^[a-zA-Z0-9]+/", "", model_name)))
|
|
|
|
| 126 |
with YoudaoRerank._model_lock:
|
| 127 |
if not YoudaoRerank._model:
|
| 128 |
try:
|
|
|
|
| 129 |
YoudaoRerank._model = RerankerModel(model_name_or_path=os.path.join(
|
| 130 |
get_home_cache_dir(),
|
| 131 |
re.sub(r"^[a-zA-Z0-9]+/", "", model_name)))
|