legionio 1.9.30 → 1.9.33

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ba59d5cee637b2d8c6e78dfeb20d481f432d65e22ef4d76346f1aa6539a86752
-  data.tar.gz: 8d80bd7e851ecf13e2a9f798c4465a2d5e7f9e21110b82b9ebf08cef6bf87901
+  metadata.gz: a0e7ab0c576dcf1950fb2dd7d38d1113dfd5f584883cd30286cd990f851ce6a3
+  data.tar.gz: a4dc69a591dd00dbb757f5e1bc5b8913005e374fed607b63d5f28088e51018a1
 SHA512:
-  metadata.gz: a12da66d229b6a3d6c1bf6cb84d84cc09d97e0b773220faafa5d15d73d57796d7c655dec36b6f258661c2a71c4f883897cba18ba7c111ee446db46792e65c3c3
-  data.tar.gz: 9dc487e8582733e0a5cf0fec2e57c6d80fe0f78641ce3ba0e9b816d17747cbb99c78ae051c6ab59b53ab294af0304968fdf077ce85504dd29ee6403b1ea88e7c
+  metadata.gz: ef50ae89f614aeec8385558444afb27f57aa1797e415ce25361ec57dec5d87ecf1181fdaf9e59bf0e696cc4ec8d68c0aa9c641bb30b1c01c08a72dd5792d0e6e
+  data.tar.gz: e9049722257ccdc3415ec4e33af0100d692dbf85806cd2c1fb1f4094c1189364d575943e0f39907bcc295ec7fecabfd8065aae57c0857997a22dd0904ea958ea

data/CHANGELOG.md

@@ -2,6 +2,36 @@
 
 ## [Unreleased]
 
+## [1.9.33] - 2026-05-15
+
+### Added
+- `Legion::Identity::Process` stores and exposes `db_principal_id` and `db_identity_id` integer PKs — present in `EMPTY_STATE`, persisted through `bind!`, and included in `identity_hash`. Both default to nil until an identity provider populates them.
+
+## [1.9.32] - 2026-05-14
+
+### Removed
+- Removed `gem 'ruby_llm'` dependency from Gemfile; all 40 CLI chat tools now use `Legion::Tools::Base` natively.
+
+### Changed
+- Migrated all 40 CLI chat tool classes from `RubyLLM::Tool` to `Legion::Tools::Base`:
+  - `param` DSL replaced with `input_schema` (JSON Schema hash)
+  - `def execute` instance method replaced with `def self.call` class method
+  - Explicit `tool_name 'legion.<snake_case>'` added to each tool
+  - Private instance helpers converted to class methods
+- Updated `tool_registry.rb`: removed `require 'ruby_llm'` and `begin/rescue LoadError` guard.
+- Updated `extension_tool_loader.rb`: `klass < RubyLLM::Tool` changed to `klass < Legion::Tools::Base`.
+- Updated `generate_command.rb` tool template to emit `Legion::Tools::Base` with `input_schema` and `def self.call`.
+- `Permissions::Gate` now prepends on the singleton class to intercept `self.call` correctly.
+
+## [1.9.31] - 2026-05-14
+
+### Added
+- `GET /api/identity` endpoint returning live process identity, provider resolution status, and registered provider metadata.
+- `autobuild_submodules` recursive walk in `Legion::Extensions` — nested sub-modules (e.g. `Delegated`, `Application`, `ManagedIdentity`, `WorkloadIdentity` inside `lex-identity-entra`) now have their actors autobuilt and started.
+
+### Fixed
+- `Extensions::Helpers::Base#full_path` now walks up gem name segments to find the parent gem when a sub-module gem doesn't exist as a standalone gem (e.g. `lex-identity-entra-delegated`).
+
 ## [1.9.30] - 2026-05-12
 
 ### Changed

data/Gemfile

@@ -3,64 +3,41 @@
 source 'https://rubygems.org'
 
 gemspec
