Added support for passing Vec<&Vector> with SQLx

pgvector · Mar 18, 2024 · 53330c1 · 53330c1
1 parent 33144d2
commit 53330c1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/sqlx_ext.rs b/src/sqlx_ext.rs
@@ -39,6 +39,12 @@ impl PgHasArrayType for Vector {
     }
 }
 
+impl PgHasArrayType for &Vector {
+    fn array_type_info() -> PgTypeInfo {
+        PgTypeInfo::with_name("_vector")
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use crate::Vector;
@@ -108,6 +114,12 @@ mod tests {
             .execute(&pool)
             .await?;
 
+        let vecs = vec![&vec, &vec2];
+        sqlx::query("INSERT INTO sqlx_items (factors) VALUES ($1)")
+            .bind(&vecs)
+            .execute(&pool)
+            .await?;
+
         let vecs = vec![vec, vec2];
         sqlx::query("INSERT INTO sqlx_items (factors) VALUES ($1)")
             .bind(&vecs)