legionio 1.9.31 → 1.9.33

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 765d0bb57b5a10135d49ccbaca56669a33b201118421fc90242acd4f12d752ec
-  data.tar.gz: 464deff220c28495e29c7def972d4c5a7c925abdee6572991fd01d84e6791878
+  metadata.gz: a0e7ab0c576dcf1950fb2dd7d38d1113dfd5f584883cd30286cd990f851ce6a3
+  data.tar.gz: a4dc69a591dd00dbb757f5e1bc5b8913005e374fed607b63d5f28088e51018a1
 SHA512:
-  metadata.gz: aba6393b1e60a435ff0e752ae6b7d5a035387a1e6012ec997d3c0f1c0ee31d408d7fb78f8ed4340e1594963b0510189047ea1fd0f6d49f18b7d914ff247fa60b
-  data.tar.gz: dd3c0bd6abd74759c9858c8192b8938dbdf17333093e9173bdc681868c1f0fd039fb87f9659e8a0388bde76e47cfd71a97937e24f619c31a9d2b32476aded675
+  metadata.gz: ef50ae89f614aeec8385558444afb27f57aa1797e415ce25361ec57dec5d87ecf1181fdaf9e59bf0e696cc4ec8d68c0aa9c641bb30b1c01c08a72dd5792d0e6e
+  data.tar.gz: e9049722257ccdc3415ec4e33af0100d692dbf85806cd2c1fb1f4094c1189364d575943e0f39907bcc295ec7fecabfd8065aae57c0857997a22dd0904ea958ea

data/CHANGELOG.md

@@ -2,6 +2,27 @@
 
 ## [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

data/Gemfile

@@ -49,6 +49,5 @@ group :test do
   gem 'rubocop'
   gem 'rubocop-legion'
   gem 'rubocop-rspec'
-  gem 'ruby_llm'
   gem 'simplecov'
 end

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

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

@@ -1,6 +1,5 @@
 # frozen_string_literal: true
 
-require 'ruby_llm'
 require 'net/http'
 require 'json'
 
@@ -14,20 +13,25 @@ module Legion
   module CLI
     class Chat
       module Tools
-        class CostSummary < RubyLLM::Tool
+        class CostSummary < Legion::Tools::Base
+          tool_name 'legion.cost_summary'
           description 'Get cost and token usage summary from the running Legion daemon. Shows spending ' \
                       'for today, this week, and this month, plus top cost consumers by worker. ' \
                       'Use this to monitor LLM spending and identify expensive operations.'
-          param :action, type:     'string',
-                         desc:     'Action: "summary" (default), "top" (top consumers), or "worker" (specific worker)',
-                         required: false
-          param :worker_id, type: 'string', desc: 'Worker ID (required for "worker" action)', required: false
-          param :limit, type: 'integer', desc: 'Number of top consumers to show (default: 5)', required: false
+          input_schema({
+                         type:       'object',
+                         properties: {
+                           action:    { type: 'string', description: 'Action: "summary" (default), "top" (top consumers), or "worker" (specific worker)' },
+                           worker_id: { type: 'string', description: 'Worker ID (required for "worker" action)' },
+                           limit:     { type: 'integer', description: 'Number of top consumers to show (default: 5)' }
+                         },
+                         required:   []
+                       })
 
           DEFAULT_PORT = 4567
           DEFAULT_HOST = '127.0.0.1'
 
-          def execute(action: 'summary', worker_id: nil, limit: 5)
+          def self.call(action: 'summary', worker_id: nil, limit: 5)
             case action.to_s
             when 'top'
               handle_top(limit.to_i.clamp(1, 20))
@@ -45,9 +49,7 @@ module Legion
             "Error fetching cost data: #{e.message}"
           end
 
-          private
-
-          def handle_summary
+          def self.handle_summary
             data = api_get('/api/costs/summary?period=month')
             return "API error: #{data[:error]}" if data[:error]
 
@@ -60,7 +62,7 @@ module Legion
             lines.join("\n")
           end
 
-          def handle_top(limit)
+          def self.handle_top(limit)
             data = api_get('/api/workers')
             return "API error: #{data[:error]}" if data[:error]
 
@@ -77,7 +79,7 @@ module Legion
             lines.join("\n")
           end
 
-          def handle_worker(worker_id)
+          def self.handle_worker(worker_id)
             data = api_get("/api/workers/#{worker_id}/value")
             return "API error: #{data[:error]}" if data[:error]
 
@@ -91,7 +93,7 @@ module Legion
             lines.join("\n")
           end
 
-          def fetch_worker_cost(worker_id)
+          def self.fetch_worker_cost(worker_id)
             data = api_get("/api/workers/#{worker_id}/value")
             data = data[:data] || data
             (data[:total_cost_usd] || 0).to_f
@@ -99,7 +101,7 @@ module Legion
             0.0
           end
 
-          def api_get(path)
+          def self.api_get(path)
             uri = URI("http://#{DEFAULT_HOST}:#{api_port}#{path}")
             http = Net::HTTP.new(uri.host, uri.port)
             http.open_timeout = 2
@@ -108,7 +110,7 @@ module Legion
             ::JSON.parse(response.body, symbolize_names: true)
           end
 
-          def api_port
+          def self.api_port
             return DEFAULT_PORT unless defined?(Legion::Settings)
 
             Legion::Settings[:api]&.dig(:port) || DEFAULT_PORT