lex-swarm 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b72b8c33768eb24380924cb9f33a342dd5e0e07fa1daaf151089de1d38e141e
-  data.tar.gz: 790c2c25675b41bb049410a15f21585fc361a00963d771af73752b612c2d15e9
+  metadata.gz: 57a1719c0d8b1b40310d6d060c85d6a1e3f7ed54cd35b6de22eae1a63bd79361
+  data.tar.gz: ed02e732a76958b725dc538bd66e9a1c1605f3a5343f725262e61415e6def46e
 SHA512:
-  metadata.gz: 9ca0b32d4b9c8af87086c777c71822c32228b195e0e1c7cb825c89c9a883dd97cee6e8815e228ca4d4a8a07ce1d31813eca91e09191405f06b45c9ed0c347453
-  data.tar.gz: 6765b85f142bebdb96f537860d4984e56c762551d87da9b6fd3ab0a3b028c6f6fc3236e46b50bf445a020bf1d570c6fae3791af408c6cdc77d6ea77f68382c64
+  metadata.gz: 98f84a0186ecfcaa0ba734e506f5d71f604759012aec2104490d3d5b2dd74f9ebe28e1c987ea8a4752e298a581f6584990ce830e1c575696f68c783a265fdd31
+  data.tar.gz: f1edc98f2e926bb94998fd4cce549d4d817d44b2a52e7466552324298ef186e8b1366cd27f4eb5bf0c8344ce9c9de0dca7091d4c25fb4df13a0a178ac964ed8e

data/Gemfile

@@ -8,4 +8,4 @@ gem 'rspec', '~> 3.13'
 gem 'rubocop', '~> 1.75', require: false
 gem 'rubocop-rspec', require: false
 
-gem 'legion-gaia', path: '../../legion-gaia'
+gem 'legion-gaia'

data/lib/legion/extensions/swarm/actors/orphan_sweeper.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Swarm
+      module Actor
+        class OrphanSweeper < Legion::Extensions::Actors::Every
+          def time = 300
+          def run_now? = false
+          def use_runner? = false
+          def check_subtask? = false
+          def generate_task? = false
+
+          def action(**_opts)
+            return { swept: 0 } unless defined?(Legion::Data)
+
+            tasks = Legion::Data.connection[:tasks]
+            orphans = tasks.exclude(parent_id: nil)
+                           .exclude(status: %w[complete failed])
+                           .all
+                           .select do |t|
+              parent = tasks.where(id: t[:parent_id]).first
+              parent.nil? || %w[complete failed].include?(parent[:status])
+            end
+
+            orphans.each do |orphan|
+              tasks.where(id: orphan[:id]).update(status: 'failed')
+            end
+
+            { swept: orphans.size }
+          rescue StandardError => e
+            { swept: 0, error: e.message }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/swarm/helpers/sub_agent.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Swarm
+      module Helpers
+        module SubAgent
+          def spawn(runner:, function:, payload:, parent_task_id:, **)
+            return { success: false, reason: :data_unavailable } unless defined?(Legion::Data::Model::Task)
+
+            parent = Legion::Data::Model::Task[parent_task_id]
+            return { success: false, reason: :parent_not_found } unless parent
+
+            new_depth = (parent.respond_to?(:depth) ? parent.depth.to_i : 0) + 1
+
+            return { success: false, reason: :depth_exceeded } if new_depth > max_depth
+            return { success: false, reason: :concurrent_exceeded } if concurrent_children_count > max_concurrent
+            return { success: false, reason: :per_parent_exceeded } if children_of(parent_task_id) > max_per_parent
+
+            if defined?(Legion::Ingress)
+              result = Legion::Ingress.run(
+                payload:       payload,
+                runner_class:  runner,
+                function:      function,
+                source:        'sub_agent',
+                generate_task: true,
+                check_subtask: false
+              )
+              { success: true, task_id: result[:task_id], depth: new_depth }
+            else
+              { success: false, reason: :ingress_unavailable }
+            end
+          rescue StandardError => e
+            { success: false, reason: :error, message: e.message }
+          end
+
+          private
+
+          def max_depth
+            Legion::Settings.dig(:swarm, :max_depth) || 3
+          rescue StandardError
+            3
+          end
+
+          def max_concurrent
+            Legion::Settings.dig(:swarm, :max_concurrent) || 20
+          rescue StandardError
+            20
+          end
+
+          def max_per_parent
+            Legion::Settings.dig(:swarm, :max_per_parent) || 10
+          rescue StandardError
+            10
+          end
+
+          def concurrent_children_count
+            return 0 unless defined?(Legion::Data)
+
+            Legion::Data.connection[:tasks]
+                        .exclude(parent_id: nil)
+                        .exclude(status: %w[complete failed])
+                        .count
+          rescue StandardError
+            0
+          end
+
+          def children_of(parent_task_id)
+            return 0 unless defined?(Legion::Data)
+
+            Legion::Data.connection[:tasks]
+                        .where(parent_id: parent_task_id)
+                        .exclude(status: %w[complete failed])
+                        .count
+          rescue StandardError
+            0
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/swarm/runners/spawn_child.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Swarm
+      module Runners
+        module SpawnChild
+          include Helpers::SubAgent if defined?(Helpers::SubAgent)
+
+          def spawn_child(runner:, function:, payload:, parent_task_id:, **)
+            spawn(runner: runner, function: function, payload: payload, parent_task_id: parent_task_id)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/swarm/version.rb

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Swarm
-      VERSION = '0.1.1'
+      VERSION = '0.1.2'
     end
   end
 end

data/lib/legion/extensions/swarm.rb

