Update embeddings when populating genres
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user