lex-pilot-knowledge-assist 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 91485d59b7e75c54a9286e9d00980805194dd57211a7773a3dba1dbbcac1dcce
+  data.tar.gz: e37c2062983e8884623cbce58a2a7ba706349c8fcd37501f7a4e039a13d0c109
+SHA512:
+  metadata.gz: 74ad9e20db90b0e32ce186b50a3f271284ab3115ebe2b9ef5b6d5d41c8b233bc6c3c178cdeac4031e593d3238fd607fd104f5913ed547df3ffd2b2dce526a928
+  data.tar.gz: 84046dbe69849dabe06e924534a72fd115a1438b3d2953daa231cd3b1d7c41992be93ae20c0916086a07c6d5185484581e48f6e12c22c7a127fb1f87e617cc4a

data/lib/legion/extensions/pilot_knowledge_assist/runners/assistant.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module PilotKnowledgeAssist
+      module Runners
+        module Assistant
+          def answer_question(question:, agent_id: 'knowledge-assist')
+            context = retrieve_context(question, agent_id)
+            answer = generate_answer(question, context)
+
+            {
+              question: question,
+              answer: answer,
+              sources: context.map { |c| c[:id] },
+              confidence: context.any? ? 0.8 : 0.3
+            }
+          end
+
+          private
+
+          def retrieve_context(question, agent_id)
+            return [] unless defined?(Legion::Extensions::Apollo::Client)
+
+            client = Legion::Extensions::Apollo::Client.new(agent_id: agent_id)
+            client.query_knowledge(query: question, limit: 5)
+          rescue StandardError
+            []
+          end
+
+          def generate_answer(question, context)
+            return 'LLM unavailable' unless defined?(Legion::LLM)
+
+            context_text = context.map { |c| c[:content] }.join("\n\n")
+            prompt = if context_text.empty?
+                       question
+                     else
+                       "Context:\n#{context_text}\n\nQuestion: #{question}"
+                     end
+
+            result = Legion::LLM.chat(message: prompt)
+            result[:content]
+          rescue StandardError => e
+            "Error: #{e.message}"
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module PilotKnowledgeAssist
+      VERSION = '0.1.0'
+    end
+  end
+end

data/lib/legion/extensions/pilot_knowledge_assist.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+require_relative 'pilot_knowledge_assist/version'
+require_relative 'pilot_knowledge_assist/runners/assistant'
+
+module Legion
+  module Extensions
+    module PilotKnowledgeAssist
+    end
+  end
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: lex-pilot-knowledge-assist
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Esity
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies: []
+description: RAG-based knowledge assistant using Apollo and LLM for LegionIO
+email:
+- matthewdiverson@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/legion/extensions/pilot_knowledge_assist.rb
+- lib/legion/extensions/pilot_knowledge_assist/runners/assistant.rb
+- lib/legion/extensions/pilot_knowledge_assist/version.rb
+homepage: https://github.com/LegionIO
+licenses:
+- MIT
+metadata:
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: LEX::PilotKnowledgeAssist
+test_files: []