swarm_memory 2.2.2 → 2.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4216c2f9b10d9accb547d94709305359009d8b8a7440c3023c97d49cc66b5cd4
-  data.tar.gz: 670b1af5ec7880caa86fbceed37a030537cd863ec6a32155c144f706f1f96172
+  metadata.gz: 6217058c4241029650e057e4f28c75eb1c6105f797657eb7c039eb307fe9bcab
+  data.tar.gz: a9f5d039fe4dd776dff184fce3a539bf40e3bfe9f150698b26847e2953bc5e46
 SHA512:
-  metadata.gz: 747e338d2569036d590c783c674895fdb9420af52655e34a412605f4085ced7c0bc198bd70698df2c84689211846095e381622f6bbb139376c70e3d71b3e6871
-  data.tar.gz: bf9c87e64f5e149d7dbb00157abd6ba39f005e16cf29304382235c1ede975a13a76d1afc72ce12e83dfcec072610536a53e7abc457d8e0a7dff0061892e29759
+  metadata.gz: 883afdf0f09de8425226e1ef1138b9f657de2a92b9efbb2b2262315479c3bc2a83542151a73d484ee632d6e5e5921e1f2232d7361d6c55457d530d982569e681
+  data.tar.gz: 963e4ba618cde5bdeb496815044b9c791fb68c9fb28f7543fcba75d93f77d095638d05417338132143b57dc0c6b845b7fad4f8853b433b845545272c36aa2e37

data/lib/swarm_memory/core/semantic_index.rb

@@ -181,6 +181,9 @@ module SwarmMemory
 
       # Calculate hybrid scores combining semantic similarity and keyword matching
       #
+      # When keyword_score is 0 (no tag matches), falls back to pure semantic scoring
+      # to avoid penalizing results that have excellent semantic matches but no tag overlap.
+      #
       # @param results [Array<Hash>] Results with semantic :similarity scores
       # @param query_keywords [Array<String>] Keywords from query
       # @return [Array<Hash>] Results with updated :similarity (hybrid score) and debug info
@@ -189,8 +192,13 @@ module SwarmMemory
           semantic_score = result[:similarity]
           keyword_score = calculate_keyword_score(result, query_keywords)
 
