Module: RubyLLM::Providers::OpenAI::Images

Included in:
RubyLLM::Providers::OpenAI
Defined in:
lib/ruby_llm/providers/openai/images.rb

Overview

Image generation methods for the OpenAI API integration

Class Method Summary collapse

Class Method Details

.images_url ⇒ Object 



10
11
12
 
# File 'lib/ruby_llm/providers/openai/images.rb', line 10

def images_url
  'images/generations'
end

.parse_image_response(response, model:) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/ruby_llm/providers/openai/images.rb', line 23

def parse_image_response(response, model:)
  data = response.body
  image_data = data['data'].first

  Image.new(
    url: image_data['url'],
    mime_type: 'image/png', # DALL-E typically returns PNGs
    revised_prompt: image_data['revised_prompt'],
    model_id: model,
    data: image_data['b64_json']
  )
end

.render_image_payload(prompt, model:, size:) ⇒ Object 



14
15
16
17
18
19
20
21
 
# File 'lib/ruby_llm/providers/openai/images.rb', line 14

def render_image_payload(prompt, model:, size:)
  {
    model: model,
    prompt: prompt,
    n: 1,
    size: size
  }
end