defmodule EmbedbaseEx do
  @moduledoc """
      It's easy to get a prototype up-and-running. But, that's not where it stops. How do you persist data? Which LLM should you use? How can you keep up with the ever advancing pace of AI?

      Embedbase provides you with all the tools you need to create native production-ready applications powered by LLMs.

      > https://embedbase.xyz/
  """

  alias EmbedbaseEx.ExHttp

  @api_url "https://api.embedbase.xyz"
  @default_retries 5

  require Logger

  def embedbase_api() do
    System.fetch_env("EMBEDBASE_KEY")
  end

  # Inserting Data
  def insert_data(dataset_id, data, embedbase_api) when is_list(data) do
      url = "#{@api_url}/v1/#{dataset_id}"
      Enum.map(data, fn %{content: content, tag: tag} ->
          body = %{documents: [%{data: content}]}
          {:ok, %{id: id}} =ExHttp.http_post(url, body, embedbase_api, @default_retries)
          {tag, id}
      end)

  end

  def insert_data(dataset_id, data, embedbase_api) do
      url = "#{@api_url}/v1/#{dataset_id}"
      body = %{documents: [%{data: data}]}
      ExHttp.http_post(url, body, embedbase_api, @default_retries)
  end

  def insert_data(dataset_id, data, metadata, embedbase_api) do
      url = "#{@api_url}/v1/#{dataset_id}"
      body = %{documents: [%{data: data, metadata: metadata}]}
      ExHttp.http_post(url, body, embedbase_api, @default_retries)
  end

  # Updating Data
  def update_data(dataset_id, data, id, embedbase_api) do
        url = "#{@api_url}/v1/#{dataset_id}"
        body = %{documents: [%{data: data, id: id}]}
        ExHttp.http_post(url, body, embedbase_api, @default_retries)
  end

  # Using Bing Search
  def search_data(question, :bing, embedbase_api) do
      url = "#{@api_url}/v1/internet-search"
      body = %{query: question, engine: "bing"}
      ExHttp.http_post(url, body, embedbase_api, @default_retries)
  end


  def search_data(dataset_id, question, embedbase_api) do
      url = "#{@api_url}/v1/#{dataset_id}/search"
      body = %{query: question}
      ExHttp.http_post(url, body, embedbase_api, @default_retries)
  end

  # Delete data
  def delete_data(dataset_id, ids, embedbase_api) do
      url = "#{@api_url}/v1/#{dataset_id}"
      body = %{ids: ids}
      ExHttp.http_delete(url, body, embedbase_api, @default_retries)
  end

  # Delete dataset
  def delete_dataset(dataset_id, embedbase_api) do
     url = "#{@api_url}/v1/#{dataset_id}/clear"
     ExHttp.http_delete(url, embedbase_api, @default_retries)
  end
end