Преглед изворни кода

Refine error message while embedding model error, (#5490)

### What problem does this PR solve?

### Type of change

- [x] Refactoring
tags/v0.17.0
Kevin Hu пре 8 месеци
родитељ
комит
21943ce0e2
No account linked to committer's email address
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2
    2
      rag/svr/task_executor.py

+ 2
- 2
rag/svr/task_executor.py Прегледај датотеку

@@ -516,14 +516,14 @@ def do_handle_task(task):
try:
# bind embedding model
embedding_model = LLMBundle(task_tenant_id, LLMType.EMBEDDING, llm_name=task_embedding_id, lang=task_language)
vts, _ = embedding_model.encode(["ok"])
vector_size = len(vts[0])
except Exception as e:
error_message = f'Fail to bind embedding model: {str(e)}'
progress_callback(-1, msg=error_message)
logging.exception(error_message)
raise

vts, _ = embedding_model.encode(["ok"])
vector_size = len(vts[0])
init_kb(task, vector_size)

# Either using RAPTOR or Standard chunking methods

Loading…
Откажи
Сачувај