Packages

oapi_open_ai

0.2.2
1.0.2 1.0.1 1.0.0 0.4.4 0.4.3 0.4.2 0.4.1 0.4.0 0.3.3 0.3.2 0.3.1 0.3.0 0.2.5 0.2.4 0.2.3 0.2.2 0.2.1 0.2.0 0.1.6 0.1.5 0.1.4 0.1.3 0.1.2 0.1.1

OpenAI REST API Client

HexDocs

Current section

Files

Jump to
Readme
24 Versions
2 Dependencies
Dependants
Files
Activity
oapi_open_ai test mocks embedding.ex
Raw

test/mocks/embedding.ex 

defmodule OpenAi.Mocks.Embedding do

  def embedding(data) do

    %{

      embedding: [0.1, 0.2, 0.3],

      index: 0,

      object: "embedding"

    }

    |> Map.merge(data)

  end



  def create_response(data \\ [], count \\ 2) do

    data =

      Enum.map(1..count, fn index ->

        embedding(data |> Enum.at(index - 1, %{}))

      end)



    %HTTPoison.Response{

      status_code: 200,

      body:

        %{

          object: "list",

          data: data,

          model: "text-embedding-ada-002",

          usage: %{

            prompt_tokens: 123,

            total_tokens: 456

          }

        }

        |> embedding()

        |> Poison.encode!()

    }

  end

end