Class: RunApi::Imagen4::Resources::TextToImage

Inherits:
Object
  • Object
show all
Includes:
Core::ResourceHelpers
Defined in:
lib/runapi/imagen_4/resources/text_to_image.rb

Constant Summary collapse

ENDPOINT =
"/api/v1/imagen_4/text_to_image"
RESPONSE_CLASS =
Types::TextToImageResponse
COMPLETED_RESPONSE_CLASS =
Types::CompletedTextToImageResponse
TEXT_ONLY_FIELDS =
%i[negative_prompt seed num_images].freeze
PRO_ONLY_FIELDS =
%i[image_input resolution output_format].freeze

Instance Method Summary collapse

Constructor Details

#initialize(http) ⇒ TextToImage

Returns a new instance of TextToImage.



15
16
17
# File 'lib/runapi/imagen_4/resources/text_to_image.rb', line 15

def initialize(http)
  @http = http
end

Instance Method Details

#create(**params) ⇒ Object



24
25
26
27
28
# File 'lib/runapi/imagen_4/resources/text_to_image.rb', line 24

def create(**params)
  params = compact_params(params)
  validate_params!(params)
  request(:post, ENDPOINT, body: params)
end

#get(id) ⇒ Object



30
31
32
# File 'lib/runapi/imagen_4/resources/text_to_image.rb', line 30

def get(id)
  request(:get, "#{ENDPOINT}/#{id}")
end

#run(**params) ⇒ Object



19
20
21
22
# File 'lib/runapi/imagen_4/resources/text_to_image.rb', line 19

def run(**params)
  task = create(**params)
  poll_until_complete { get(task.id) }
end