mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-08 01:06:59 +08:00
feat: import Sample SQL
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user