From 7abeaa62dffb2d1619a085c4cd00380a749b9f93 Mon Sep 17 00:00:00 2001 From: ulleo Date: Mon, 17 Nov 2025 19:40:48 +0800 Subject: [PATCH] feat: import Sample SQL --- backend/apps/data_training/curd/data_training.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/apps/data_training/curd/data_training.py b/backend/apps/data_training/curd/data_training.py index a32f6700..12bf0704 100644 --- a/backend/apps/data_training/curd/data_training.py +++ b/backend/apps/data_training/curd/data_training.py @@ -425,10 +425,6 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo] inserted_ids.append(obj.id) success_count += 1 - # 批量处理embedding - if inserted_ids: - run_save_data_training_embeddings(inserted_ids) - except Exception as e: session.rollback() # 将所有的有效记录标记为失败 @@ -440,6 +436,10 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo] }) success_count = 0 + # 批量处理embedding + if success_count > 0 and inserted_ids: + run_save_data_training_embeddings(inserted_ids) + # 返回结果，包含去重统计信息 return { 'success_count': success_count,