chromadb 1.7.1 → 1.7.3

package/README.md

@@ -24,7 +24,7 @@ const collection = await chroma.createCollection({ name: "test-from-js" });
 for (let i = 0; i < 20; i++) {
   await collection.add({
     ids: ["test-id-" + i.toString()],
-    embeddings, [1, 2, 3, 4, 5],
+    embeddings: [1, 2, 3, 4, 5],
     documents: ["test"],
   });
 }

package/dist/chromadb.d.ts

@@ -1,5 +1,4 @@
 import * as openai from 'openai';
-import * as cohere_ai from 'cohere-ai';
 import * as _google_generative_ai from '@google/generative-ai';
 
 /**
@@ -1242,19 +1241,15 @@ declare class OpenAIEmbeddingFunction implements IEmbeddingFunction {
 }
 
 declare class CohereEmbeddingFunction implements IEmbeddingFunction {
-    private api_key;
-    private model;
     private cohereAiApi?;
-    constructor({ cohere_api_key, model }: {
+    private model;
+    private apiKey;
+    constructor({ cohere_api_key, model, }: {
         cohere_api_key: string;
         model?: string;
     });
-    private loadClient;
-    generate(texts: string[]): Promise<any>;
-    /** @ignore */
-    static import(): Promise<{
-        cohere: typeof cohere_ai;
-    }>;
+    private initCohereClient;
+    generate(texts: string[]): Promise<number[][]>;
 }
 
 declare class GoogleGenerativeAiEmbeddingFunction implements IEmbeddingFunction {

package/dist/chromadb.legacy-esm.js

@@ -2949,46 +2949,83 @@ var OpenAIEmbeddingFunction = class _OpenAIEmbeddingFunction {
 };
 
 // src/embeddings/CohereEmbeddingFunction.ts
-var CohereAiApi;
-var CohereEmbeddingFunction = class _CohereEmbeddingFunction {
-  constructor({ cohere_api_key, model }) {
-    this.api_key = cohere_api_key;
-    this.model = model || "large";
+var CohereAISDK56 = class {
+  constructor(configuration) {
+    this.apiKey = configuration.apiKey;
   }
   async loadClient() {
-    if (this.cohereAiApi)
+    if (this.cohereClient)
       return;
-    try {
-      const { cohere } = await _CohereEmbeddingFunction.import();
-      CohereAiApi = cohere;
-      CohereAiApi.init(this.api_key);
-    } catch (_a) {
-      if (_a.code === "MODULE_NOT_FOUND") {
-        throw new Error("Please install the cohere-ai package to use the CohereEmbeddingFunction, `npm install -S cohere-ai`");
-      }
-      throw _a;
-    }
-    this.cohereAiApi = CohereAiApi;
+    const { default: cohere } = await import("cohere-ai");
+    cohere.init(this.apiKey);
+    this.cohereClient = cohere;
   }
-  async generate(texts) {
+  async createEmbedding(params) {
     await this.loadClient();
-    const response = await this.cohereAiApi.embed({
-      texts,
-      model: this.model
+    return await this.cohereClient.embed({
+      texts: params.input,
+      model: params.model
+    }).then((response) => {
+      return response.body.embeddings;
     });
-    return response.body.embeddings;
   }
-  /** @ignore */
-  static async import() {
+};
+var CohereAISDK7 = class {
+  constructor(configuration) {
+    this.apiKey = configuration.apiKey;
+  }
+  async loadClient() {
+    if (this.cohereClient)
+      return;
+    const cohere = await import("cohere-ai").then((cohere2) => {
+      return cohere2;
+    });
+    this.cohereClient = new cohere.CohereClient({
+      token: this.apiKey
+    });
+  }
+  async createEmbedding(params) {
+    await this.loadClient();
+    return await this.cohereClient.embed({ texts: params.input, model: params.model }).then((response) => {
+      return response.embeddings;
+    });
+  }
+};
+var CohereEmbeddingFunction = class {
+  constructor({
+    cohere_api_key,
+    model
+  }) {
+    this.model = model || "large";
+    this.apiKey = cohere_api_key;
+  }
+  async initCohereClient() {
+    if (this.cohereAiApi)
+      return;
     try {
-      const { default: cohere } = await import("cohere-ai");
-      return { cohere };
+      this.cohereAiApi = await import("cohere-ai").then((cohere) => {
+        if (cohere.CohereClient) {
+          return new CohereAISDK7({ apiKey: this.apiKey });
+        } else {
+          return new CohereAISDK56({ apiKey: this.apiKey });
+        }
+      });
     } catch (e) {
-      throw new Error(
-        "Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`"
-      );
+      if (e.code === "MODULE_NOT_FOUND") {
+        throw new Error(
+          "Please install the cohere-ai package to use the CohereEmbeddingFunction, `npm install -S cohere-ai`"
+        );
+      }
+      throw e;
     }
   }
+  async generate(texts) {
+    await this.initCohereClient();
+    return await this.cohereAiApi.createEmbedding({
+      model: this.model,
+      input: texts
+    });
+  }
 };
 
 // src/embeddings/GoogleGeminiEmbeddingFunction.ts

package/dist/chromadb.mjs

@@ -2949,46 +2949,83 @@ var OpenAIEmbeddingFunction = class _OpenAIEmbeddingFunction {
 };
 
 // src/embeddings/CohereEmbeddingFunction.ts
-var CohereAiApi;
-var CohereEmbeddingFunction = class _CohereEmbeddingFunction {
-  constructor({ cohere_api_key, model }) {
-    this.api_key = cohere_api_key;
-    this.model = model || "large";
+var CohereAISDK56 = class {
+  constructor(configuration) {
+    this.apiKey = configuration.apiKey;
   }
   async loadClient() {
-    if (this.cohereAiApi)
+    if (this.cohereClient)
       return;
-    try {
-      const { cohere } = await _CohereEmbeddingFunction.import();
-      CohereAiApi = cohere;
-      CohereAiApi.init(this.api_key);
-    } catch (_a) {
-      if (_a.code === "MODULE_NOT_FOUND") {
-        throw new Error("Please install the cohere-ai package to use the CohereEmbeddingFunction, `npm install -S cohere-ai`");
-      }
-      throw _a;
-    }
-    this.cohereAiApi = CohereAiApi;
+    const { default: cohere } = await import("cohere-ai");
+    cohere.init(this.apiKey);
+    this.cohereClient = cohere;
   }
-  async generate(texts) {
+  async createEmbedding(params) {
     await this.loadClient();
-    const response = await this.cohereAiApi.embed({
-      texts,
-      model: this.model
+    return await this.cohereClient.embed({
+      texts: params.input,
+      model: params.model
+    }).then((response) => {
+      return response.body.embeddings;
     });
-    return response.body.embeddings;
   }
-  /** @ignore */
-  static async import() {
+};
+var CohereAISDK7 = class {
+  constructor(configuration) {
+    this.apiKey = configuration.apiKey;
+  }
+  async loadClient() {
+    if (this.cohereClient)
+      return;
+    const cohere = await import("cohere-ai").then((cohere2) => {
+      return cohere2;
+    });
+    this.cohereClient = new cohere.CohereClient({
+      token: this.apiKey
+    });
+  }
+  async createEmbedding(params) {
+    await this.loadClient();
+    return await this.cohereClient.embed({ texts: params.input, model: params.model }).then((response) => {
+      return response.embeddings;
+    });
+  }
+};
+var CohereEmbeddingFunction = class {
+  constructor({
+    cohere_api_key,
+    model
+  }) {
+    this.model = model || "large";
+    this.apiKey = cohere_api_key;
+  }
+  async initCohereClient() {
+    if (this.cohereAiApi)
+      return;
     try {
-      const { default: cohere } = await import("cohere-ai");
-      return { cohere };
+      this.cohereAiApi = await import("cohere-ai").then((cohere) => {
+        if (cohere.CohereClient) {
+          return new CohereAISDK7({ apiKey: this.apiKey });
+        } else {
+          return new CohereAISDK56({ apiKey: this.apiKey });
+        }
+      });
     } catch (e) {
-      throw new Error(
-        "Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`"
-      );
+      if (e.code === "MODULE_NOT_FOUND") {
+        throw new Error(
+          "Please install the cohere-ai package to use the CohereEmbeddingFunction, `npm install -S cohere-ai`"
+        );
+      }
+      throw e;
     }
   }
+  async generate(texts) {
+    await this.initCohereClient();
+    return await this.cohereAiApi.createEmbedding({
+      model: this.model,
+      input: texts
+    });
+  }
 };
 
 // src/embeddings/GoogleGeminiEmbeddingFunction.ts