From 7bd653dccd1857773c5f9f696fd114b21edecb2a Mon Sep 17 00:00:00 2001 From: ulleo Date: Mon, 17 Nov 2025 20:01:20 +0800 Subject: [PATCH] feat: import Sample SQL --- backend/apps/data_training/curd/data_training.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/backend/apps/data_training/curd/data_training.py b/backend/apps/data_training/curd/data_training.py index 12bf0704..ede18941 100644 --- a/backend/apps/data_training/curd/data_training.py +++ b/backend/apps/data_training/curd/data_training.py @@ -416,14 +416,14 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo] try: # 批量插入 - session.bulk_save_objects(data_training_objects) + session.bulk_save_objects(data_training_objects, return_defaults=True) session.commit() - # 获取插入的ID（需要刷新对象） + # 获取插入的ID for obj in data_training_objects: - session.refresh(obj) - inserted_ids.append(obj.id) - success_count += 1 + if obj.id is not None: # 确保ID已经被赋值 + inserted_ids.append(obj.id) + success_count += 1 except Exception as e: session.rollback()