Class: Roast::Cogs::Agent::Providers::Pi::PiInvocation::Context

Inherits:
Object
  • Object
show all
Defined in:
lib/roast/cogs/agent/providers/pi/pi_invocation.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Context

Returns a new instance of Context.



21
22
23
 
# File 'lib/roast/cogs/agent/providers/pi/pi_invocation.rb', line 21

def initialize
  @tool_calls = {} #: Hash[String, Messages::ToolCallMessage]
end

Instance Method Details

#add_tool_call(tool_call_message) ⇒ Object

: (Messages::ToolCallMessage) -> void



31
32
33
34
 
# File 'lib/roast/cogs/agent/providers/pi/pi_invocation.rb', line 31

def add_tool_call(tool_call_message)
  id = tool_call_message.id
  @tool_calls[id] = tool_call_message if id
end

#tool_call(tool_call_id) ⇒ Object

: (String?) -> Messages::ToolCallMessage?



26
27
28
 
# File 'lib/roast/cogs/agent/providers/pi/pi_invocation.rb', line 26

def tool_call(tool_call_id)
  @tool_calls[tool_call_id] if tool_call_id
end