legionio 1.4.50 → 1.4.51

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9f4884c77ed8ea44e74bdc3c9860dac014693cc90a3029fb06f1bc9e5a7edc27
-  data.tar.gz: 8da5570f871da743755e0c1c0fc1f81a8a5085061a6e27b5a8d9c2befe2400e0
+  metadata.gz: 1a711d9a3b6b942849185b3ec8eb16e4ecaa8b054c4c157799140f23d0bfc5f9
+  data.tar.gz: 64f138151ce16a625846a982fe66ab03d3e6403aba144509a62218a34c75fbae
 SHA512:
-  metadata.gz: be184813bae9bba912f9d190a06dbb9e3bc0a5b9c317019743fcaf1787de5c8aaad28fec85dcd14e1ffe27115cb80be3ea0970dd183cc002042aac1f2a12a7f1
-  data.tar.gz: e844ec3f09dd2f5bc6d8c908f884def2a6e4d71aa0ec5178d0e1ec1440c3a4cfe9d5d58996736acca92c67b7bced45ed8e8d2ffbc00a5f37a1ac4c6c4fb3ca37
+  metadata.gz: 9acd3652f07de505bfa8b3262c20b65715668eb7a2a012dcb1a222a27cf0b94d91facef2bd961468ee93cab3f89e5b06f100a6bf32eaf8dd2368da5c3979a675
+  data.tar.gz: ebd3a23ee70a99b2e86d709c931221262656f3eb3be4f638c20e4f8c24f7f8ad08c9dd0d54928be0f3f781725380ea298a13f93da05771a19b71fdd452333f7b

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Legion Changelog
 
+## [1.4.51] - 2026-03-17
+
+### Added
+- `Legion::TraceSearch`: natural language to safe JSON filter translation via legion-llm structured output
+- `legion trace search "query"`: CLI command for NL trace search
+- Column allowlist enforcement for query safety (no eval, JSON-only filter DSL)
+- Schema-aware prompt for metering_records table
+
 ## [1.4.50] - 2026-03-17
 
 ### Added

data/lib/legion/cli/trace_command.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'thor'
+
+module Legion
+  module CLI
+    class TraceCommand < Thor
+      namespace 'trace'
+
+      desc 'search QUERY', 'Search traces with natural language'
+      option :limit, type: :numeric, default: 50
+      def search(*query_parts)
+        require 'legion/trace_search'
+        query = query_parts.join(' ')
+        say "Searching: #{query}", :yellow
+
+        result = Legion::TraceSearch.search(query, limit: options[:limit])
+        if result[:error]
+          say "Error: #{result[:error]}", :red
+          return
+        end
+
+        say "Found #{result[:count]} results", :green
+        result[:results].first(20).each do |r|
+          say "  #{r[:created_at]} #{r[:extension]}.#{r[:runner_function]} #{r[:status]} $#{r[:cost_usd] || 0}"
+        end
+      end
+
+      default_task :search
+    end
+  end
+end

data/lib/legion/trace_search.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+module Legion
+  module TraceSearch
+    SCHEMA_CONTEXT = <<~PROMPT
+      You translate natural language queries into JSON filter objects for the metering_records table.
+
+      Columns: id (integer), worker_id (string), event_type (string), extension (string),
+      runner_function (string), status (string: success/failure), tokens_in (integer),
+      tokens_out (integer), cost_usd (float), wall_clock_ms (integer), created_at (datetime)
+
+      Return ONLY a valid JSON object with these possible keys:
+      - "where": hash of column => value filters (e.g. {"status": "failure"})
+      - "order": column name to sort by (prefix with "-" for descending, e.g. "-cost_usd")
+      - "limit": integer limit (default 50)
+      - "date_from": ISO date string for created_at >= filter
+      - "date_to": ISO date string for created_at <= filter
+
+      Examples:
+      - "failed tasks" => {"where": {"status": "failure"}}
+      - "most expensive calls" => {"order": "-cost_usd", "limit": 20}
+      - "tasks by worker-1 today" => {"where": {"worker_id": "worker-1"}, "date_from": "2026-03-16"}
+
+      Return ONLY the JSON object, no explanation.
+    PROMPT
+
+    FILTER_SCHEMA = {
+      type:       'object',
+      properties: {
+        where:     { type: 'object' },
+        order:     { type: 'string' },
+        limit:     { type: 'integer' },
+        date_from: { type: 'string' },
+        date_to:   { type: 'string' }
+      }
+    }.freeze
+
+    ALLOWED_COLUMNS = %w[
+      id worker_id event_type extension runner_function status
+      tokens_in tokens_out cost_usd wall_clock_ms created_at
+    ].freeze
+
+    class << self
+      def search(query, limit: 50)
+        parsed = generate_filter(query)
+        return { results: [], error: 'no filter generated' } unless parsed
+
+        execute_filter(parsed, limit)
+      rescue StandardError => e
+        { results: [], error: e.message }
+      end
+
+      def generate_filter(query)
+        return nil unless defined?(Legion::LLM) && Legion::LLM.respond_to?(:structured)
+
+        result = Legion::LLM.structured(
+          messages: [
+            { role: 'system', content: SCHEMA_CONTEXT },
+            { role: 'user',   content: query }
+          ],
+          schema:   FILTER_SCHEMA
+        )
+        result[:data] if result[:valid]
+      end
+
+      def execute_filter(parsed, default_limit)
+        return { results: [], error: 'data unavailable' } unless defined?(Legion::Data) && Legion::Data.respond_to?(:connection) && Legion::Data.connection
+
+        ds = Legion::Data.connection[:metering_records]
+
+        if parsed[:where].is_a?(Hash)
+          safe_where = parsed[:where].select { |k, _| ALLOWED_COLUMNS.include?(k.to_s) }
+          ds = ds.where(safe_where.transform_keys(&:to_sym))
+        end
+
+        ds = ds.where { created_at >= parsed[:date_from] } if parsed[:date_from]
+        ds = ds.where { created_at <= parsed[:date_to] } if parsed[:date_to]
+
+        if parsed[:order].is_a?(String)
+          col = parsed[:order].delete_prefix('-')
+          if ALLOWED_COLUMNS.include?(col)
+            ds = parsed[:order].start_with?('-') ? ds.order(Sequel.desc(col.to_sym)) : ds.order(col.to_sym)
+          end
+        end
+
+        limit = [parsed[:limit] || default_limit, 200].min
+        results = ds.limit(limit).all
+        { results: results, count: results.size, filter: parsed }
+      end
+    end
+  end
+end

data/lib/legion/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Legion
-  VERSION = '1.4.50'
+  VERSION = '1.4.51'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: legionio
 version: !ruby/object:Gem::Version
-  version: 1.4.50
+  version: 1.4.51
 platform: ruby
 authors:
 - Esity
@@ -474,6 +474,7 @@ files:
 - lib/legion/cli/telemetry_command.rb
 - lib/legion/cli/templates/core.json.erb
 - lib/legion/cli/theme.rb
+- lib/legion/cli/trace_command.rb
 - lib/legion/cli/trigger.rb
 - lib/legion/cli/update_command.rb
 - lib/legion/cli/version.rb
@@ -573,6 +574,7 @@ files:
 - lib/legion/telemetry.rb
 - lib/legion/tenant_context.rb
 - lib/legion/tenants.rb
+- lib/legion/trace_search.rb
 - lib/legion/version.rb
 - lib/legion/webhooks.rb
 - scripts/rollout-ci-workflow.sh