Added Hamming and Jaccard distances

pgvector · May 17, 2024 · f6cff70 · f6cff70
1 parent 95cc23a
commit f6cff70
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,6 @@
 ## 0.2.0 (unreleased)
 
-- Added `L1` distance
+- Added `L1`, `Hamming`, and `Jaccard` distances
 - Changed `Distance` to enum
 - Dropped support for PHP < 8.1
 - Dropped support for Laravel < 10

diff --git a/src/laravel/Distance.php b/src/laravel/Distance.php
@@ -8,4 +8,6 @@ enum Distance
     case InnerProduct;
     case Cosine;
     case L1;
+    case Hamming;
+    case Jaccard;
 }
diff --git a/src/laravel/HasNeighbors.php b/src/laravel/HasNeighbors.php
@@ -22,13 +22,19 @@ public function scopeNearestNeighbors(Builder $query, string $column, mixed $val
             case Distance::L1:
                 $op = '<+>';
                 break;
+            case Distance::Hamming:
+                $op = '<~>';
+                break;
+            case Distance::Jaccard:
+                $op = '<%>';
+                break;
             default:
                 throw new \InvalidArgumentException("Invalid distance");
         }
         $wrapped = $query->getGrammar()->wrap($column);
         $order = "$wrapped $op ?";
         $neighborDistance = $distance == Distance::InnerProduct ? "($order) * -1" : $order;
-        $vector = $value instanceof Vector ? $value : new Vector($value);
+        $vector = is_array($value) ? new Vector($value) : $value;
 
         // ideally preserve existing select, but does not appear to be a way to get columns
         $query->select()