Class: Ask::Agent::Test::Mode

Inherits:
Object
  • Object
show all
Defined in:
lib/ask/agent/test.rb

Overview

Holds stubs, tracks called tools and final response for a test session.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Mode

Returns a new instance of Mode.



10
11
12
13
14
 
# File 'lib/ask/agent/test.rb', line 10

def initialize
  @stubs = []
  @called_tools = []
  @final_response = nil
end

Instance Attribute Details

#called_tools ⇒ Object (readonly)

Returns the value of attribute called_tools.



8
9
10
 
# File 'lib/ask/agent/test.rb', line 8

def called_tools
  @called_tools
end

#final_response ⇒ Object (readonly)

Returns the value of attribute final_response.



8
9
10
 
# File 'lib/ask/agent/test.rb', line 8

def final_response
  @final_response
end

#stubs ⇒ Object (readonly)

Returns the value of attribute stubs.



8
9
10
 
# File 'lib/ask/agent/test.rb', line 8

def stubs
  @stubs
end

Instance Method Details

#record_final_response(response) ⇒ Object 



28
29
30
 
# File 'lib/ask/agent/test.rb', line 28

def record_final_response(response)
  @final_response = response
end

#record_tool_call(name) ⇒ Object 



24
25
26
 
# File 'lib/ask/agent/test.rb', line 24

def record_tool_call(name)
  @called_tools << name
end

#stub_text(content) ⇒ Object 



20
21
22
 
# File 'lib/ask/agent/test.rb', line 20

def stub_text(content)
  @stubs << { type: :text, content: content }
end

#stub_tool_call(name, arguments = {}) ⇒ Object 



16
17
18
 
# File 'lib/ask/agent/test.rb', line 16

def stub_tool_call(name, arguments = {})
  @stubs << { type: :tool_call, name: name.to_s, arguments: arguments }
end

#unused_stubs ⇒ Object 



32
33
34
 
# File 'lib/ask/agent/test.rb', line 32

def unused_stubs
  @stubs
end