-
-def local_gem_version(path, version_file)
-  version_path = File.expand_path(File.join(path, version_file), __dir__)
-  return unless File.file?(version_path)
-
-  version_source = File.read(version_path)
-  version_source[/VERSION\s*=\s*['"]([^'"]+)['"]/, 1]
-end
-
-def local_gem_satisfies?(path, version_file, requirement)
-  version = local_gem_version(path, version_file)
-  version && Gem::Requirement.new(requirement).satisfied_by?(Gem::Version.new(version))
-end
-
-def local_gem_path(name, default_path, version_file, requirement)
-  env_name = "#{name.upcase.tr('-', '_')}_PATH"
-  env_path = ENV.fetch(env_name, nil)
-  return env_path if env_path && File.exist?(File.expand_path(env_path, __dir__))
-
-  return unless File.exist?(File.expand_path(default_path, __dir__))
-  return unless local_gem_satisfies?(default_path, version_file, requirement)
-
-  default_path
-end
-
-gem 'legion-apollo', path: '../legion-apollo' if File.exist?(File.expand_path('../legion-apollo', __dir__))
-gem 'legion-data', path: '../legion-data' if File.exist?(File.expand_path('../legion-data', __dir__))
-gem 'legion-logging', path: '../legion-logging' if File.exist?(File.expand_path('../legion-logging', __dir__))
-gem 'legion-settings', path: '../legion-settings' if File.exist?(File.expand_path('../legion-settings', __dir__))
-if (legion_tty_path = local_gem_path('legion-tty', '../legion-tty', 'lib/legion/tty/version.rb', '>= 0.5.4'))
-  gem 'legion-tty', path: legion_tty_path
-end
-
-gem 'legion-gaia', path: '../legion-gaia' if File.exist?(File.expand_path('../legion-gaia', __dir__))
-if (legion_llm_path = local_gem_path('legion-llm', '../legion-llm', 'lib/legion/llm/version.rb', '>= 0.8.47'))
-  gem 'legion-llm', path: legion_llm_path
-end
-gem 'legion-mcp', path: '../legion-mcp' if File.exist?(File.expand_path('../legion-mcp', __dir__))
-
-gem 'lex-kerberos'
-
-gem 'lex-apollo', path: '../extensions/lex-apollo' if File.exist?(File.expand_path('../extensions/lex-apollo', __dir__))
-gem 'lex-llm', path: '../extensions-ai/lex-llm' if File.exist?(File.expand_path('../extensions-ai/lex-llm', __dir__))
-gem 'lex-llm-ledger', path: '../extensions-ai/lex-llm-ledger' if File.exist?(File.expand_path('../extensions-ai/lex-llm-ledger', __dir__))
-
-%w[anthropic azure-foundry bedrock gemini mlx ollama openai vertex vllm].each do |provider|
-  provider_path = "../extensions-ai/lex-llm-#{provider}"
-  gem "lex-llm-#{provider}", path: provider_path if File.exist?(File.expand_path(provider_path, __dir__))
-end
-
-# gem 'lex-microsoft_teams', path: '../extensions/lex-microsoft_teams' if File.exist?(File.expand_path('../extensions/lex-microsoft_teams', __dir__))
-
 gem 'pg'
 
 gem 'kramdown', '>= 2.0'
 gem 'mysql2'
 
 group :test do
