Add similar records

Initial stab by Claude
This commit is contained in:
Claudio Ortolina
2025-10-11 21:39:59 +02:00
parent 07218d16ca
commit 9cae972bc5
18 changed files with 1869 additions and 5 deletions
@@ -0,0 +1,39 @@
defmodule MusicLibrary.Worker.GenerateRecordEmbedding do
use Oban.Worker, queue: :heavy_writes, max_attempts: 3
require Logger
alias MusicLibrary.Records
alias MusicLibrary.Records.Similarity
@impl Oban.Worker
def perform(%Oban.Job{args: %{"record_id" => record_id}}) do
record = Records.get_record!(record_id)
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
end
end
defp generate_embedding(record) do
text = Similarity.text_representation(record)
case OpenAI.embeddings(text) do
{:ok, embedding} ->
{:ok, {embedding, text}}
{:error, reason} ->
{:error, reason}
end
end
defp store_embedding(record, {embedding, text_representation}) do
Similarity.store_embedding(record.id, embedding, text_representation)
end
end