Class: NitroIntelligence::Client::Handlers::Observed::ImageHandler

Inherits:
Object
  • Object
show all
Defined in:
lib/nitro_intelligence/client/handlers/observed/image_handler.rb

Instance Method Summary collapse

Constructor Details

#initialize(base_handler:, observer:) ⇒ ImageHandler

Returns a new instance of ImageHandler.



8
9
10
11
 
# File 'lib/nitro_intelligence/client/handlers/observed/image_handler.rb', line 8

def initialize(base_handler:, observer:)
  @base_handler = base_handler
  @observer = observer
end

Instance Method Details

#create(message: "", target_image: nil, reference_images: [], parameters: {}) ⇒ Object 



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/nitro_intelligence/client/handlers/observed/image_handler.rb', line 13

def create(message: "", target_image: nil, reference_images: [], parameters: {})
  image_generation = build_image_generation(message:, target_image:, reference_images:, parameters:)

  @base_handler.validate_and_resolve!(parameters, image_generation)

  # Modifies parameters in place
  prompt = handle_prompt(parameters:)
  trace_name = parameters[:trace_name] || prompt&.name || @observer.project_client.project.slug

  @observer.observe(
    "image-generation",
    type: :generation,
    parameters:,
    trace_name:,
    prompt:
  ) do |generation|
    workflow(generation:, image_generation:, parameters:)
  end

  image_generation
end