-          # Hybrid score: weighted combination
-          hybrid_score = (@semantic_weight * semantic_score) + (@keyword_weight * keyword_score)
+          # Fallback to pure semantic when no keyword matches
+          # This prevents penalizing results with excellent semantic matches but no tag overlap
+          hybrid_score = if keyword_score.zero?
+            semantic_score
+          else
+            (@semantic_weight * semantic_score) + (@keyword_weight * keyword_score)
+          end
 
           # Update result with hybrid score and debug info
           result.merge(

data/lib/swarm_memory/core/storage.rb

@@ -18,7 +18,9 @@ module SwarmMemory
       #
       # @param adapter [Adapters::Base] Storage adapter
       # @param embedder [Embeddings::Embedder, nil] Optional embedder for semantic search
-      def initialize(adapter:, embedder: nil)
+      # @param semantic_weight [Float, nil] Weight for semantic similarity in hybrid search (0.0-1.0)
+      # @param keyword_weight [Float, nil] Weight for keyword matching in hybrid search (0.0-1.0)
+      def initialize(adapter:, embedder: nil, semantic_weight: nil, keyword_weight: nil)
         raise ArgumentError, "adapter is required" unless adapter.is_a?(Adapters::Base)
 
         @adapter = adapter
@@ -26,7 +28,10 @@ module SwarmMemory
 
         # Create semantic index if embedder is provided
         @semantic_index = if embedder
-          SemanticIndex.new(adapter: adapter, embedder: embedder)
+          index_options = { adapter: adapter, embedder: embedder }
+          index_options[:semantic_weight] = semantic_weight if semantic_weight
+          index_options[:keyword_weight] = keyword_weight if keyword_weight
+          SemanticIndex.new(**index_options)
         end
       end
 

data/lib/swarm_memory/dsl/memory_config.rb

@@ -84,6 +84,43 @@ module SwarmMemory
         @mode = value.to_sym
       end
 
+      # DSL method to set/get semantic weight for hybrid search
+      #
+      # Controls how much semantic (embedding) similarity affects search results.
+      # Default is 0.5 (50%). Set to 1.0 for pure semantic search.
+      #
+      # @param value [Float, nil] Weight between 0.0 and 1.0
+      # @return [Float, nil] Current semantic weight
+      #
+      # @example Pure semantic search (no keyword penalty)
+      #   semantic_weight 1.0
+      #   keyword_weight 0.0
+      def semantic_weight(value = nil)
+        if value.nil?
+          @adapter_options[:semantic_weight]
+        else
+          @adapter_options[:semantic_weight] = value.to_f
+        end
+      end
+
+      # DSL method to set/get keyword weight for hybrid search
+      #
+      # Controls how much keyword (tag) matching affects search results.
+      # Default is 0.5 (50%). Set to 0.0 to disable keyword matching.
+      #
+      # @param value [Float, nil] Weight between 0.0 and 1.0
+      # @return [Float, nil] Current keyword weight
+      #
+      # @example Disable keyword matching
+      #   keyword_weight 0.0
+      def keyword_weight(value = nil)
+        if value.nil?
+          @adapter_options[:keyword_weight]
+        else
+          @adapter_options[:keyword_weight] = value.to_f
+        end
+      end
+
       # Check if memory is enabled
       #
       # @return [Boolean] True if adapter is configured with required options

data/lib/swarm_memory/integration/sdk_plugin.rb

@@ -110,10 +110,12 @@ module SwarmMemory
           # MemoryConfig object (from DSL)
           [config.adapter_type, config.adapter_options]
         elsif config.is_a?(Hash)
-          # Hash (from YAML)
+          # Hash (from YAML) - symbolize keys for adapter compatibility
           adapter = (config[:adapter] || config["adapter"] || :filesystem).to_sym
-          options = config.reject { |k, _v| k == :adapter || k == "adapter" || k == :mode || k == "mode" }
-          [adapter, options]
+          options = config.reject { |k, _v| [:adapter, "adapter", :mode, "mode"].include?(k) }
+          # Symbolize keys so adapter receives keyword arguments correctly
+          symbolized_options = options.transform_keys { |k| k.to_s.to_sym }
+          [adapter, symbolized_options]
         else
           raise SwarmSDK::ConfigurationError, "Invalid memory configuration for #{agent_name}"
         end
@@ -125,7 +127,17 @@ module SwarmMemory
           raise SwarmSDK::ConfigurationError, "#{e.message} for agent #{agent_name}"
         end
 
-        # Instantiate adapter with options
+        # Extract hybrid search weights and other SDK-level config (before passing to adapter)
+        # Keys are already symbolized at this point
+        semantic_weight = adapter_options.delete(:semantic_weight)
+        keyword_weight = adapter_options.delete(:keyword_weight)
+
+        # Remove other SDK-level threshold configs that shouldn't go to adapter
+        adapter_options.delete(:discovery_threshold)
+        adapter_options.delete(:discovery_threshold_short)
+        adapter_options.delete(:adaptive_word_cutoff)
+
+        # Instantiate adapter with options (weights removed, adapter doesn't need them)
         # Note: Adapter is responsible for validating its own requirements
         begin
           adapter = adapter_class.new(**adapter_options)
@@ -137,8 +149,13 @@ module SwarmMemory
         # Create embedder for semantic search
         embedder = Embeddings::InformersEmbedder.new
 
-        # Create storage with embedder (enables semantic features)
-        Core::Storage.new(adapter: adapter, embedder: embedder)
+        # Create storage with embedder and hybrid search weights
+        Core::Storage.new(
+          adapter: adapter,
+          embedder: embedder,
+          semantic_weight: semantic_weight,
+          keyword_weight: keyword_weight,
+        )
       end
 
       # Parse memory configuration

data/lib/swarm_memory/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SwarmMemory
-  VERSION = "2.2.2"
+  VERSION = "2.2.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: swarm_memory
 version: !ruby/object:Gem::Version
-  version: 2.2.2
+  version: 2.2.3
 platform: ruby
 authors:
 - Paulo Arruda