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