This commit is contained in:
@@ -114,6 +114,34 @@ defmodule ElixirAi.AiProvider do
|
||||
end
|
||||
end
|
||||
|
||||
def find_by_id(id) do
|
||||
case Ecto.UUID.dump(id) do
|
||||
{:ok, binary_id} ->
|
||||
sql = """
|
||||
SELECT id, name, model_name, api_token, completions_url
|
||||
FROM ai_providers
|
||||
WHERE id = $(id)
|
||||
LIMIT 1
|
||||
"""
|
||||
|
||||
params = %{"id" => binary_id}
|
||||
|
||||
case DbHelpers.run_sql(sql, params, providers_topic(), AiProviderSchema.schema()) do
|
||||
{:error, _} ->
|
||||
{:error, :db_error}
|
||||
|
||||
[] ->
|
||||
{:error, :not_found}
|
||||
|
||||
[row | _] ->
|
||||
{:ok, row |> convert_uuid_to_string() |> then(&struct(AiProviderSchema, &1))}
|
||||
end
|
||||
|
||||
:error ->
|
||||
{:error, :invalid_uuid}
|
||||
end
|
||||
end
|
||||
|
||||
def delete(id) do
|
||||
sql = "DELETE FROM ai_providers WHERE id = $(id)::uuid"
|
||||
params = %{"id" => id}
|
||||
|
||||
@@ -173,6 +173,31 @@ defmodule ElixirAi.Conversation do
|
||||
end
|
||||
end
|
||||
|
||||
def update_provider(name, provider_id) when is_binary(provider_id) do
|
||||
case Ecto.UUID.dump(provider_id) do
|
||||
{:ok, binary_id} ->
|
||||
sql = """
|
||||
UPDATE conversations
|
||||
SET ai_provider_id = $(ai_provider_id), updated_at = $(updated_at)
|
||||
WHERE name = $(name)
|
||||
"""
|
||||
|
||||
params = %{
|
||||
"name" => name,
|
||||
"ai_provider_id" => binary_id,
|
||||
"updated_at" => now()
|
||||
}
|
||||
|
||||
case DbHelpers.run_sql(sql, params, "conversations") do
|
||||
{:error, :db_error} -> {:error, :db_error}
|
||||
_ -> :ok
|
||||
end
|
||||
|
||||
:error ->
|
||||
{:error, :invalid_uuid}
|
||||
end
|
||||
end
|
||||
|
||||
def find_id(name) do
|
||||
sql = "SELECT id FROM conversations WHERE name = $(name) LIMIT 1"
|
||||
params = %{"name" => name}
|
||||
|
||||
Reference in New Issue
Block a user