@vectorstores/supabase 0.1.4 → 0.1.6

package/dist/index.cjs

@@ -71,6 +71,26 @@ class SupabaseVectorStore extends core.BaseVectorStore {
    * @returns Object containing matched nodes, similarity scores, and document IDs
    * @throws Error if query embedding is not provided or if the query fails
    */ async query(query, options) {
+        switch(query.mode){
+            case "bm25":
+                return this.bm25Search(query);
+            case "hybrid":
+                {
+                    // Calculate prefetch limit for sub-searches
+                    const prefetchK = query.hybridPrefetch ?? query.similarityTopK * core.DEFAULT_HYBRID_PREFETCH_MULTIPLIER;
+                    const prefetchQuery = {
+                        ...query,
+                        similarityTopK: prefetchK
+                    };
+                    const vectorResult = await this.vectorSearch(prefetchQuery);
+                    const bm25Result = await this.bm25Search(prefetchQuery);
+                    return core.combineResults(vectorResult, bm25Result, query.alpha ?? 0.5, query.similarityTopK);
+                }
+            default:
+                return this.vectorSearch(query);
+        }
+    }
+    async vectorSearch(query) {
         if (!query.queryEmbedding) {
             throw new Error("Query embedding is required");
         }
@@ -104,6 +124,32 @@ class SupabaseVectorStore extends core.BaseVectorStore {
             ids: searchedEmbeddingResponses.map((item)=>item.id)
         };
     }
+    async bm25Search(query) {
+        if (!query.queryStr) {
+            throw new Error("Query string is required for BM25 search");
+        }
+        const { data, error } = await this.supabaseClient.from(this.table).select("*").textSearch("content", query.queryStr).limit(query.similarityTopK);
+        if (error) {
+            throw new Error(`Error querying vector store: ${JSON.stringify(error, null, 2)}`);
+        }
+        const nodes = (data || []).map((item)=>{
+            const node = core.metadataDictToNode(item.metadata ?? {}, {
+                fallback: {
+                    id: item.id,
+                    text: item.content,
+                    metadata: item.metadata
+                }
+            });
+            node.embedding = item.embedding;
+            node.setContent(item.content);
+            return node;
+        });
+        return {
+            nodes,
+            similarities: nodes.map(()=>1.0),
+            ids: (data || []).map((item)=>item.id)
+        };
+    }
     /**
    * Converts metadata filters to supabase query filter format
    * @param queryFilters - Metadata filters to convert

package/dist/index.d.cts

@@ -46,6 +46,8 @@ declare class SupabaseVectorStore extends BaseVectorStore {
      * @throws Error if query embedding is not provided or if the query fails
      */
     query(query: VectorStoreQuery, options?: object): Promise<VectorStoreQueryResult>;
+    private vectorSearch;
+    private bm25Search;
     /**
      * Converts metadata filters to supabase query filter format
      * @param queryFilters - Metadata filters to convert

package/dist/index.d.ts

@@ -46,6 +46,8 @@ declare class SupabaseVectorStore extends BaseVectorStore {
      * @throws Error if query embedding is not provided or if the query fails
      */
     query(query: VectorStoreQuery, options?: object): Promise<VectorStoreQueryResult>;
+    private vectorSearch;
+    private bm25Search;
     /**
      * Converts metadata filters to supabase query filter format
      * @param queryFilters - Metadata filters to convert

package/dist/index.edge-light.d.ts

@@ -46,6 +46,8 @@ declare class SupabaseVectorStore extends BaseVectorStore {
      * @throws Error if query embedding is not provided or if the query fails
      */
     query(query: VectorStoreQuery, options?: object): Promise<VectorStoreQueryResult>;
+    private vectorSearch;
+    private bm25Search;
     /**
      * Converts metadata filters to supabase query filter format
      * @param queryFilters - Metadata filters to convert

package/dist/index.edge-light.js

@@ -1,5 +1,5 @@
 import { createClient } from '@supabase/supabase-js';
-import { BaseVectorStore, nodeToMetadata, MetadataMode, metadataDictToNode } from '@vectorstores/core';
+import { BaseVectorStore, nodeToMetadata, MetadataMode, DEFAULT_HYBRID_PREFETCH_MULTIPLIER, combineResults, metadataDictToNode } from '@vectorstores/core';
 import { getEnv } from '@vectorstores/env';
 
 class SupabaseVectorStore extends BaseVectorStore {
@@ -69,6 +69,26 @@ class SupabaseVectorStore extends BaseVectorStore {
    * @returns Object containing matched nodes, similarity scores, and document IDs
    * @throws Error if query embedding is not provided or if the query fails
    */ async query(query, options) {
+        switch(query.mode){
+            case "bm25":
+                return this.bm25Search(query);
+            case "hybrid":
+                {
+                    // Calculate prefetch limit for sub-searches
+                    const prefetchK = query.hybridPrefetch ?? query.similarityTopK * DEFAULT_HYBRID_PREFETCH_MULTIPLIER;
+                    const prefetchQuery = {
+                        ...query,
+                        similarityTopK: prefetchK
+                    };
+                    const vectorResult = await this.vectorSearch(prefetchQuery);
+                    const bm25Result = await this.bm25Search(prefetchQuery);
+                    return combineResults(vectorResult, bm25Result, query.alpha ?? 0.5, query.similarityTopK);
+                }
+            default:
+                return this.vectorSearch(query);
+        }
+    }
+    async vectorSearch(query) {
         if (!query.queryEmbedding) {
             throw new Error("Query embedding is required");
         }
@@ -102,6 +122,32 @@ class SupabaseVectorStore extends BaseVectorStore {
             ids: searchedEmbeddingResponses.map((item)=>item.id)
         };
     }
+    async bm25Search(query) {
+        if (!query.queryStr) {
+            throw new Error("Query string is required for BM25 search");
+        }
+        const { data, error } = await this.supabaseClient.from(this.table).select("*").textSearch("content", query.queryStr).limit(query.similarityTopK);
+        if (error) {
+            throw new Error(`Error querying vector store: ${JSON.stringify(error, null, 2)}`);
+        }
+        const nodes = (data || []).map((item)=>{
+            const node = metadataDictToNode(item.metadata ?? {}, {
+                fallback: {
+                    id: item.id,
+                    text: item.content,
+                    metadata: item.metadata
+                }
+            });
+            node.embedding = item.embedding;
+            node.setContent(item.content);
+            return node;
+        });
+        return {
+            nodes,
+            similarities: nodes.map(()=>1.0),
+            ids: (data || []).map((item)=>item.id)
+        };
+    }
     /**
    * Converts metadata filters to supabase query filter format
    * @param queryFilters - Metadata filters to convert

package/dist/index.js

@@ -1,5 +1,5 @@
 import { createClient } from '@supabase/supabase-js';
-import { BaseVectorStore, nodeToMetadata, MetadataMode, metadataDictToNode } from '@vectorstores/core';
+import { BaseVectorStore, nodeToMetadata, MetadataMode, DEFAULT_HYBRID_PREFETCH_MULTIPLIER, combineResults, metadataDictToNode } from '@vectorstores/core';
 import { getEnv } from '@vectorstores/env';
 
 class SupabaseVectorStore extends BaseVectorStore {
@@ -69,6 +69,26 @@ class SupabaseVectorStore extends BaseVectorStore {
    * @returns Object containing matched nodes, similarity scores, and document IDs
    * @throws Error if query embedding is not provided or if the query fails
    */ async query(query, options) {
+        switch(query.mode){
+            case "bm25":
+                return this.bm25Search(query);
+            case "hybrid":
+                {
+                    // Calculate prefetch limit for sub-searches
+                    const prefetchK = query.hybridPrefetch ?? query.similarityTopK * DEFAULT_HYBRID_PREFETCH_MULTIPLIER;
+                    const prefetchQuery = {
+                        ...query,
+                        similarityTopK: prefetchK
+                    };
+                    const vectorResult = await this.vectorSearch(prefetchQuery);
+                    const bm25Result = await this.bm25Search(prefetchQuery);
+                    return combineResults(vectorResult, bm25Result, query.alpha ?? 0.5, query.similarityTopK);
+                }
+            default:
+                return this.vectorSearch(query);
+        }
+    }
+    async vectorSearch(query) {
         if (!query.queryEmbedding) {
             throw new Error("Query embedding is required");
         }
@@ -102,6 +122,32 @@ class SupabaseVectorStore extends BaseVectorStore {
             ids: searchedEmbeddingResponses.map((item)=>item.id)
         };
     }
+    async bm25Search(query) {
+        if (!query.queryStr) {
+            throw new Error("Query string is required for BM25 search");
+        }
+        const { data, error } = await this.supabaseClient.from(this.table).select("*").textSearch("content", query.queryStr).limit(query.similarityTopK);
+        if (error) {
+            throw new Error(`Error querying vector store: ${JSON.stringify(error, null, 2)}`);
+        }
+        const nodes = (data || []).map((item)=>{
+            const node = metadataDictToNode(item.metadata ?? {}, {
+                fallback: {
+                    id: item.id,
+                    text: item.content,
+                    metadata: item.metadata
+                }
+            });
+            node.embedding = item.embedding;
+            node.setContent(item.content);
+            return node;
+        });
+        return {
+            nodes,
+            similarities: nodes.map(()=>1.0),
+            ids: (data || []).map((item)=>item.id)
+        };
+    }
     /**
    * Converts metadata filters to supabase query filter format
    * @param queryFilters - Metadata filters to convert

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@vectorstores/supabase",
   "description": "Supabase Storage for vectorstores",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -35,11 +35,11 @@
   },
   "devDependencies": {
     "vitest": "2.1.0",
-    "@vectorstores/core": "0.1.4",
+    "@vectorstores/core": "0.1.6",
     "@vectorstores/env": "0.1.0"
   },
   "peerDependencies": {
-    "@vectorstores/core": "0.1.4",
+    "@vectorstores/core": "0.1.6",
     "@vectorstores/env": "0.1.0"
   },
   "dependencies": {