Sha256: ccade03ffc6ab3899f23626af886e77fea2e5c06f9676688994bf4cdd3faa6af

Contents?: true

Size: 885 Bytes

Versions: 1

Compression:

Stored size: 885 Bytes

Contents

module OpenAI
  class Images
    def initialize(access_token: nil, organization_id: nil)
      Ruby::OpenAI.configuration.access_token = access_token if access_token
      Ruby::OpenAI.configuration.organization_id = organization_id if organization_id
    end

    def generate(parameters: {})
      OpenAI::Client.post(path: "/images/generations", parameters: parameters)
    end

    def edit(parameters: {})
      OpenAI::Client.post(path: "/images/edits", parameters: open_files(parameters))
    end

    def variations(parameters: {})
      OpenAI::Client.post(path: "/images/variations", parameters: open_files(parameters))
    end

    private

    def open_files(parameters)
      parameters = parameters.merge(image: File.open(parameters[:image]))
      parameters = parameters.merge(mask: File.open(parameters[:mask])) if parameters[:mask]
      parameters
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
ruby-openai-3.0.2 lib/ruby/openai/images.rb