lex-apollo 0.3.4 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e7d785a9fed9656eb22b307620a384f64e52646bca14728dc0de3ef2b1adb5bd
-  data.tar.gz: c64bcf1b75ee0ed43a747722a610c5cee8de823995d5f7455e31ae5c9174c56d
+  metadata.gz: 6156122baf8989f96a09685918e8ed6f6608730c8e19e442a632f5faa8853ee7
+  data.tar.gz: 478d0003003a76abc6ffdc0c39a7be3b06d8e75f91e1c88a6edfa5e46a05634a
 SHA512:
-  metadata.gz: 0ff89d33e4993dd8c71e392202c66add619ab04045fa674b5f8f0c4d891b727ca7401865d349c68bdc3a0344e1773db4f86651ed7e96ecbcaf0098a9d05cdbc9
-  data.tar.gz: 29b69744b9870bee8e1fbfc030652aede63a51c3a12cd02a3bc69bfc805b6267cf8045346cb37dc2565e1e103ae6f3b559972536b02f3a62dd4b6f12223361a5
+  metadata.gz: 898147c2dbe69430a0dc4606db21a82100de37a1865b9d56157499530c7e7a205e8cc44cd39f23410293a8f3874c9dadacf9c5aa47c89bd5b61a7a39b9067951
+  data.tar.gz: 8ed9fe61c718b12abfd8b5b2c6a75e249e47491c6bf8068df984f6f18c8943ee5d3d56b3b2def551703b4cd7b0463942e20a4803a740ca7da65d25a87b90c344

data/lib/legion/extensions/apollo/runners/knowledge.rb

@@ -130,6 +130,39 @@ module Legion
             { success: false, error: e.message }
           end
 
+          def redistribute_knowledge(agent_id:, min_confidence: 0.5, **)
+            return { success: false, error: 'apollo_data_not_available' } unless defined?(Legion::Data::Model::ApolloEntry)
+
+            entries = Legion::Data::Model::ApolloEntry
+                      .where(source_agent: agent_id, status: 'confirmed')
+                      .where { confidence > min_confidence }
+                      .all
+
+            return { success: true, redistributed: 0 } if entries.empty?
+
+            store = (Legion::Extensions::Agentic::Memory::Trace.shared_store if defined?(Legion::Extensions::Agentic::Memory::Trace))
+
+            redistributed = 0
+            entries.each do |entry|
+              if store
+                trace = Legion::Extensions::Agentic::Memory::Trace::Helpers::Trace.new_trace(
+                  type:            :semantic,
+                  content_payload: { content: entry.content, source_agent: agent_id,
+                                     content_type: entry.content_type, tags: Array(entry.tags) },
+                  strength:        entry.confidence.to_f,
+                  domain_tag:      Array(entry.tags).first || 'general'
+                )
+                store.store(trace)
+              end
+              redistributed += 1
+            end
+
+            Legion::Logging.info "[apollo] redistributed #{redistributed} entries from departing agent=#{agent_id}"
+            { success: true, redistributed: redistributed, agent_id: agent_id }
+          rescue Sequel::Error => e
+            { success: false, error: e.message }
+          end
+
           def retrieve_relevant(query: nil, limit: 5, min_confidence: 0.3, tags: nil, skip: false, **)
             return { status: :skipped } if skip
 

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

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Apollo
-      VERSION = '0.3.4'
+      VERSION = '0.3.5'
     end
   end
 end

data/spec/legion/extensions/apollo/runners/knowledge_spec.rb

@@ -307,4 +307,88 @@ RSpec.describe Legion::Extensions::Apollo::Runners::Knowledge do
       end
     end
   end
+
+  describe '#redistribute_knowledge' do
+    let(:host) { Object.new.extend(described_class) }
+
+    context 'when Apollo data is not available' do
+      before { hide_const('Legion::Data::Model::ApolloEntry') if defined?(Legion::Data::Model::ApolloEntry) }
+
+      it 'returns a structured error' do
+        result = host.redistribute_knowledge(agent_id: 'agent-x')
+        expect(result[:success]).to be false
+        expect(result[:error]).to eq('apollo_data_not_available')
+      end
+    end
+
+    context 'when the departing agent has no confirmed entries' do
+      let(:mock_entry_class) { double('ApolloEntry') }
+
+      before do
+        stub_const('Legion::Data::Model::ApolloEntry', mock_entry_class)
+        chain = double('chain')
+        allow(mock_entry_class).to receive(:where).and_return(chain)
+        allow(chain).to receive(:where).and_return(double(all: []))
+      end
+
+      it 'returns success with zero redistributed' do
+        result = host.redistribute_knowledge(agent_id: 'departed-1')
+        expect(result[:success]).to be true
+        expect(result[:redistributed]).to eq(0)
+      end
+    end
+
+    context 'when the departing agent has confirmed entries' do
+      let(:mock_entry_class) { double('ApolloEntry') }
+      let(:mock_entry) do
+        double('entry', content: 'Ruby is fast', content_type: 'fact',
+                        confidence: 0.8, tags: ['ruby'])
+      end
+
+      before do
+        stub_const('Legion::Data::Model::ApolloEntry', mock_entry_class)
+        chain = double('chain')
+        allow(mock_entry_class).to receive(:where).and_return(chain)
+        allow(chain).to receive(:where).and_return(double(all: [mock_entry]))
+      end
+
+      it 'returns the count of redistributed entries' do
+        result = host.redistribute_knowledge(agent_id: 'departed-2')
+        expect(result[:success]).to be true
+        expect(result[:redistributed]).to eq(1)
+        expect(result[:agent_id]).to eq('departed-2')
+      end
+
+      it 'stores into trace shared_store when Memory::Trace is available' do
+        mock_store = double('store')
+        mock_trace_helpers = Module.new do
+          def self.new_trace(type:, content_payload: nil, **kwargs) # rubocop:disable Lint/UnusedMethodArgument
+            { trace_id: 'trace-abc', trace_type: type, strength: kwargs[:strength] || 0.5 }
+          end
+        end
+        stub_const('Legion::Extensions::Agentic::Memory::Trace', Module.new)
+        stub_const('Legion::Extensions::Agentic::Memory::Trace::Helpers::Trace', mock_trace_helpers)
+        allow(Legion::Extensions::Agentic::Memory::Trace).to receive(:shared_store).and_return(mock_store)
+        allow(mock_store).to receive(:store)
+
+        result = host.redistribute_knowledge(agent_id: 'departed-3')
+        expect(result[:redistributed]).to eq(1)
+        expect(mock_store).to have_received(:store).once
+      end
+    end
+
+    context 'when Sequel raises an error' do
+      before do
+        stub_const('Legion::Data::Model::ApolloEntry', Class.new)
+        allow(Legion::Data::Model::ApolloEntry).to receive(:where)
+          .and_raise(Sequel::Error, 'db error')
+      end
+
+      it 'returns a structured error' do
+        result = host.redistribute_knowledge(agent_id: 'agent-x')
+        expect(result[:success]).to be false
+        expect(result[:error]).to eq('db error')
+      end
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-apollo
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.3.5
 platform: ruby
 authors:
 - Esity