Skip to content

Commit 7abeaa6

Browse files
committed
feat: import Sample SQL
1 parent 5b436c7 commit 7abeaa6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

backend/apps/data_training/curd/data_training.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -425,10 +425,6 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo]
425425
inserted_ids.append(obj.id)
426426
success_count += 1
427427

428-
# 批量处理embedding
429-
if inserted_ids:
430-
run_save_data_training_embeddings(inserted_ids)
431-
432428
except Exception as e:
433429
session.rollback()
434430
# 将所有的有效记录标记为失败
@@ -440,6 +436,10 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo]
440436
})
441437
success_count = 0
442438

439+
# 批量处理embedding
440+
if success_count > 0 and inserted_ids:
441+
run_save_data_training_embeddings(inserted_ids)
442+
443443
# 返回结果,包含去重统计信息
444444
return {
445445
'success_count': success_count,

0 commit comments

Comments
 (0)