Class: AgentClientProtocol::Contrib::PermissionBroker

Inherits:
Object
  • Object
show all
Defined in:
lib/agent_client_protocol/contrib/permission_broker.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(requester:, tracker: nil) ⇒ PermissionBroker

Returns a new instance of PermissionBroker.



6
7
8
9
 
# File 'lib/agent_client_protocol/contrib/permission_broker.rb', line 6

def initialize(requester:, tracker: nil)
  @requester = requester
  @tracker = tracker
end

Class Method Details

.default_permission_options ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/agent_client_protocol/contrib/permission_broker.rb', line 30

def self.default_permission_options
  [
    Schema::PermissionOption.new(
      option_id: "approve",
      name: "Approve",
      kind: Schema::PermissionOptionKind::ALLOW_ONCE
    ),
    Schema::PermissionOption.new(
      option_id: "approve_session",
      name: "Approve for session",
      kind: Schema::PermissionOptionKind::ALLOW_ALWAYS
    ),
    Schema::PermissionOption.new(
      option_id: "reject",
      name: "Reject",
      kind: Schema::PermissionOptionKind::REJECT_ONCE
    )
  ]
end

Instance Method Details

#request_for(external_id, session_id:, description: nil, options: nil, content: nil, tool_call: nil) ⇒ Object 



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

def request_for(external_id, session_id:, description: nil, options: nil, content: nil, tool_call: nil)
  tc = tool_call
  if tc.nil? && @tracker
    tc = @tracker.tool_call_model(external_id)
  end
  tc ||= Schema::ToolCallUpdate.new(
    tool_call_id: external_id,
    title: description || external_id
  )

  opts = options || default_permission_options

  @requester.call(
    session_id: session_id,
    tool_call: tc,
    options: opts
  )
end