Method: OpenaiEmbeddingCreator#initialize

Defined in:
lib/embedding_engines/openai_embeddings.rb

#initialize(api_key = nil, chunker = BasicTextChunker.new, model = "text-embedding-ada-002") ⇒ OpenaiEmbeddingCreator

Returns a new instance of OpenaiEmbeddingCreator.



8
9
10
11
12
13
14
15
16
17
18
19
 
# File 'lib/embedding_engines/openai_embeddings.rb', line 8

def initialize(api_key = nil, chunker = BasicTextChunker.new, model = "text-embedding-ada-002")
  @chunker = chunker
  @model = model
  @api_key = api_key || ENV['OPENAI_API_KEY']
  raise 'API key not found. Please set the OPENAI_API_KEY environment variable.' if api_key.nil? || api_key.empty?

  if @api_key
    @llm = OpenAI::Client.new(access_token: @api_key)
  else
    Rails.logger.error "OpenAI API key not provided. Set the OPENAI_API_KEY in the ENV variables or pass it as an argument."
  end
end