Migrate Pinecone to class (#4726)

migrate pinecone to class

Author: shatfield4

server/utils/helpers/index.js

@@ -87,7 +87,7 @@ function getVectorDbClass(getExactly = null) {
   switch (vectorSelection) {
     case "pinecone":
       const { Pinecone } = require("../vectorDbProviders/pinecone");
-      return Pinecone;
+      return new Pinecone();
     case "chroma":
       const { Chroma } = require("../vectorDbProviders/chroma");
       return Chroma;

server/utils/vectorDbProviders/pinecone/index.js

@@ -6,9 +6,12 @@ const { v4: uuidv4 } = require("uuid");
 const { toChunks, getEmbeddingEngineSelection } = require("../../helpers");
 const { sourceIdentifier } = require("../../chats");
 
-const PineconeDB = {
-  name: "Pinecone",
-  connect: async function () {
+class PineconeDB {
+  constructor() {
+    this.name = "Pinecone";
+  }
+
+  async connect() {
     if (process.env.VECTOR_DB !== "pinecone")
       throw new Error("Pinecone::Invalid ENV settings");
 
@@ -21,22 +24,25 @@ const PineconeDB = {
 
     if (!status.ready) throw new Error("Pinecone::Index not ready.");
     return { client, pineconeIndex, indexName: process.env.PINECONE_INDEX };
-  },
-  totalVectors: async function () {
+  }
+
+  async totalVectors() {
     const { pineconeIndex } = await this.connect();
     const { namespaces } = await pineconeIndex.describeIndexStats();
 
     return Object.values(namespaces).reduce(
       (a, b) => a + (b?.recordCount || 0),
       0
     );
-  },
-  namespaceCount: async function (_namespace = null) {
+  }
+
+  async namespaceCount(_namespace = null) {
     const { pineconeIndex } = await this.connect();
     const namespace = await this.namespace(pineconeIndex, _namespace);
     return namespace?.recordCount || 0;
-  },
-  similarityResponse: async function ({
+  }
+
+  async similarityResponse({
     client,
     namespace,
     queryVector,
@@ -75,28 +81,33 @@ const PineconeDB = {
     });
 
     return result;
-  },
-  namespace: async function (index, namespace = null) {
+  }
+
+  async namespace(index, namespace = null) {
     if (!namespace) throw new Error("No namespace value provided.");
     const { namespaces } = await index.describeIndexStats();
     return namespaces.hasOwnProperty(namespace) ? namespaces[namespace] : null;
-  },
-  hasNamespace: async function (namespace = null) {
+  }
+
+  async hasNamespace(namespace = null) {
     if (!namespace) return false;
     const { pineconeIndex } = await this.connect();
     return await this.namespaceExists(pineconeIndex, namespace);
-  },
-  namespaceExists: async function (index, namespace = null) {
+  }
+
+  async namespaceExists(index, namespace = null) {
     if (!namespace) throw new Error("No namespace value provided.");
     const { namespaces } = await index.describeIndexStats();
     return namespaces.hasOwnProperty(namespace);
-  },
-  deleteVectorsInNamespace: async function (index, namespace = null) {
+  }
+
+  async deleteVectorsInNamespace(index, namespace = null) {
     const pineconeNamespace = index.namespace(namespace);
     await pineconeNamespace.deleteAll();
     return true;
-  },
-  addDocumentToNamespace: async function (
+  }
+
+  async addDocumentToNamespace(
     namespace,
     documentData = {},
     fullFilePath = null,
@@ -197,8 +208,9 @@ const PineconeDB = {
       console.error("addDocumentToNamespace", e.message);
       return { vectorized: false, error: e.message };
     }
-  },
-  deleteDocumentFromNamespace: async function (namespace, docId) {
+  }
+
+  async deleteDocumentFromNamespace(namespace, docId) {
     const { DocumentVectors } = require("../../../models/vectors");
     const { pineconeIndex } = await this.connect();
     if (!(await this.namespaceExists(pineconeIndex, namespace))) return;
@@ -216,8 +228,9 @@ const PineconeDB = {
     const indexes = knownDocuments.map((doc) => doc.id);
     await DocumentVectors.deleteIds(indexes);
     return true;
-  },
-  "namespace-stats": async function (reqBody = {}) {
+  }
+
+  async "namespace-stats"(reqBody = {}) {
     const { namespace = null } = reqBody;
     if (!namespace) throw new Error("namespace required");
     const { pineconeIndex } = await this.connect();
@@ -227,8 +240,9 @@ const PineconeDB = {
     return stats
       ? stats
       : { message: "No stats were able to be fetched from DB" };
-  },
-  "delete-namespace": async function (reqBody = {}) {
+  }
+
+  async "delete-namespace"(reqBody = {}) {
     const { namespace = null } = reqBody;
     const { pineconeIndex } = await this.connect();
     if (!(await this.namespaceExists(pineconeIndex, namespace)))
@@ -239,8 +253,9 @@ const PineconeDB = {
     return {
       message: `Namespace ${namespace} was deleted along with ${details.vectorCount} vectors.`,
     };
-  },
-  performSimilaritySearch: async function ({
+  }
+
+  async performSimilaritySearch({
     namespace = null,
     input = "",
     LLMConnector = null,
@@ -275,8 +290,9 @@ const PineconeDB = {
       sources: this.curateSources(sources),
       message: false,
     };
-  },
-  curateSources: function (sources = []) {
+  }
+
+  curateSources(sources = []) {
     const documents = [];
     for (const source of sources) {
       const { metadata = {} } = source;
@@ -290,7 +306,7 @@ const PineconeDB = {
       }
     }
     return documents;
-  },
-};
+  }
+}
 
 module.exports.Pinecone = PineconeDB;