lex-pilot-knowledge-assist 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 45d9ed95157a12405daa38f237db9b6a58f80a496202e54d2f0562fc1867d9e7
-  data.tar.gz: d0737978bd94e8bef510663cf82c13fde3b0a239ffcbf6f6e0e0e69fa8f0239b
+  metadata.gz: 9f398e78d2e02cceb805989cab08ce330e62fcaa458afa2c07d864553789b708
+  data.tar.gz: 1a677c86d87bbe95617c95d55dfe20ce20aeec0452f9adef8a723200576e77d7
 SHA512:
-  metadata.gz: a63ee4128a40e890eb4adbac50063a1c5f196238ed4e6e15f8babc7134675f019e228ff66cc72e63da4f000cad5a2782691e44ffce778060005e35d44e8b9472
-  data.tar.gz: '018f5c696c16ec5f683c31d531acaf902723aa03c05cf2fe1cf8eea964aa34478aa135cee95bf8ba00bbbd627b9f989948b02e4f7c79f6aecc38d3d5bcd696ed'
+  metadata.gz: c328df777ead50e852bc92486d54193d56eea53526cc03ad9fec78556c75aad8bbd11deb0b37fc4b25fd0955f025cf170a414f0fbf9e5b132e8bec5355d51dbe
+  data.tar.gz: 45acdab5dc6c139132499fa24c5ea48e10be2fa93b1bef3972175f8badc07d4a6115e38bd05b502a5771aac253d32f4f8338f9b7d23b107a12ca241f9ef6e859

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

@@ -13,12 +13,21 @@ module Legion
               question: question,
               answer: answer,
               sources: context.map { |c| c[:id] },
-              confidence: context.any? ? 0.8 : 0.3
+              confidence: derive_confidence(context)
             }
           end
 
           private
 
+          def derive_confidence(context)
+            return 0.3 if context.empty?
+
+            scores = context.filter_map { |c| c[:confidence] }
+            return 0.6 if scores.empty?
+
+            scores.max.clamp(0.1, 1.0)
+          end
+
           def retrieve_context(question, agent_id)
             return [] unless defined?(Legion::Extensions::Apollo::Client)
 
@@ -38,7 +47,7 @@ module Legion
                        "Context:\n#{context_text}\n\nQuestion: #{question}"
                      end
 
-            result = Legion::LLM.chat(message: prompt)
+            result = Legion::LLM.chat(message: prompt, caller: { extension: 'lex-pilot-knowledge-assist' })
             result[:content]
           rescue StandardError => e
             "Error: #{e.message}"

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

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module PilotKnowledgeAssist
-      VERSION = '0.1.1'
+      VERSION = '0.1.3'
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-pilot-knowledge-assist
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Esity