+  gem 'legion-data', path: '../legion-data' if File.exist?(File.expand_path('../legion-data', __dir__))
+  gem 'legion-logging', path: '../legion-logging' if File.exist?(File.expand_path('../legion-logging', __dir__))
+  gem 'legion-settings', path: '../legion-settings' if File.exist?(File.expand_path('../legion-settings', __dir__))
+
+  gem 'legion-apollo', path: '../legion-apollo' if File.exist?(File.expand_path('../legion-apollo', __dir__))
+  gem 'legion-gaia', path: '../legion-gaia' if File.exist?(File.expand_path('../legion-gaia', __dir__))
+  gem 'legion-llm', path: '../legion-llm' if File.exist?(File.expand_path('../legion-llm', __dir__))
+  gem 'legion-mcp', path: '../legion-mcp' if File.exist?(File.expand_path('../legion-mcp', __dir__))
+  gem 'legion-tty', path: '../legion-tty' if File.exist?(File.expand_path('../legion-tty', __dir__))
+
+  gem 'lex-apollo', path: '../extensions/lex-apollo' if File.exist?(File.expand_path('../extensions/lex-apollo', __dir__))
+  gem 'lex-llm', path: '../extensions-ai/lex-llm' if File.exist?(File.expand_path('../extensions-ai/lex-llm', __dir__))
+  gem 'lex-llm-ledger', path: '../extensions-ai/lex-llm-ledger' if File.exist?(File.expand_path('../extensions-ai/lex-llm-ledger', __dir__))
+
+  if File.exist?(File.expand_path('../extensions-identity/lex-identity-entra', __dir__))
+    gem 'lex-identity-entra', path: '../extensions-identity/lex-identity-entra'
+  end
+  if File.exist?(File.expand_path('../extensions-identity/lex-identity-kerberos', __dir__))
+    gem 'lex-identity-kerberos', path: '../extensions-identity/lex-identity-kerberos'
+  end
+  if File.exist?(File.expand_path('../extensions-identity/lex-identity-system', __dir__))
+    gem 'lex-identity-system', path: '../extensions-identity/lex-identity-system'
+  end
+
+  %w[anthropic azure-foundry bedrock gemini mlx ollama openai vertex vllm].each do |provider|
+    provider_path = "../extensions-ai/lex-llm-#{provider}"
+    gem "lex-llm-#{provider}", path: provider_path if File.exist?(File.expand_path(provider_path, __dir__))
+  end
+
   gem 'faraday'
   gem 'faraday-net_http'
   gem 'graphql'
@@ -72,6 +49,5 @@ group :test do
   gem 'rubocop'
   gem 'rubocop-legion'
   gem 'rubocop-rspec'
-  gem 'ruby_llm'
   gem 'simplecov'
 end

data/lib/legion/api/identity_audit.rb

@@ -7,6 +7,26 @@ module Legion
         def self.registered(app)
           app.helpers IdentityAuditHelpers
 
+          app.get '/api/identity' do
+            identity = defined?(Legion::Identity::Process) ? Legion::Identity::Process.identity_hash : {}
+
+            registered_providers = if defined?(Legion::Identity::Resolver)
+                                     Legion::Identity::Resolver.providers.map do |p|
+                                       {
+                                         name:         p.provider_name,
+                                         type:         p.provider_type,
+                                         trust_level:  p.trust_level,
+                                         priority:     p.respond_to?(:priority) ? p.priority : nil,
+                                         capabilities: p.respond_to?(:capabilities) ? p.capabilities : []
+                                       }
+                                     end
+                                   else
+                                     []
+                                   end
+
+            json_response(identity.merge(registered_providers: registered_providers))
+          end
+
           app.get '/api/identity/audit' do
             require_data!
             halt 503, json_error('unavailable', 'identity audit log not available') unless defined?(Legion::Data::Model::Identity::AuditLog)

data/lib/legion/cli/chat/extension_tool_loader.rb

@@ -24,7 +24,7 @@ module Legion
           def collect_tool_classes(tools_module)
             tools_module.constants.filter_map do |const_name|
               klass = tools_module.const_get(const_name)
-              klass if klass.is_a?(Class) && klass < RubyLLM::Tool
+              klass if klass.is_a?(Class) && klass < Legion::Tools::Base
             end
           end
 

data/lib/legion/cli/chat/permissions.rb

@@ -58,16 +58,15 @@ module Legion
           def apply!(tool_classes)
             tool_classes.each do |klass|
               tier = tier_for(klass)
-              klass.prepend(Gate) unless tier == :read
+              klass.singleton_class.prepend(Gate) unless tier == :read
             end
           end
         end
 
         module Gate
-          def call(args)
-            normalized = normalize_args(args)
-            desc = permission_description(normalized)
-            return 'Tool execution denied by user.' unless Permissions.confirm?(desc)
+          def call(**args)
+            desc = permission_description(args)
+            return error_response('Tool execution denied by user.') unless Permissions.confirm?(desc)
 
             super
           end
@@ -75,11 +74,11 @@ module Legion
           private
 
           def permission_description(args)
-            tier = Permissions.tier_for(self.class)
+            tier = Permissions.tier_for(self)
             case tier
             when :write
               path = args[:path] || '(unknown)'
