Class: Aidp::Providers::Kilocode

Inherits:
Base
  • Object
show all
Includes:
DebugMixin
Defined in:
lib/aidp/providers/kilocode.rb

Constant Summary

Constants included from DebugMixin

DebugMixin::DEBUG_BASIC, DebugMixin::DEBUG_OFF, DebugMixin::DEBUG_VERBOSE

Constants inherited from Base

Base::ACTIVITY_STATES, Base::DEFAULT_STUCK_TIMEOUT, Base::TIMEOUT_ARCHITECTURE_ANALYSIS, Base::TIMEOUT_DEFAULT, Base::TIMEOUT_DOCUMENTATION_ANALYSIS, Base::TIMEOUT_FUNCTIONALITY_ANALYSIS, Base::TIMEOUT_IMPLEMENTATION, Base::TIMEOUT_QUICK_MODE, Base::TIMEOUT_REFACTORING_RECOMMENDATIONS, Base::TIMEOUT_REPOSITORY_ANALYSIS, Base::TIMEOUT_STATIC_ANALYSIS, Base::TIMEOUT_TEST_ANALYSIS

Constants included from MessageDisplay

MessageDisplay::COLOR_MAP

Instance Attribute Summary

Attributes inherited from Base

#activity_state, #last_activity_time, #start_time, #step_name, #stuck_timeout

Class Method Summary collapse

Instance Method Summary collapse

Methods included from DebugMixin

#debug_basic?, #debug_command, #debug_enabled?, #debug_error, #debug_execute_command, #debug_level, #debug_log, #debug_logger, #debug_provider, #debug_step, #debug_timing, #debug_verbose?, included, shared_logger

Methods inherited from Base

#activity_summary, #available?, #execution_time, #fetch_mcp_servers, #harness_config, #harness_health_status, #harness_healthy?, #harness_metrics, #harness_mode?, #initialize, #record_harness_request, #send_with_harness, #set_harness_context, #set_job_context, #stuck?, #supports_activity_monitoring?, #supports_mcp?, #time_since_last_activity, #update_activity_state

Methods included from Adapter

#available?, #capabilities, #classify_error, #dangerous_mode=, #dangerous_mode_enabled?, #dangerous_mode_flags, #error_metadata, #error_patterns, #fetch_mcp_servers, #health_status, #logging_metadata, #redact_secrets, #retryable_error?, #supports_dangerous_mode?, #supports_mcp?, #validate_config

Methods included from MessageDisplay

#display_message, #in_test_environment?, included, #message_display_prompt, #suppress_display_message?

Constructor Details

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

Class Method Details

.available? ⇒ Boolean

Returns:

  • (Boolean) 


13
14
15
 
# File 'lib/aidp/providers/kilocode.rb', line 13

def self.available?
  !!Aidp::Util.which("kilocode")
end

Instance Method Details

#display_name ⇒ Object 



21
22
23
 
# File 'lib/aidp/providers/kilocode.rb', line 21

def display_name
  "Kilocode"
end

#name ⇒ Object 



17
18
19
 
# File 'lib/aidp/providers/kilocode.rb', line 17

def name
  "kilocode"
end

#send_message(prompt:, session: nil) ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
 
# File 'lib/aidp/providers/kilocode.rb', line 25

def send_message(prompt:, session: nil)
  raise "kilocode not available" unless self.class.available?

  # Smart timeout calculation
  timeout_seconds = calculate_timeout

  debug_provider("kilocode", "Starting execution", {timeout: timeout_seconds})
  debug_log("📝 Sending prompt to kilocode (length: #{prompt.length})", level: :info)

  # Check if streaming mode is enabled
  streaming_enabled = ENV["AIDP_STREAMING"] == "1" || ENV["DEBUG"] == "1"
  if streaming_enabled
    display_message("📺 Display streaming enabled - output buffering reduced", type: :info)
  end

  # Check if prompt is too large and warn
  if prompt.length > 3000
    debug_log("⚠️  Large prompt detected (#{prompt.length} chars) - this may cause rate limiting", level: :warn)
  end

  # Set up activity monitoring
  setup_activity_monitoring("kilocode", method(:activity_callback))
  record_activity("Starting kilocode execution")

  # Create a spinner for activity display
   = TTY::Spinner.new("[:spinner] :title", format: :dots, hide_cursor: true)
  .auto_spin

  activity_display_thread = Thread.new do
    start_time = Time.now
    loop do
      sleep 0.5 # Update every 500ms to reduce spam
      elapsed = Time.now - start_time

      # Break if we've been running too long or state changed
      break if elapsed > timeout_seconds || @activity_state == :completed || @activity_state == :failed

      (, elapsed, "🔄 kilocode")
    end
  end

  begin
    # Build kilocode command arguments
    args = ["--auto"]

    # Add model if specified
    model = ENV["KILOCODE_MODEL"]
    if model
      args.concat(["-m", model])
    end

    # Add workspace detection if needed
    if Dir.exist?(".git") && ENV["KILOCODE_WORKSPACE"]
      args.concat(["--workspace", ENV["KILOCODE_WORKSPACE"]])
    end

    # Set authentication via environment variable
    env_vars = {}
    if ENV["KILOCODE_TOKEN"]
      env_vars["KILOCODE_TOKEN"] = ENV["KILOCODE_TOKEN"]
    end

    # Use debug_execute_command for better debugging
    result = debug_execute_command("kilocode", args: args, input: prompt, timeout: timeout_seconds, streaming: streaming_enabled, env: env_vars)

    # Log the results
    debug_command("kilocode", args: args, input: prompt, output: result.out, error: result.err, exit_code: result.exit_status)

    if result.exit_status == 0
      .success("✓")
      mark_completed
      result.out
    else
      .error("✗")
      mark_failed("kilocode failed with exit code #{result.exit_status}")
      debug_error(StandardError.new("kilocode failed"), {exit_code: result.exit_status, stderr: result.err})
      raise "kilocode failed with exit code #{result.exit_status}: #{result.err}"
    end
  rescue => e
    &.error("✗")
    mark_failed("kilocode execution failed: #{e.message}")
    debug_error(e, {provider: "kilocode", prompt_length: prompt.length})
    raise
  ensure
    cleanup_activity_display(activity_display_thread, )
  end
end