Class: AgentHarness::Providers::Kilocode

Inherits:
Base
  • Object
show all
Defined in:
lib/agent_harness/providers/kilocode.rb

Overview

Kilocode CLI provider

Provides integration with the Kilocode CLI tool.

Constant Summary collapse

PACKAGE_NAME = 
"@kilocode/cli"
DEFAULT_VERSION = 
"7.1.3"
SUPPORTED_VERSION_REQUIREMENT = 
"= #{DEFAULT_VERSION}"
STRUCTURED_EVENT_TYPES = 
%w[text error step_finish result usage].freeze
DEFAULT_PERMISSION_EXTERNAL_DIRECTORY_PATTERNS =

Kilo CLI (an OpenCode fork) ships the same external_directory permission category as OpenCode, defaulting to "ask" for anything outside the project dir. In non-interactive execution there is no human to answer the prompt, so dedicated read/write/edit tool calls targeting paths outside the project dir are auto-rejected and the agent silently loses access to its own scratch output or its own config/data files, killing the run without a recoverable error.

This default rule broadens the allowlist to all of /tmp (so the agent can read back files it or its sub-agents wrote there) and to the full agent home directory (so the agent can inspect/maintain its own config, cache, and data files such as ~/.config/kilocode, ~/.local/share/kilo, ~/.cache). The container is already isolated (Docker, non-root user) and the agent runs with --auto which approves everything inside the project dir. See #289 (precedent: #282/#277).

["/tmp/**", "/home/agent/**"].freeze
DEFAULT_PERMISSION_CONFIG = 
{
  "external_directory" => DEFAULT_PERMISSION_EXTERNAL_DIRECTORY_PATTERNS
    .to_h { |pattern| [pattern, "allow"] }
    .freeze
}.freeze
USAGE_EVENT_TYPES = 
%w[result usage].freeze
TOKEN_USAGE_KEYS = 
%w[
  input_tokens
  output_tokens
  total_tokens
  total
  reasoning_tokens
  cache_creation_input_tokens
  cache_read_input_tokens
  cache_write_input_tokens
].freeze

Constants inherited from Base

Base::COMMON_ERROR_PATTERNS, Base::DEFAULT_SMOKE_TEST_CONTRACT

Instance Attribute Summary

Attributes inherited from Base

#config, #executor, #logger

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Base

#api_key_env_var_names, #api_key_unset_vars, #cli_env_overrides, #configure, #initialize, #parse_container_output, #parse_test_error, #plan_execution, #preflight_check, #sandboxed_environment?, #send_chat_message, #send_message, #subscription_unset_vars

Methods included from RateLimitResetParsing

#parse_rate_limit_reset

Methods included from Adapter

#auth_lock_config, #auth_type, #build_mcp_flags, #chat_transport, #chat_transport_type, #configuration_schema, #dangerous_mode_flags, #error_classification_patterns, #fetch_mcp_servers, #health_status, included, metadata_package_name, #noisy_error_patterns, normalize_metadata_installation, normalize_metadata_source_type, normalize_metadata_version_requirement, #notify_hook_content, #parse_rate_limit_reset, #plan_execution, #preflight_check, #send_message, #session_flags, #smoke_test, #smoke_test_contract, #supported_mcp_transports, #supports_chat?, #supports_dangerous_mode?, #supports_mcp?, #supports_message_tool_injection?, #supports_sessions?, #supports_text_mode?, #supports_token_counting?, #supports_tool_control?, #token_usage_from_api_response, #translate_error, #validate_config, #validate_mcp_servers!

Constructor Details

This class inherits a constructor from AgentHarness::Providers::Base

Class Method Details

.available? ⇒ Boolean

Returns:

  • (Boolean) 


58
59
60
61
 
# File 'lib/agent_harness/providers/kilocode.rb', line 58

def available?
  executor = AgentHarness.configuration.command_executor
  !!executor.which(binary_name)
end

.binary_name ⇒ Object 



54
55
56
 
# File 'lib/agent_harness/providers/kilocode.rb', line 54

def binary_name
  "kilo"
end

.discover_models ⇒ Object 



74
75
76
77
 
# File 'lib/agent_harness/providers/kilocode.rb', line 74

def discover_models
  return [] unless available?
  []
end

.firewall_requirements ⇒ Object 



63
64
65
66
67
68
 
# File 'lib/agent_harness/providers/kilocode.rb', line 63

def firewall_requirements
  {
    domains: [],
    ip_ranges: []
  }
end

.install_command(version: DEFAULT_VERSION) ⇒ Object 



96
97
98
 
# File 'lib/agent_harness/providers/kilocode.rb', line 96

def install_command(version: DEFAULT_VERSION)
  installation_contract(version: version)[:install_command]
end

.installation_contract(version: DEFAULT_VERSION) ⇒ Object 



