RCBC-489: Support for base64 encoded vector types (#146)

Author: DemetrisChr

ext/couchbase

@@ -1071,13 +1071,23 @@ class VectorQuery
 
     # Constructs a +VectorQuery+ instance
     #
-    # @param [String] vector_field_name the document field that contains the vector.
-    # @param [Array<Float>] vector_query the vector query to run.
+    # @overload initialize(vector_field_name, vector_query)
+    #   @param [String] vector_field_name the document field that contains the vector.
+    #   @param [Array<Float>] vector_query the vector query.
     #
-    # @yieldparam [MatchPhraseQuery] self
+    # @overload initialize(vector_field_name, base64_vector_query)
+    #   @param [String] vector_field_name the document field that contains the vector.
+    #   @param [String] base64_vector_query the vector query represented as a base64-encoded sequence of little-endian IEEE 754 floats.
+    #
+    # @yieldparam [VectorQuery] self
     def initialize(vector_field_name, vector_query)
       @vector_field_name = vector_field_name
-      @vector_query = vector_query
+
+      if vector_query.respond_to?(:to_str)
+        @base64_vector_query = vector_query.to_str
+      else
+        @vector_query = vector_query
+      end
 
       yield self if block_given?
     end
@@ -1092,6 +1102,7 @@ def to_h
       {
         field: @vector_field_name,
         vector: @vector_query,
+        vector_base64: @base64_vector_query,
         k: num_candidates || 3,
         boost: boost,
       }.compact

lib/couchbase/search_options.rb

@@ -237,6 +237,15 @@ def test_vector_search_query_defaults_to_match_none
       assert_equal SearchQuery.match_none.to_json, enc_query
     end
 
+    def test_vector_search_query_base64
+      base64_query = "aOeYBEXJ4kI="
+      enc_vector_query = VectorQuery.new("foo", base64_query).to_h
+
+      refute enc_vector_query.key?(:vector)
+      assert enc_vector_query.key?(:vector_base64)
+      assert_equal enc_vector_query[:vector_base64], base64_query
+    end
+
     def test_vector_search_not_supported
       skip("#{name}: Server supports vector search") if !env.protostellar? && env.server_version.supports_vector_search?