module OpenAI

def configuration
  @configuration ||= OpenAI::Configuration.new
end

def configure
  yield(configuration)
end

def log_message(prefix, message, level = :warn)
  color = level == :error ? "\033[31m" : "\033[33m"
  logger = Logger.new($stdout)
  logger.formatter = proc do |_severity, _datetime, _progname, msg|
    "#{color}#{prefix} (spotted in ruby-openai #{VERSION}): #{msg}\n\033[0m"
  end
  logger.send(level, message)
end

def rough_token_count(content = "")
  raise ArgumentError, "rough_token_count requires a string" unless content.is_a? String
  return 0 if content.empty?
  count_by_chars = content.size / 4.0
  count_by_words = content.split.size * 4.0 / 3
  estimate = ((count_by_chars + count_by_words) / 2.0).round
  [1, estimate].max
end