discourse-ai/lib/shared/inference/openai_completions.rb

# frozen_string_literal: true

module ::DiscourseAi
  module Inference
    class OpenAICompletions
      def self.perform!(model, content, api_key)
        headers = {
          "Authorization" => "Bearer #{SiteSetting.ai_openai_api_key}",
          "Content-Type" => "application/json",
        }

        model ||= "gpt-3.5-turbo"

        response =
          Faraday.post(
            "https://api.openai.com/v1/chat/completions",
            { model: model, messages: content }.to_json,
            headers,
          )

        raise Net::HTTPBadResponse unless response.status == 200

        JSON.parse(response.body, symbolize_names: true)
      end
    end
  end
end