Added cast method for vector and halfvec

ankane · May 3, 2024 · 5f0720b · 5f0720b
1 parent 052204e
commit 5f0720b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/neighbor/type/halfvec.rb b/lib/neighbor/type/halfvec.rb
@@ -5,6 +5,14 @@ def type
         :halfvec
       end
 
+      def cast(value)
+        if value.is_a?(Array)
+          "[#{value.join(",")}]"
+        else
+          super
+        end
+      end
+
       def deserialize(value)
         value[1..-1].split(",").map(&:to_f) unless value.nil?
       end

diff --git a/lib/neighbor/type/vector.rb b/lib/neighbor/type/vector.rb
@@ -5,6 +5,14 @@ def type
         :vector
       end
 
+      def cast(value)
+        if value.is_a?(Array)
+          "[#{value.join(",")}]"
+        else
+          super
+        end
+      end
+
       def deserialize(value)
         value[1..-1].split(",").map(&:to_f) unless value.nil?
       end