fix:worker

Author: LinceMathew

.gitignore

@@ -0,0 +1 @@
+dprompts

README.md

@@ -1,8 +1,8 @@
-# dPrompt
+# dPrompts
 
 ## Overview
 
-dPrompt is a distributed job processing system using River and PostgreSQL, with support for LLM (Ollama) integration.
+dPrompts enables teams to perform distributed, bulk LLM operations locally using Ollama, which is cost-effective and works on most laptops with an integrated GPU.
 
 ## Installation
 

dprompts

@@ -54,8 +54,14 @@ func (w *DPromptsWorker) Work(ctx context.Context, job *river.Job[DPromptsJobArg
 		return err
 	}
 
-	_, err = w.db.Exec(ctx,
-		"INSERT INTO dprompts_results (job_id, response) VALUES ($1, $2)",
+	tx, err := w.db.Begin(ctx)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback(ctx)
+
+	_, err = tx.Exec(ctx,
+		"INSERT INTO dprompts_results (job_id, response) VALUES ($1, $2) ON CONFLICT (job_id) DO NOTHING",
 		job.ID,
 		jsonResponse,
 	)
@@ -64,6 +70,17 @@ func (w *DPromptsWorker) Work(ctx context.Context, job *river.Job[DPromptsJobArg
 		return err
 	}
 
+	_, err = river.JobCompleteTx[*riverpgxv5.Driver](ctx, tx, job)
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to complete job transactionally")
+		return err
+	}
+
+	if err := tx.Commit(ctx); err != nil {
+		log.Error().Err(err).Msg("Failed to commit transaction")
+		return err
+	}
+
 	log.Info().
 		Str("job_id", strconv.FormatInt(job.ID, 10)).
 		Msg("Job completed and saved")