Class: OrchestraAI::Step::Summarize

Inherits:
Object
  • Object
show all
Defined in:
lib/orchestra_ai/step/summarize.rb

Instance Method Summary collapse

Constructor Details

#initialize(results:, block: nil) ⇒ Summarize

Returns a new instance of Summarize.



6
7
8
9
 
# File 'lib/orchestra_ai/step/summarize.rb', line 6

def initialize(results:, block: nil)
  @block = block
  @results = results
end

Instance Method Details

#message ⇒ Object 



26
27
28
 
# File 'lib/orchestra_ai/step/summarize.rb', line 26

def message
  @response.match(/<Message>(.+)<\/Message>/m)[1]
end

#perform ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/orchestra_ai/step/summarize.rb', line 11

def perform
  prompt = Prompt::Summarize.new(results: @results)
  @response = @block.call(prompt.to_s)

  if @response.match?(/<Error>/)
    raise Error, "Error: #{@response.match(/<Message>(.+)<\/Message>/m)[1]}"
  end

  unless @response.match?(/<Message>(.+)<\/Message>/m)
    raise NoMessageError, "Error: No message found in the response"
  end

  @response
end