-              action = self.class.name.split('::').last.gsub(/([a-z])([A-Z])/, '\1 \2')
+              action = name.split('::').last.gsub(/([a-z])([A-Z])/, '\1 \2')
               "#{action}: #{path}"
             when :shell
               "Run command: #{args[:command]}"

data/lib/legion/cli/chat/tool_registry.rb

@@ -2,52 +2,46 @@
 
 require 'legion/cli/chat_command'
 
-begin
-  require 'ruby_llm'
-
-  require 'legion/cli/chat/tools/read_file'
-  require 'legion/cli/chat/tools/write_file'
-  require 'legion/cli/chat/tools/edit_file'
-  require 'legion/cli/chat/tools/search_files'
-  require 'legion/cli/chat/tools/search_content'
-  require 'legion/cli/chat/tools/run_command'
-  require 'legion/cli/chat/tools/save_memory'
-  require 'legion/cli/chat/tools/search_memory'
-  require 'legion/cli/chat/tools/web_search'
-  require 'legion/cli/chat/tools/spawn_agent'
-  require 'legion/cli/chat/tools/search_traces'
-  require 'legion/cli/chat/tools/query_knowledge'
-  require 'legion/cli/chat/tools/ingest_knowledge'
-  require 'legion/cli/chat/tools/consolidate_memory'
-  require 'legion/cli/chat/tools/relate_knowledge'
-  require 'legion/cli/chat/tools/knowledge_maintenance'
-  require 'legion/cli/chat/tools/knowledge_stats'
-  require 'legion/cli/chat/tools/summarize_traces'
-  require 'legion/cli/chat/tools/list_extensions'
-  require 'legion/cli/chat/tools/manage_tasks'
-  require 'legion/cli/chat/tools/system_status'
-  require 'legion/cli/chat/tools/view_events'
-  require 'legion/cli/chat/tools/cost_summary'
-  require 'legion/cli/chat/tools/reflect'
-  require 'legion/cli/chat/tools/manage_schedules'
-  require 'legion/cli/chat/tools/worker_status'
-  require 'legion/cli/chat/tools/detect_anomalies'
-  require 'legion/cli/chat/tools/view_trends'
-  require 'legion/cli/chat/tools/trigger_dream'
-  require 'legion/cli/chat/tools/generate_insights'
-  require 'legion/cli/chat/tools/budget_status'
-  require 'legion/cli/chat/tools/provider_health'
-  require 'legion/cli/chat/tools/model_comparison'
-  require 'legion/cli/chat/tools/shadow_eval_status'
-  require 'legion/cli/chat/tools/entity_extract'
-  require 'legion/cli/chat/tools/arbitrage_status'
-  require 'legion/cli/chat/tools/escalation_status'
-  require 'legion/cli/chat/tools/graph_explore'
-  require 'legion/cli/chat/tools/scheduling_status'
-  require 'legion/cli/chat/tools/memory_status'
-rescue LoadError => e
-  Legion::Logging.debug("ToolRegistry ruby_llm not available, chat tools will not be registered: #{e.message}") if defined?(Legion::Logging)
-end
+require 'legion/cli/chat/tools/read_file'
+require 'legion/cli/chat/tools/write_file'
+require 'legion/cli/chat/tools/edit_file'
+require 'legion/cli/chat/tools/search_files'
+require 'legion/cli/chat/tools/search_content'
+require 'legion/cli/chat/tools/run_command'
+require 'legion/cli/chat/tools/save_memory'
+require 'legion/cli/chat/tools/search_memory'
+require 'legion/cli/chat/tools/web_search'
+require 'legion/cli/chat/tools/spawn_agent'
+require 'legion/cli/chat/tools/search_traces'
+require 'legion/cli/chat/tools/query_knowledge'
+require 'legion/cli/chat/tools/ingest_knowledge'
+require 'legion/cli/chat/tools/consolidate_memory'
+require 'legion/cli/chat/tools/relate_knowledge'
+require 'legion/cli/chat/tools/knowledge_maintenance'
+require 'legion/cli/chat/tools/knowledge_stats'
+require 'legion/cli/chat/tools/summarize_traces'
+require 'legion/cli/chat/tools/list_extensions'
+require 'legion/cli/chat/tools/manage_tasks'
+require 'legion/cli/chat/tools/system_status'
+require 'legion/cli/chat/tools/view_events'
+require 'legion/cli/chat/tools/cost_summary'
+require 'legion/cli/chat/tools/reflect'
+require 'legion/cli/chat/tools/manage_schedules'
+require 'legion/cli/chat/tools/worker_status'
+require 'legion/cli/chat/tools/detect_anomalies'
+require 'legion/cli/chat/tools/view_trends'
+require 'legion/cli/chat/tools/trigger_dream'
+require 'legion/cli/chat/tools/generate_insights'
+require 'legion/cli/chat/tools/budget_status'
+require 'legion/cli/chat/tools/provider_health'
+require 'legion/cli/chat/tools/model_comparison'
+require 'legion/cli/chat/tools/shadow_eval_status'
+require 'legion/cli/chat/tools/entity_extract'
+require 'legion/cli/chat/tools/arbitrage_status'
+require 'legion/cli/chat/tools/escalation_status'
+require 'legion/cli/chat/tools/graph_explore'
+require 'legion/cli/chat/tools/scheduling_status'
+require 'legion/cli/chat/tools/memory_status'
 
 require 'legion/cli/chat/permissions'
 
