fix(api): fix foreign key constraint error deletion

This commit is contained in:
Vito Sansevero
2025-05-25 21:37:02 -07:00
parent e0ac927a0d
commit 27ca3b0555
+3
View File
@@ -738,6 +738,9 @@ class PromptManagerAPI:
deleted_count = 0
with self.db.model.get_connection() as conn:
for prompt_id in prompt_ids:
# First delete related images to avoid foreign key constraint
conn.execute("DELETE FROM generated_images WHERE prompt_id = ?", (prompt_id,))
# Then delete the prompt
cursor = conn.execute(
"DELETE FROM prompts WHERE id = ?", (prompt_id,)
)