Class: RunApi::Flux2::Resources::TextToImage

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

Overview

Flux 2 text-to-image resource. Generate images from text or image prompts with Flux 2 models.

Constant Summary collapse

ENDPOINT = 
"/api/v1/flux_2/text_to_image"
RESPONSE_CLASS = 
Types::TextToImageResponse
COMPLETED_RESPONSE_CLASS = 
Types::CompletedTextToImageResponse

Instance Method Summary collapse

Constructor Details

#initialize(http) ⇒ TextToImage

Returns a new instance of TextToImage.



16
17
18
 
# File 'lib/runapi/flux_2/resources/text_to_image.rb', line 16

def initialize(http)
  @http = http
end

Instance Method Details

#create(**params) ⇒ RunApi::Flux2::Types::TextToImageResponse

Create a text-to-image task.

Parameters:

  • params (Hash) —

    text-to-image parameters

Returns:



33
34
35
36
37
 
# File 'lib/runapi/flux_2/resources/text_to_image.rb', line 33

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

#get(id) ⇒ RunApi::Flux2::Types::TextToImageResponse

Get text-to-image status by task ID.

Parameters:

  • id (String) —

    task ID

Returns:



43
44
45
 
# File 'lib/runapi/flux_2/resources/text_to_image.rb', line 43

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

#run(**params) ⇒ RunApi::Flux2::Types::CompletedTextToImageResponse

Generate an image and wait until complete.

Parameters:

  • params (Hash) —

    text-to-image parameters

Returns:



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

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