79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
 
# File 'lib/agent_harness/providers/kilocode.rb', line 79

def installation_contract(version: DEFAULT_VERSION)
  version = version.strip if version.respond_to?(:strip)
  validate_install_version!(version)
  package_spec = "#{PACKAGE_NAME}@#{version}"

  {
    source: {
      type: :npm,
      package: PACKAGE_NAME
    },
    install_command: ["npm", "install", "-g", "--ignore-scripts", package_spec],
    binary_name: binary_name,
    default_version: DEFAULT_VERSION,
    supported_version_requirement: SUPPORTED_VERSION_REQUIREMENT
  }
end

.instruction_file_paths ⇒ Object 



70
71
72
 
# File 'lib/agent_harness/providers/kilocode.rb', line 70

def instruction_file_paths
  []
end

.provider_name ⇒ Object 



50
51
52
 
# File 'lib/agent_harness/providers/kilocode.rb', line 50

def provider_name
  :kilocode
end

.smoke_test_contract ⇒ Object 



100
101
102
 
# File 'lib/agent_harness/providers/kilocode.rb', line 100

def smoke_test_contract
  Base::DEFAULT_SMOKE_TEST_CONTRACT
end

Instance Method Details

#capabilities ⇒ Object 



142
143
144
145
146
147
148
149
150
151
152
 
# File 'lib/agent_harness/providers/kilocode.rb', line 142

def capabilities
  {
    streaming: false,
    file_upload: false,
    vision: false,
    tool_use: false,
    json_mode: false,
    mcp: false,
    dangerous_mode: false
  }
end

#config_file_content(options = {}) ⇒ Object 



187
188
189
190
191
192
193
194
195
196
197
198
199
 
# File 'lib/agent_harness/providers/kilocode.rb', line 187

def config_file_content(options = {})
  # Only use explicit provider_name or default to "openai".
  # api_provider is a generic backend label (e.g. "openrouter") that is not
  # a valid Kilo built-in provider ID, so we must not fall back to it here.
  provider_name = options[:provider_name] || "openai"
  model_id = options[:model_id]

  config = {"provider" => {provider_name => {}}}
  config["model"] = "#{provider_name}/#{model_id}" if model_id
  apply_default_external_directory_permission(config, options)

  config.to_json
end

#display_name ⇒ Object 



134
135
136
 
# File 'lib/agent_harness/providers/kilocode.rb', line 134

def display_name
  "Kilocode CLI"
end

#error_patterns ⇒ Object 



201
202
203
 
# File 'lib/agent_harness/providers/kilocode.rb', line 201

def error_patterns
  COMMON_ERROR_PATTERNS
end

#execution_semantics ⇒ Object 



205
206
207
208
209
210
211
212
213
214
215
216
 
# File 'lib/agent_harness/providers/kilocode.rb', line 205

def execution_semantics
  {
    prompt_delivery: :arg,
    output_format: :json,
    sandbox_aware: false,
    uses_subcommand: true,
    non_interactive_flag: nil,
    legitimate_exit_codes: [0],
    stderr_is_diagnostic: true,
    parses_rate_limit_reset: true
  }
end

#heartbeat_integration(heartbeat_file_path:) ⇒ Object 



158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
 
# File 'lib/agent_harness/providers/kilocode.rb', line 158

def heartbeat_integration(heartbeat_file_path:)
  unless heartbeat_file_path.is_a?(String) && !heartbeat_file_path.strip.empty?
    raise ArgumentError, "heartbeat_file_path must be a non-empty String"
  end
  unless heartbeat_file_path.start_with?("/")
    raise ArgumentError, "heartbeat_file_path must be an absolute path (got #{heartbeat_file_path.inspect})"
  end

  hook_script = heartbeat_hook_script(heartbeat_file_path)
  config_payload = merge_heartbeat_hooks(hook_script)

  preparation = ExecutionPreparation.new(
    file_writes: [
      {
        path: heartbeat_hook_config_path,
        content: JSON.pretty_generate(config_payload),
        mode: 0o600
      }
    ]
  )

  {
    supported: true,
    env: {"KILO_HEARTBEAT_FILE" => heartbeat_file_path},
    preparation: preparation,
    granularity: :tool_call
  }
end

#name ⇒ Object 



130
131
132
 
# File 'lib/agent_harness/providers/kilocode.rb', line 130

def name
  "kilocode"
end

#supports_activity_heartbeat? ⇒ Boolean

Returns:

  • (Boolean) 


154
155
156
 
# File 'lib/agent_harness/providers/kilocode.rb', line 154

def supports_activity_heartbeat?
  true
end

#test_command_overrides ⇒ Object 



138
139
140
 
# File 'lib/agent_harness/providers/kilocode.rb', line 138

def test_command_overrides
  ["--auto", "--print-logs"]
end