Claudio Ortolina
28 lines
655 B
Elixir
28 lines
655 B
Elixir
defmodule OpenAI do
|
|
alias OpenAI.API
|
|
|
|
def gpt(completion) do
|
|
{:ok, collector} = Agent.start_link(fn -> "" end)
|
|
|
|
API.gpt_stream(completion, api_key(), fn data ->
|
|
case get_in(data, ["choices", Access.at(0), "delta", "content"]) do
|
|
nil -> :ok
|
|
data -> Agent.update(collector, fn current -> current <> data end)
|
|
end
|
|
end)
|
|
|
|
result = Agent.get(collector, & &1) |> JSON.decode!()
|
|
Agent.stop(collector)
|
|
{:ok, result}
|
|
end
|
|
|
|
def embeddings(text) do
|
|
API.get_embeddings(text, api_key())
|
|
end
|
|
|
|
defp api_key do
|
|
Application.get_env(:music_library, __MODULE__)
|
|
|> Keyword.fetch!(:api_key)
|
|
end
|
|
end
|