fix: change <-> to <=> in the SELECT query (#804)

Co-authored-by: Alex Yang <[email protected]>
run-llama · May 3, 2024 · e74fe88 · e74fe88
1 parent f1862cc
commit e74fe88
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.changeset/silly-horses-carry.md b/.changeset/silly-horses-carry.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+fix: change <-> to <=> in the SELECT query
diff --git a/packages/core/src/storage/vectorStore/PGVectorStore.ts b/packages/core/src/storage/vectorStore/PGVectorStore.ts
@@ -250,7 +250,7 @@ export class PGVectorStore implements VectorStore {
     options?: any,
   ): Promise<VectorStoreQueryResult> {
     // TODO QUERY TYPES:
-    //    Distance:       SELECT embedding <-> $1 AS distance FROM items;
+    //    Distance:       SELECT embedding <=> $1 AS distance FROM items;
     //    Inner Product:  SELECT (embedding <#> $1) * -1 AS inner_product FROM items;
     //    Cosine Sim:     SELECT 1 - (embedding <=> $1) AS cosine_similarity FROM items;
 
@@ -273,7 +273,7 @@ export class PGVectorStore implements VectorStore {
 
     const sql = `SELECT 
         v.*, 
-        embeddings <-> $1 s 
+        embeddings <=> $1 s 
       FROM ${this.schemaName}.${this.tableName} v
       ${where}
       ORDER BY s