endee-llamaindex 1.0.1 → 1.0.2

package/dist/endeeClient.d.ts

@@ -1,27 +1,30 @@
+import type { Index, Endee } from "endee";
 import { BaseNode, BaseVectorStore, Metadata, VectorStoreBaseParams, VectorStoreQuery, VectorStoreQueryResult } from "llamaindex";
 type EndeeParams = {
     indexName: string;
     authToken: string;
     chunkSize?: number;
+    textKey?: string;
 } & VectorStoreBaseParams;
 export declare class EndeeVectorStore extends BaseVectorStore {
     storesText: boolean;
     private authToken;
     private indexName;
     private chunkSize;
-    private db;
+    private db?;
+    private textKey;
     constructor(params: EndeeParams);
     private getDB;
-    client(): any;
-    index(): Promise<any>;
+    client(): Promise<Endee>;
+    index(): Promise<Index>;
     add(embeddingResults: BaseNode<Metadata>[]): Promise<string[]>;
-    protected saveChunk(idx: any, chunk: any): Promise<boolean>;
+    protected saveChunk(idx: Index, chunk: any): Promise<boolean>;
     nodeToRecord(node: BaseNode<Metadata>): {
         id: string;
         vector: number[];
         meta: Metadata;
     };
     query(query: VectorStoreQuery, _options?: object): Promise<VectorStoreQueryResult>;
-    delete(refDocId: string, deleteKwargs?: object): Promise<void>;
+    delete(refDocId: string): Promise<void>;
 }
 export {};

package/dist/endeeClient.js

@@ -1,4 +1,3 @@
-import { Endee } from "endee";
 import { BaseVectorStore, metadataDictToNode, nodeToMetadata, } from "llamaindex";
 export class EndeeVectorStore extends BaseVectorStore {
     constructor(params) {
@@ -7,21 +6,21 @@ export class EndeeVectorStore extends BaseVectorStore {
         this.authToken = params.authToken;
         this.indexName = params.indexName;
         this.chunkSize = params.chunkSize ?? 100;
-        this.db = new Endee(params.authToken);
+        this.textKey = params.textKey ?? "text";
     }
-    getDB() {
+    async getDB() {
         if (!this.db) {
+            const { Endee } = await import("endee");
             this.db = new Endee(this.authToken);
         }
-        return this.db;
+        return Promise.resolve(this.db);
     }
     client() {
         return this.getDB();
     }
     async index() {
-        const db = this.getDB();
-        const index = await db.getIndex(this.indexName);
-        return index;
+        const db = await this.getDB();
+        return await db.getIndex(this.indexName);
     }
     async add(embeddingResults) {
         if (embeddingResults.length == 0) {
@@ -43,7 +42,7 @@ export class EndeeVectorStore extends BaseVectorStore {
                 return {
                     id: node.id,
                     vector: node.vector,
-                    meta: { text: nodeContent.text },
+                    meta: { [this.textKey]: nodeContent.text },
                     filter: nodeContent.relationships.SOURCE.metadata,
                 };
             });
@@ -60,7 +59,7 @@ export class EndeeVectorStore extends BaseVectorStore {
             return true;
         }
         catch (err) {
-            const msg = `${err}`;
+            console.error(err);
             return false;
         }
     }
@@ -119,12 +118,8 @@ export class EndeeVectorStore extends BaseVectorStore {
             ids: results.map((row) => row.id),
         };
     }
-    async delete(refDocId, deleteKwargs) {
-        const [idx, index] = await Promise.all([
-            this.index(),
-            //to get the information about the index
-            this.db?.describeIndex(this.indexName),
-        ]);
+    async delete(refDocId) {
+        const idx = await this.index();
         await idx.deleteVector(refDocId);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "endee-llamaindex",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "LlamaIndex integration with endee vector database",
   "type": "module",
   "main": "dist/endeeClient.js",