Update embeddings when populating genres

This commit is contained in:
Claudio Ortolina
2025-10-12 22:45:12 +02:00
parent 0e80bf8ba1
commit 7a1061a783
5 changed files with 39 additions and 22 deletions
+1 -1
View File
@@ -121,7 +121,7 @@ defmodule MusicLibrary.Records.Similarity do
params
|> GenerateRecordEmbedding.new(meta: meta)
|> Oban.insert!()
|> Oban.insert()
end
def generate_all_embeddings_async do
@@ -1,8 +1,6 @@
defmodule MusicLibrary.Worker.GenerateRecordEmbedding do
use Oban.Worker, queue: :heavy_writes, max_attempts: 3
require Logger
alias MusicLibrary.Records
alias MusicLibrary.Records.Similarity
@@ -12,12 +10,7 @@ defmodule MusicLibrary.Worker.GenerateRecordEmbedding do
with {:ok, embedding} <- generate_embedding(record),
{:ok, _} <- store_embedding(record, embedding) do
Logger.info("Generated embedding for record #{record_id}")
:ok
else
{:error, reason} = error ->
Logger.error("Failed to generate embedding for record #{record_id}: #{inspect(reason)}")
error
Records.notify_update(record)
end
end
@@ -28,8 +21,8 @@ defmodule MusicLibrary.Worker.GenerateRecordEmbedding do
{:ok, embedding} ->
{:ok, {embedding, text}}
{:error, reason} ->
{:error, reason}
error ->
error
end
end
+7 -3
View File
@@ -1,12 +1,16 @@
defmodule MusicLibrary.Worker.PopulateGenres do
use Oban.Worker, queue: :heavy_writes, max_attempts: 10
alias MusicLibrary.Records
@impl Oban.Worker
def perform(%Oban.Job{args: %{"id" => record_id}}) do
record = MusicLibrary.Records.get_record!(record_id)
record = Records.get_record!(record_id)
with {:ok, updated_record} <- MusicLibrary.Records.populate_genres(record) do
MusicLibrary.Records.notify_update(updated_record)
with {:ok, updated_record} <- Records.populate_genres(record),
{:ok, _worker} <-
Records.Similarity.generate_embedding_async(updated_record) do
Records.notify_update(updated_record)
end
end
end