Class: Legion::CLI::Chat::Tools::BudgetStatus

Inherits:
RubyLLM::Tool
  • Object
show all
Defined in:
lib/legion/cli/chat/tools/budget_status.rb

Instance Method Summary collapse

Instance Method Details

#execute(action: 'status') ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/legion/cli/chat/tools/budget_status.rb', line 23

def execute(action: 'status')
  return 'Legion::LLM not available.' unless llm_available?

  case action.to_s
  when 'summary' then format_summary
  else format_status
  end
rescue StandardError => e
  Legion::Logging.warn("BudgetStatus#execute failed: #{e.message}") if defined?(Legion::Logging)
  "Error checking budget: #{e.message}"
end