@@ -55,54 +49,50 @@ module Legion
   module CLI
     class Chat
       module ToolRegistry
-        BUILTIN_TOOLS = if defined?(Tools::ReadFile)
-                          [
-                            Tools::ReadFile,
-                            Tools::WriteFile,
-                            Tools::EditFile,
-                            Tools::SearchFiles,
-                            Tools::SearchContent,
-                            Tools::RunCommand,
-                            Tools::SaveMemory,
-                            Tools::SearchMemory,
-                            Tools::WebSearch,
-                            Tools::SpawnAgent,
-                            Tools::SearchTraces,
-                            Tools::QueryKnowledge,
-                            Tools::IngestKnowledge,
-                            Tools::ConsolidateMemory,
-                            Tools::RelateKnowledge,
-                            Tools::KnowledgeMaintenance,
-                            Tools::KnowledgeStats,
-                            Tools::SummarizeTraces,
-                            Tools::ListExtensions,
-                            Tools::ManageTasks,
-                            Tools::SystemStatus,
-                            Tools::ViewEvents,
-                            Tools::CostSummary,
-                            Tools::Reflect,
-                            Tools::ManageSchedules,
-                            Tools::WorkerStatus,
-                            Tools::DetectAnomalies,
-                            Tools::ViewTrends,
-                            Tools::TriggerDream,
-                            Tools::GenerateInsights,
-                            Tools::BudgetStatus,
-                            Tools::ProviderHealth,
-                            Tools::ModelComparison,
-                            Tools::ShadowEvalStatus,
-                            Tools::EntityExtract,
-                            Tools::ArbitrageStatus,
-                            Tools::EscalationStatus,
-                            Tools::GraphExplore,
-                            Tools::SchedulingStatus,
-                            Tools::MemoryStatus
-                          ].freeze
-                        else
-                          [].freeze
-                        end
+        BUILTIN_TOOLS = [
+          Tools::ReadFile,
+          Tools::WriteFile,
+          Tools::EditFile,
+          Tools::SearchFiles,
+          Tools::SearchContent,
+          Tools::RunCommand,
+          Tools::SaveMemory,
+          Tools::SearchMemory,
+          Tools::WebSearch,
+          Tools::SpawnAgent,
+          Tools::SearchTraces,
+          Tools::QueryKnowledge,
+          Tools::IngestKnowledge,
+          Tools::ConsolidateMemory,
+          Tools::RelateKnowledge,
+          Tools::KnowledgeMaintenance,
+          Tools::KnowledgeStats,
+          Tools::SummarizeTraces,
+          Tools::ListExtensions,
+          Tools::ManageTasks,
+          Tools::SystemStatus,
+          Tools::ViewEvents,
+          Tools::CostSummary,
+          Tools::Reflect,
+          Tools::ManageSchedules,
+          Tools::WorkerStatus,
+          Tools::DetectAnomalies,
+          Tools::ViewTrends,
+          Tools::TriggerDream,
+          Tools::GenerateInsights,
+          Tools::BudgetStatus,
+          Tools::ProviderHealth,
+          Tools::ModelComparison,
+          Tools::ShadowEvalStatus,
+          Tools::EntityExtract,
+          Tools::ArbitrageStatus,
+          Tools::EscalationStatus,
+          Tools::GraphExplore,
+          Tools::SchedulingStatus,
+          Tools::MemoryStatus
+        ].freeze
 
