From 21943ce0e201492e12444893d2a7047bf9529416 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Fri, 28 Feb 2025 17:52:38 +0800
Subject: [PATCH] Refine error message while embedding model error, (#5490)

### What problem does this PR solve?

### Type of change

- [x] Refactoring
---
 rag/svr/task_executor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py
index 0d9eac99b..3fc69f3f5 100644
--- a/rag/svr/task_executor.py
+++ b/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