Browse Source

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 months ago
parent
commit
21943ce0e2
No account linked to committer's email address
1 changed files with 2 additions and 2 deletions
  1. 2
    2
      rag/svr/task_executor.py

+ 2
- 2
rag/svr/task_executor.py View File

@@ -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…
Cancel
Save