-        Permissions.apply!(BUILTIN_TOOLS) unless BUILTIN_TOOLS.empty?
+        Permissions.apply!(BUILTIN_TOOLS)
 
         def self.builtin_tools
           BUILTIN_TOOLS.dup

data/lib/legion/cli/chat/tools/arbitrage_status.rb

@@ -4,17 +4,20 @@ module Legion
   module CLI
     class Chat
       module Tools
-        class ArbitrageStatus < RubyLLM::Tool
+        class ArbitrageStatus < Legion::Tools::Base
+          tool_name 'legion.arbitrage_status'
           description 'Show LLM cost arbitrage status: model pricing table, cheapest model per capability tier'
-
-          param :capability,
-                type:     :string,
-                desc:     'Capability tier to check: basic, moderate, or reasoning (default: show all)',
-                required: false
+          input_schema({
+                         type:       'object',
+                         properties: {
+                           capability: { type: 'string', description: 'Capability tier to check: basic, moderate, or reasoning (default: show all)' }
+                         },
+                         required:   []
+                       })
 
           TIERS = %i[basic moderate reasoning].freeze
 
-          def execute(capability: nil)
+          def self.call(capability: nil)
             return 'LLM arbitrage module not available.' unless arbitrage_available?
 
             if capability
@@ -24,13 +27,11 @@ module Legion
             end
           end
 
-          private
-
-          def arbitrage_available?
+          def self.arbitrage_available?
             defined?(Legion::LLM::Arbitrage)
           end
 
-          def format_overview
+          def self.format_overview
             arb = Legion::LLM::Arbitrage
             lines = ["LLM Cost Arbitrage\n"]
             lines << format('  Enabled: %<v>s', v: arb.enabled? ? 'YES' : 'no')
@@ -56,7 +57,7 @@ module Legion
             lines.join("\n")
           end
 
-          def format_tier(tier)
+          def self.format_tier(tier)
             arb = Legion::LLM::Arbitrage
             return format('Invalid tier: %<t>s. Use: %<valid>s', t: tier, valid: TIERS.join(', ')) unless TIERS.include?(tier)
 

data/lib/legion/cli/chat/tools/budget_status.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'ruby_llm'
-
 begin
   require 'legion/cli/chat_command'
 rescue LoadError
@@ -12,15 +10,20 @@ module Legion
   module CLI
     class Chat
       module Tools
-        class BudgetStatus < RubyLLM::Tool
+        class BudgetStatus < Legion::Tools::Base
+          tool_name 'legion.budget_status'
           description 'Check the current LLM session cost budget status. Shows how much has been spent, ' \
                       'remaining budget, and whether the budget guard is enforcing limits. Works locally ' \
                       'without needing the Legion daemon. Use this when the user asks about spending or budget.'
-          param :action, type:     'string',
-                         desc:     'Action: "status" (default), "summary" (cost breakdown by model)',
-                         required: false
-
-          def execute(action: 'status')
+          input_schema({
+                         type:       'object',
+                         properties: {
+                           action: { type: 'string', description: 'Action: "status" (default), "summary" (cost breakdown by model)' }
+                         },
+                         required:   []
+                       })
+
+          def self.call(action: 'status')
             return 'Legion::LLM not available.' unless llm_available?
 
             case action.to_s
@@ -32,9 +35,7 @@ module Legion
             "Error checking budget: #{e.message}"
           end
 
-          private
-
-          def format_status
+          def self.format_status
             guard = budget_guard_status
             tracker = cost_summary
             lines = ["Session Budget Status:\n"]
@@ -49,7 +50,7 @@ module Legion
             lines.join("\n")
           end
 
-          def format_summary
+          def self.format_summary
             tracker = cost_summary
             return 'No LLM requests recorded this session.' if tracker[:total_requests].zero?
 
