Class: Exa::Services::AnswerStream

Inherits:
Object
  • Object
show all
Defined in:
lib/exa/services/answer_stream.rb

Instance Method Summary collapse

Constructor Details

#initialize(connection, **params) ⇒ AnswerStream

Returns a new instance of AnswerStream.



6
7
8
9
 
# File 'lib/exa/services/answer_stream.rb', line 6

def initialize(connection, **params)
  @connection = connection
  @params = params.merge(stream: true)
end

Instance Method Details

#call(&block) ⇒ Object

Raises:

  • (ArgumentError) 


11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/exa/services/answer_stream.rb', line 11

def call(&block)
  raise ArgumentError, "block required for streaming" unless block_given?

  # Use instance variable to track buffer across on_data callbacks
  @buffer = ""

  # Configure the request to stream chunks via on_data callback
  @connection.post("/answer", @params) do |req|
    req.options.on_data = proc do |chunk|
      # Add chunk to buffer and process complete SSE events
      @buffer += chunk
      process_sse_buffer(&block)
    end
  end

  # Process any remaining data in buffer after stream ends
  process_remaining_buffer(&block) if @buffer.length.positive?
end