whoosh 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 03b756c310f71435839a3628bfdfa9499bb70869b9475e75eb6cce861caa9583
-  data.tar.gz: a848d792014e80cc6bee413ad14277881895a62a40854644f3a088bfe0ea164a
+  metadata.gz: 133af39d011ac77fc6e0773ea19652cb990f5c8d129d40b7f301796062e23e1f
+  data.tar.gz: 5145a7c3177f2ddef7667f8faa6a6c2681b38fd279351c95ceda93ce69999c2a
 SHA512:
-  metadata.gz: b764fa79a9e8297c66e0ce0c56be1127c1d1780ccb3840a0868a55841c4cb1108efffeff90cdaf4105a7df9259764aff7a8afe1047a42809e0c039b1e74be11d
-  data.tar.gz: 297cc80e8ca6ba22df946e07204f114bda7634480f5af72520caa0bdfde7b8604e6b66bf1b081f3527bede997a320b9ffc1b44c14fba21c2a05e5778101451c5
+  metadata.gz: 4e99f75634ea05e7e2673164f6bdd8b71ea704e6ee02023370b8b77207666bee6291a02d7423d8d036c4014651ba17673c3f23639e64a2a68b5038799b1dca99
+  data.tar.gz: 4427cce1a58e9d35841a372ca782f831d6aa3ef513e2309f14e4f6b284a4d12461b102fe5a963768e1a01af2e98db7fc82f7084e970f1a150ef7092f2033c8bf

data/lib/whoosh/app.rb

@@ -29,6 +29,7 @@ module Whoosh
       auto_register_database
       auto_register_storage
       auto_register_http
+      auto_register_vectors
       auto_configure_jobs
       @metrics = Metrics.new
       auto_register_metrics
@@ -302,6 +303,10 @@ module Whoosh
       @di.provide(:http) { HTTP }
     end
 
+    def auto_register_vectors
+      @di.provide(:vectors) { VectorStore.build(@config.data) }
+    end
+
     def auto_configure_jobs
       backend = Jobs.build_backend(@config.data)
       Jobs.configure(backend: backend, di: @di)

data/lib/whoosh/cli/project_generator.rb

@@ -175,8 +175,10 @@ module Whoosh
               max_connections: 10
               log_level: debug
 
+            # Cache & Jobs auto-detect:
+            # No REDIS_URL → in-memory (just works)
+            # Set REDIS_URL → auto-switches to Redis
             cache:
-              store: memory
               default_ttl: 300
 
             jobs:
@@ -188,6 +190,12 @@ module Whoosh
               level: info
               format: json
 
+            # Vector store auto-detect:
+            # zvec gem installed → uses zvec, otherwise → in-memory
+            # vector:
+            #   adapter: auto
+            #   path: db/vectors
+
             docs:
               enabled: true
 

data/lib/whoosh/vector_store/memory_store.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+module Whoosh
+  module VectorStore
+    class MemoryStore
+      def initialize
+        @collections = {}
+        @mutex = Mutex.new
+      end
+
+      # Store a vector with metadata
+      def insert(collection, id:, vector:, metadata: {})
+        @mutex.synchronize do
+          @collections[collection] ||= {}
+          @collections[collection][id] = { vector: vector, metadata: metadata }
+        end
+      end
+
+      # Search by cosine similarity, return top-k results
+      def search(collection, vector:, limit: 10)
+        @mutex.synchronize do
+          items = @collections[collection]
+          return [] unless items && !items.empty?
+
+          scored = items.map do |id, data|
+            score = cosine_similarity(vector, data[:vector])
+            { id: id, score: score, metadata: data[:metadata] }
+          end
+
+          scored.sort_by { |r| -r[:score] }.first(limit)
+        end
+      end
+
+      # Delete a vector
+      def delete(collection, id:)
+        @mutex.synchronize do
+          @collections[collection]&.delete(id)
+        end
+      end
+
+      # Count vectors in a collection
+      def count(collection)
+        @mutex.synchronize do
+          @collections[collection]&.size || 0
+        end
+      end
+
+      # Drop a collection
+      def drop(collection)
+        @mutex.synchronize do
+          @collections.delete(collection)
+        end
+      end
+
+      def close
+        # No-op
+      end
+
+      private
+
+      def cosine_similarity(a, b)
+        return 0.0 if a.empty? || b.empty? || a.length != b.length
+
+        dot = 0.0
+        mag_a = 0.0
+        mag_b = 0.0
+
+        a.length.times do |i|
+          dot += a[i] * b[i]
+          mag_a += a[i] * a[i]
+          mag_b += b[i] * b[i]
+        end
+
+        denom = Math.sqrt(mag_a) * Math.sqrt(mag_b)
+        denom.zero? ? 0.0 : dot / denom
+      end
+    end
+  end
+end

data/lib/whoosh/vector_store.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module Whoosh
+  module VectorStore
+    autoload :MemoryStore, "whoosh/vector_store/memory_store"
+
+    # Auto-detect: zvec gem → use it, otherwise → in-memory
+    def self.build(config_data = {})
+      vector_config = config_data["vector"] || {}
+      adapter = vector_config["adapter"] || "auto"
+
+      case adapter
+      when "auto"
+        # Try zvec first, fall back to memory
+        if zvec_available?
+          require "whoosh/vector_store/zvec_store"
+          ZvecStore.new(**zvec_options(vector_config))
+        else
+          MemoryStore.new
+        end
+      when "memory"
+        MemoryStore.new
+      when "zvec"
+        require "whoosh/vector_store/zvec_store"
+        ZvecStore.new(**zvec_options(vector_config))
+      else
+        MemoryStore.new
+      end
+    end
+
+    def self.zvec_available?
+      require "zvec"
+      true
+    rescue LoadError
+      false
+    end
+
+    def self.zvec_options(config)
+      { path: config["path"] || "db/vectors" }
+    end
+  end
+end

data/lib/whoosh/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Whoosh
-  VERSION = "1.1.0"
+  VERSION = "1.2.0"
 end

data/lib/whoosh.rb

@@ -28,6 +28,7 @@ module Whoosh
   autoload :Jobs,                "whoosh/jobs"
   autoload :Metrics,             "whoosh/metrics"
   autoload :Paginate,            "whoosh/paginate"
+  autoload :VectorStore,         "whoosh/vector_store"
 
   module Auth
     autoload :ApiKey,         "whoosh/auth/api_key"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: whoosh
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Johannes Dwi Cahyo
@@ -241,6 +241,8 @@ files:
 - lib/whoosh/test.rb
 - lib/whoosh/types.rb
 - lib/whoosh/uploaded_file.rb
+- lib/whoosh/vector_store.rb
+- lib/whoosh/vector_store/memory_store.rb
 - lib/whoosh/version.rb
 homepage: https://github.com/johannesdwicahyo/whoosh
 licenses: