lex-knowledge 0.6.14 → 0.6.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 21c773508b11dfb12b4d07bf2e37705483b82f4100382c29d84d883f3790ce5c
-  data.tar.gz: 96c6e3da83bb20eabb5db9f55b8ef4b7715ac08d3110ec4c6680ccf6b62210e4
+  metadata.gz: 28e36fec6a26f06e680ef33e0cbf8bf52d49762c1996d9257883b8b26524fab8
+  data.tar.gz: 2097de8337c7b06c8ac4a4bc1ebca123e642900991a2219e4e9adb31eca0d601
 SHA512:
-  metadata.gz: 5e2a5993ca4b4213c1dfb25eff797aaf7b89af3b5bc4c37072683d564bc49db49aad6f30702f47c19bd66c008bcdcc9b9a29b075920b66c03ab237dd4e9070e0
-  data.tar.gz: 04f3d33195e5c81c4b48c361170909b6f8a53e28660a0d30370584453a847f8d787443261ffa1b47a383c58aa387281616a29f1a2366c6c46e43308d82284808
+  metadata.gz: 5ed0345d3cb7c5cd8f52c0f2fa5fb3782f2d251a4a522709a7e5a0576ca5de4ca6b8d15897a49167caa6fe6629c81751ffaf84122a3534374edc99ab7a4b71ea
+  data.tar.gz: c684f6f7842c465b7e9c3879a6c6ba8b545ec4dc8787a87945098f7df0e9eb50fa8fc0ced89b706080a4cab64c0b23483ebc8369d13d1887ab0a75312ec91409

data/lib/legion/extensions/knowledge/actors/maintenance_runner.rb

@@ -21,16 +21,20 @@ module Legion
           end
 
           def enabled? # rubocop:disable Legion/Extension/ActorEnabledSideEffects
-            return false unless corpus_path && !corpus_path.empty?
+            return true if corpus_path && !corpus_path.empty?
 
-            true
+            Runners::Monitor.resolve_monitors.any?
           rescue StandardError => e
             handle_exception(e, level: :warn, operation: 'knowledge.maintenance_runner.enabled')
             false
           end
 
           def args
-            { path: corpus_path }
+            path = corpus_path
+            return { path: path } if path && !path.empty?
+
+            monitors = Runners::Monitor.resolve_monitors
+            monitors.any? ? { path: monitors.first[:path] } : { path: nil }
           end
 
           private

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

@@ -240,7 +240,7 @@ module Legion
           private_class_method :build_exists_map
 
           def llm_embed_available?
-            defined?(Legion::LLM) && Legion::LLM.respond_to?(:embed_batch)
+            defined?(Legion::LLM) && (Legion::LLM.respond_to?(:embed_batch) || Legion::LLM.respond_to?(:embed))
           end
           private_class_method :llm_embed_available?
 
@@ -250,9 +250,19 @@ module Legion
           private_class_method :paired_without_embed
 
           def build_embed_map(needs_embed)
-            results = Legion::LLM.embed_batch(needs_embed.map { |c| c[:content] }) # rubocop:disable Legion/HelperMigration/DirectLlm
-            results.each_with_object({}) do |r, h|
-              h[needs_embed[r[:index]][:content_hash]] = r[:vector] unless r[:error]
+            if Legion::LLM.respond_to?(:embed_batch)
+              results = Legion::LLM.embed_batch(needs_embed.map { |c| c[:content] }) # rubocop:disable Legion/HelperMigration/DirectLlm
+              results.each_with_object({}) do |r, h|
+                h[needs_embed[r[:index]][:content_hash]] = r[:vector] unless r[:error]
+              end
+            else
+              needs_embed.each_with_object({}) do |chunk, h|
+                result = Legion::LLM.embed(chunk[:content]) # rubocop:disable Legion/HelperMigration/DirectLlm
+                vector = result.is_a?(Hash) ? result[:vector] : result
+                h[chunk[:content_hash]] = vector if vector.is_a?(Array) && vector.any?
+              rescue StandardError => e
+                log.warn(e.message)
+              end
             end
           rescue StandardError => e
             handle_exception(e, level: :warn, operation: 'knowledge.ingest.build_embed_map')
@@ -328,10 +338,10 @@ module Legion
             return unless Legion::Apollo.respond_to?(:ingest) && Legion::Apollo.started?
 
             Legion::Apollo.ingest( # rubocop:disable Legion/HelperMigration/DirectKnowledge
-              content:      file_path,
-              content_type: 'document_retired',
+              content:      "Retired document: #{file_path}",
+              content_type: 'observation',
               tags:         [file_path, 'retired', 'document_chunk'].uniq,
-              metadata:     { source_file: file_path, retired: true }
+              context:      { source_file: file_path, retired: true }
             )
           rescue StandardError => e
             handle_exception(e, level: :warn, operation: 'knowledge.ingest.retire_file', file_path: file_path)

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

@@ -296,7 +296,7 @@ module Legion
           def query_count
             return 0 unless Helpers::ApolloModels.access_log_available?
 
-            Helpers::ApolloModels.access_log.where(action: 'knowledge_query').count
+            Helpers::ApolloModels.access_log.where(action: 'query').count
           rescue StandardError => e
             handle_exception(e, level: :warn, operation: 'knowledge.maintenance.query_count')
             0

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

@@ -181,11 +181,14 @@ module Legion
                          "Context:\n#{context_text}\n\nQuestion: #{question}\n\nAnswer:"
                      end
 
-            result = llm_chat(message: prompt, caller: { extension: 'lex-knowledge' })
-            result.is_a?(Hash) ? result[:content] : result
+            result = Legion::LLM.chat( # rubocop:disable Legion/HelperMigration/DirectLlm
+              message: prompt,
+              caller:  { extension: 'lex-knowledge' }
+            )
+            result.is_a?(Hash) ? result[:content] : result.content
           rescue StandardError => e
             handle_exception(e, level: :warn, operation: 'knowledge.query.synthesize_answer')
-            "Error generating answer: #{e.message}"
+            nil
           end
           private_class_method :synthesize_answer
 

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

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Knowledge
-      VERSION = '0.6.14'
+      VERSION = '0.6.15'
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-knowledge
 version: !ruby/object:Gem::Version
-  version: 0.6.14
+  version: 0.6.15
 platform: ruby
 authors:
 - Matthew Iverson