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
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_rate_limit_reset, #parse_test_error, #plan_execution, #preflight_check, #sandboxed_environment?, #send_chat_message, #send_message, #subscription_unset_vars

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_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) 


37
38
39
40
 
# File 'lib/agent_harness/providers/kilocode.rb', line 37

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

.binary_name ⇒ Object 



33
34
35
 
# File 'lib/agent_harness/providers/kilocode.rb', line 33

def binary_name
  "kilo"
end

.discover_models ⇒ Object 



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

def discover_models
  return [] unless available?
  []
end

.firewall_requirements ⇒ Object 



42
43
44
45
46
47
 
# File 'lib/agent_harness/providers/kilocode.rb', line 42

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

.install_command(version: DEFAULT_VERSION) ⇒ Object 



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

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

.installation_contract(version: DEFAULT_VERSION) ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/agent_harness/providers/kilocode.rb', line 58

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 



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

def instruction_file_paths
  []
end

.provider_name ⇒ Object 



29
30
31
 
# File 'lib/agent_harness/providers/kilocode.rb', line 29

def provider_name
  :kilocode
end

.smoke_test_contract ⇒ Object 



79
80
81
 
# File 'lib/agent_harness/providers/kilocode.rb', line 79

def smoke_test_contract
  Base::DEFAULT_SMOKE_TEST_CONTRACT
end

Instance Method Details

#capabilities ⇒ Object 



121
122
123
124
125
126
127
128
129
130
131
 
# File 'lib/agent_harness/providers/kilocode.rb', line 121

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 



166
167
168
169
170
171
172
173
174
175
176
177
 
# File 'lib/agent_harness/providers/kilocode.rb', line 166

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

  config.to_json
end

#display_name ⇒ Object 



113
114
115
 
# File 'lib/agent_harness/providers/kilocode.rb', line 113

def display_name
  "Kilocode CLI"
end

#error_patterns ⇒ Object 



179
180
181
 
# File 'lib/agent_harness/providers/kilocode.rb', line 179

def error_patterns
  COMMON_ERROR_PATTERNS
end

#execution_semantics ⇒ Object 



183
184
185
186
187
188
189
190
191
192
193
194
 
# File 'lib/agent_harness/providers/kilocode.rb', line 183

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: false
  }
end

#heartbeat_integration(heartbeat_file_path:) ⇒ Object 



137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
 
# File 'lib/agent_harness/providers/kilocode.rb', line 137

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 



109
110
111
 
# File 'lib/agent_harness/providers/kilocode.rb', line 109

def name
  "kilocode"
end

#supports_activity_heartbeat? ⇒ Boolean

Returns:

  • (Boolean) 


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

def supports_activity_heartbeat?
  true
end

#test_command_overrides ⇒ Object 



117
118
119
 
# File 'lib/agent_harness/providers/kilocode.rb', line 117

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