Ruby Critic

Updated 2025-04-25 12:53:06 -0300

app/use_cases/open_ai / embeddings.rb

24 lines of codes

2 methods

6.0 complexity/method

4 churn

12.09 complexity

0 duplications

code
1 smell

# frozen_string_literal: true

class OpenAi::Embeddings                                        IrresponsibleModule              
      OpenAi::Embeddings has no descriptive comment          
  
  def get_embedding(ai_assistent, content, model = 'text-embedding-ada-002')
    fetch_embeddings(ai_assistent, content, model)
  end

  private

  def fetch_embeddings(ai_assistent, input, model)
    url = 'https://api.openai.com/v1/embeddings'
    headers = {
      'Authorization' => "Bearer #{ai_assistent.api_key}",
      'Content-Type' => 'application/json'
    }
    data = {
      input: input,
      model: model
    }

    response = Net::HTTP.post(URI(url), data.to_json, headers)
    JSON.parse(response.body)['data']&.pick('embedding')
  end
end