Module: Jekyll::EmbeddingsGenerator::Embeddings

Extended by:

Jekyll::EmbeddingsGenerator

Defined in:

lib/jekyll/embeddings-generator/embeddings/generate.rb

Constant Summary

Constants included from Jekyll::EmbeddingsGenerator

VERSION

Class Method Summary

• .generate_embeddings(text) ⇒ Object

Methods included from Jekyll::EmbeddingsGenerator

run

Class Method Details

.generate_embeddings(text) ⇒ Object

# File 'lib/jekyll/embeddings-generator/embeddings/generate.rb', line 12

def generate_embeddings(text)
  config = Jekyll::EmbeddingsGenerator.config
  api_key = config["openai_api_key"]
  response = HTTParty.post(
    "https://api.openai.com/v1/embeddings",
    :headers => {
      "Authorization" => "Bearer #{api_key}",
      "Content-Type"  => "application/json",
    },
    :body    => {
      :model => "text-embedding-3-small",
      :input => text,
    }.to_json
  )

  raise "OpenAI API error: #{response.parsed_response["error"]["message"]}" unless response.success?

  response.parsed_response["data"][0]["embedding"]
end