fix(core): add filter to vector store retriever

LarchLiu · Jun 26, 2023 · 5ba4218 · 5ba4218 · vercel · Jun 26, 2023
1 parent 9913aea
commit 5ba4218
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/packages/core/src/storage/supabase/index.ts b/packages/core/src/storage/supabase/index.ts
@@ -114,7 +114,7 @@ export class SupabaseVectorStorage extends VectorStorage<SupabaseVectorConfig> {
 
   async save(data?: WebInfoData) {
     if (!data && !this.data)
-      throw new Error('ImageStorage error: No Storage Data')
+      throw new Error('VectorStorage error: No Storage Data')
 
     const storageData = (data || this.data)!
     const rawDoc = new Document({ pageContent: storageData.content, metadata: this.config.metaData })
@@ -126,11 +126,10 @@ export class SupabaseVectorStorage extends VectorStorage<SupabaseVectorConfig> {
     })
 
     const docs = await textSplitter.splitDocuments([rawDoc])
-    const store = new SupabaseVectorStore(this.config.embeddingsInfo.embeddings, {
+    await SupabaseVectorStore.fromDocuments(docs, this.config.embeddingsInfo.embeddings, {
       client: this.client,
       tableName: this.config.embeddingsInfo.indexName,
     })
-    await store.addDocuments(docs)
   }
 
   async getRetriever() {
@@ -140,6 +139,11 @@ export class SupabaseVectorStorage extends VectorStorage<SupabaseVectorConfig> {
         client: this.client,
         tableName: this.config.embeddingsInfo.indexName,
         queryName: this.config.embeddingsInfo.queryName,
+        filter: {
+          appName: this.config.metaData.appName,
+          botId: this.config.metaData.botId,
+          userId: this.config.metaData.userId,
+        },
       })
     return vectorStore.asRetriever()
   }

diff --git a/server/nuxt3/composables/config.ts b/server/nuxt3/composables/config.ts
@@ -486,15 +486,15 @@ export const serverConfig: BasicConfig<ModelsConfig> = {
       output: 'Server',
     },
     cloudflare: {
-      displayName: 'Cloudflare Worker',
+      displayName: 'Cloudflare Pages',
       config: {
         token: {
           label: 'Cloudflare Token',
           value: '',
           require: true,
         },
         siteid: {
-          label: 'Worker Name',
+          label: 'Project Name',
           value: '',
           require: true,
         },