@@ -3,7 +3,9 @@
 require 'legion/extensions/swarm/version'
 require 'legion/extensions/swarm/helpers/charter'
 require 'legion/extensions/swarm/helpers/swarm_store'
+require 'legion/extensions/swarm/helpers/sub_agent'
 require 'legion/extensions/swarm/runners/swarm'
+require 'legion/extensions/swarm/runners/spawn_child'
 
 module Legion
   module Extensions

data/spec/legion/extensions/swarm/actors/orphan_sweeper_spec.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Actors
+      unless defined?(Every)
+        class Every # rubocop:disable Lint/EmptyClass
+        end
+      end
+    end
+  end
+end
+
+$LOADED_FEATURES << 'legion/extensions/actors/every'
+
+require_relative '../../../../../lib/legion/extensions/swarm/actors/orphan_sweeper'
+
+RSpec.describe Legion::Extensions::Swarm::Actor::OrphanSweeper do
+  subject(:actor) { described_class.new }
+
+  describe '#time' do
+    it 'returns 300 seconds' do
+      expect(actor.time).to eq(300)
+    end
+  end
+
+  describe '#run_now?' do
+    it 'returns false' do
+      expect(actor.run_now?).to be false
+    end
+  end
+
+  describe '#action' do
+    context 'when Legion::Data is not available' do
+      it 'returns zero swept' do
+        result = actor.action
+        expect(result[:swept]).to eq(0)
+      end
+    end
+  end
+end

data/spec/legion/extensions/swarm/helpers/sub_agent_spec.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Legion::Extensions::Swarm::Helpers::SubAgent do
+  let(:helper_instance) { Class.new { include Legion::Extensions::Swarm::Helpers::SubAgent }.new }
+
+  before do
+    stub_const('Legion::Settings', Module.new do
+      def self.dig(*keys)
+        case keys
+        when %i[swarm max_depth] then 3
+        when %i[swarm max_concurrent] then 20
+        when %i[swarm max_per_parent] then 10
+        end
+      end
+    end)
+  end
+
+  describe '#max_depth' do
+    it 'returns configured max depth' do
+      expect(helper_instance.send(:max_depth)).to eq(3)
+    end
+  end
+
+  describe '#max_concurrent' do
+    it 'returns configured max concurrent' do
+      expect(helper_instance.send(:max_concurrent)).to eq(20)
+    end
+  end
+
+  describe '#max_per_parent' do
+    it 'returns configured max per parent' do
+      expect(helper_instance.send(:max_per_parent)).to eq(10)
+    end
+  end
+
+  describe '#spawn' do
+    context 'when legion-data is not available' do
+      it 'returns failure' do
+        result = helper_instance.spawn(runner: 'SomeRunner', function: 'do_thing',
+                                       payload: {}, parent_task_id: 1)
+        expect(result[:success]).to be false
+        expect(result[:reason]).to eq(:data_unavailable)
+      end
+    end
+
+    context 'when depth would be exceeded' do
+      let(:parent_task) { double('Task', depth: 3, id: 1) }
+
+      before do
+        pt = parent_task
+        task_model = Class.new do
+          define_method(:[]) { |_id| pt }
+        end.new
+        stub_const('Legion::Data::Model::Task', task_model)
+      end
+
+      it 'returns depth_exceeded' do
+        result = helper_instance.spawn(runner: 'SomeRunner', function: 'do_thing',
+                                       payload: {}, parent_task_id: 1)
+        expect(result[:success]).to be false
+        expect(result[:reason]).to eq(:depth_exceeded)
+      end
+    end
+  end
+end

data/spec/legion/extensions/swarm/runners/spawn_child_spec.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Legion::Extensions::Swarm::Runners::SpawnChild do
+  let(:runner_instance) { Class.new { include Legion::Extensions::Swarm::Runners::SpawnChild }.new }
+
+  before do
+    stub_const('Legion::Settings', Module.new do
+      def self.dig(*_keys) = nil
+    end)
+  end
+
+  describe '#spawn_child' do
+    it 'delegates to SubAgent.spawn and returns result' do
+      result = runner_instance.spawn_child(runner: 'R', function: 'f',
+                                           payload: {}, parent_task_id: 1)
+      expect(result).to be_a(Hash)
+      expect(result).to have_key(:success)
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-swarm
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Esity
@@ -33,16 +33,22 @@ files:
 - Gemfile
 - lex-swarm.gemspec
 - lib/legion/extensions/swarm.rb
+- lib/legion/extensions/swarm/actors/orphan_sweeper.rb
 - lib/legion/extensions/swarm/actors/stale_check.rb
 - lib/legion/extensions/swarm/client.rb
 - lib/legion/extensions/swarm/helpers/charter.rb
+- lib/legion/extensions/swarm/helpers/sub_agent.rb
 - lib/legion/extensions/swarm/helpers/swarm_store.rb
+- lib/legion/extensions/swarm/runners/spawn_child.rb
 - lib/legion/extensions/swarm/runners/swarm.rb
 - lib/legion/extensions/swarm/version.rb
+- spec/legion/extensions/swarm/actors/orphan_sweeper_spec.rb
 - spec/legion/extensions/swarm/actors/stale_check_spec.rb
 - spec/legion/extensions/swarm/client_spec.rb
 - spec/legion/extensions/swarm/helpers/charter_spec.rb
+- spec/legion/extensions/swarm/helpers/sub_agent_spec.rb
 - spec/legion/extensions/swarm/helpers/swarm_store_spec.rb
+- spec/legion/extensions/swarm/runners/spawn_child_spec.rb
 - spec/legion/extensions/swarm/runners/swarm_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/LegionIO/lex-swarm