@@ -63,7 +64,7 @@ module Legion
             lines.join("\n")
           end
 
-          def append_model_breakdown(lines, by_model)
+          def self.append_model_breakdown(lines, by_model)
             return unless by_model&.any?
 
             lines << "\n  By Model:"
@@ -73,31 +74,31 @@ module Legion
             end
           end
 
-          def budget_guard_status
+          def self.budget_guard_status
             return { enforcing: false, budget_usd: 0.0, ratio: 0.0 } unless budget_guard_available?
 
             Legion::LLM::Hooks::BudgetGuard.status
           end
 
-          def cost_summary
+          def self.cost_summary
             return empty_summary unless cost_tracker_available?
 
             Legion::LLM::CostTracker.summary
           end
 
-          def budget_guard_available?
+          def self.budget_guard_available?
             defined?(Legion::LLM::Hooks::BudgetGuard)
           end
 
-          def cost_tracker_available?
+          def self.cost_tracker_available?
             defined?(Legion::LLM::CostTracker)
           end
 
-          def llm_available?
+          def self.llm_available?
             defined?(Legion::LLM)
           end
 
-          def empty_summary
+          def self.empty_summary
             { total_cost_usd: 0.0, total_requests: 0, total_input_tokens: 0, total_output_tokens: 0, by_model: {} }
           end
         end

data/lib/legion/cli/chat/tools/consolidate_memory.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'ruby_llm'
-
 begin
   require 'legion/cli/chat_command'
   require 'legion/cli/chat/memory_store'
@@ -13,12 +11,19 @@ module Legion
   module CLI
     class Chat
       module Tools
-        class ConsolidateMemory < RubyLLM::Tool
+        class ConsolidateMemory < Legion::Tools::Base
+          tool_name 'legion.consolidate_memory'
           description 'Consolidate and organize memory entries by removing duplicates, merging related items, ' \
                       'and creating cleaner summaries. Use this when memory has grown cluttered or has redundant entries. ' \
                       'Pass scope "project" or "global" to target the right memory file.'
-          param :scope, type: 'string', desc: 'Memory scope: "project" or "global" (default: project)'
-          param :dry_run, type: 'string', desc: 'Set to "true" to preview without writing (default: false)', required: false
+          input_schema({
+                         type:       'object',
+                         properties: {
+                           scope:   { type: 'string', description: 'Memory scope: "project" or "global" (default: project)' },
+                           dry_run: { type: 'string', description: 'Set to "true" to preview without writing (default: false)' }
+                         },
+                         required:   []
+                       })
 
           CONSOLIDATION_PROMPT = <<~PROMPT
             You are a memory consolidation engine. Given a list of memory entries, produce a cleaned-up version that:
@@ -34,7 +39,7 @@ module Legion
             Do NOT add headers, explanations, or commentary.
           PROMPT
 
-          def execute(scope: 'project', dry_run: nil)
+          def self.call(scope: 'project', dry_run: nil)
             dry_run = dry_run.to_s == 'true'
             scope_sym = scope.to_s == 'global' ? :global : :project
 
@@ -60,9 +65,7 @@ module Legion
             "Error consolidating memory: #{e.message}"
           end
 
-          private
-
-          def consolidate_entries(entries)
+          def self.consolidate_entries(entries)
             return nil unless defined?(Legion::LLM) && Legion::LLM.respond_to?(:chat_direct)
 
             numbered = entries.map.with_index(1) { |e, i| "#{i}. #{e}" }.join("\n")
@@ -72,7 +75,7 @@ module Legion
             response.content
           end
 
-          def parse_consolidated(text)
+          def self.parse_consolidated(text)
             text.lines
                 .map(&:strip)
                 .select { |line| line.start_with?('- ') }
@@ -80,7 +83,7 @@ module Legion
                 .reject(&:empty?)
           end
 
-          def write_consolidated(entries, scope_sym)
+          def self.write_consolidated(entries, scope_sym)
             path = scope_sym == :global ? MemoryStore.global_path : MemoryStore.project_path
             header = scope_sym == :global ? "# Global Memory\n" : "# Project Memory\n"
             timestamp = Time.now.strftime('%Y-%m-%d %H:%M')