chromadb 1.7.1-beta2 → 1.7.1

package/dist/chromadb.d.ts

@@ -1,5 +1,6 @@
 import * as openai from 'openai';
 import * as cohere_ai from 'cohere-ai';
+import * as _google_generative_ai from '@google/generative-ai';
 
 /**
  * FastAPI
@@ -1211,6 +1212,16 @@ declare class AdminClient {
     }): Promise<Database>;
 }
 
+interface CloudClientParams {
+    apiKey?: string;
+    database?: string;
+    cloudHost?: string;
+    cloudPort?: string;
+}
+declare class CloudClient extends ChromaClient {
+    constructor({ apiKey, database, cloudHost, cloudPort }: CloudClientParams);
+}
+
 declare class OpenAIEmbeddingFunction implements IEmbeddingFunction {
     private api_key;
     private org_id;
@@ -1246,6 +1257,24 @@ declare class CohereEmbeddingFunction implements IEmbeddingFunction {
     }>;
 }
 
+declare class GoogleGenerativeAiEmbeddingFunction implements IEmbeddingFunction {
+    private api_key;
+    private model;
+    private googleGenAiApi?;
+    private taskType;
+    constructor({ googleApiKey, model, taskType }: {
+        googleApiKey: string;
+        model?: string;
+        taskType?: string;
+    });
+    private loadClient;
+    generate(texts: string[]): Promise<any>;
+    /** @ignore */
+    static import(): Promise<{
+        googleGenAi: typeof _google_generative_ai;
+    }>;
+}
+
 declare class HuggingFaceEmbeddingServerFunction implements IEmbeddingFunction {
     private url;
     constructor({ url }: {
@@ -1254,4 +1283,15 @@ declare class HuggingFaceEmbeddingServerFunction implements IEmbeddingFunction {
     generate(texts: string[]): Promise<any>;
 }
 
-export { type AddParams, AdminClient, ChromaClient, type ChromaClientParams, CohereEmbeddingFunction, Collection, type CollectionMetadata, type CollectionType, type CreateCollectionParams, type DeleteCollectionParams, type DeleteParams, type Document, type Documents, type Embedding, type Embeddings, type GetCollectionParams, type GetOrCreateCollectionParams, type GetParams, type GetResponse, HuggingFaceEmbeddingServerFunction, type ID, type IDs, type IEmbeddingFunction, IncludeEnum, type ListCollectionsParams, type Metadata, type Metadatas, type ModifyCollectionParams, OpenAIEmbeddingFunction, type PeekParams, type QueryParams, type QueryResponse, type UpdateParams, type UpsertParams, type Where, type WhereDocument };
+declare class JinaEmbeddingFunction implements IEmbeddingFunction {
+    private model_name;
+    private api_url;
+    private headers;
+    constructor({ jinaai_api_key, model_name }: {
+        jinaai_api_key: string;
+        model_name?: string;
+    });
+    generate(texts: string[]): Promise<any[]>;
+}
+
+export { AddParams, AdminClient, ChromaClient, ChromaClientParams, CloudClient, CohereEmbeddingFunction, Collection, CollectionMetadata, CollectionType, CreateCollectionParams, DeleteCollectionParams, DeleteParams, Document, Documents, Embedding, Embeddings, GetCollectionParams, GetOrCreateCollectionParams, GetParams, GetResponse, GoogleGenerativeAiEmbeddingFunction, HuggingFaceEmbeddingServerFunction, ID, IDs, IEmbeddingFunction, IncludeEnum, JinaEmbeddingFunction, ListCollectionsParams, Metadata, Metadatas, ModifyCollectionParams, OpenAIEmbeddingFunction, PeekParams, QueryParams, QueryResponse, UpdateParams, UpsertParams, Where, WhereDocument };

package/dist/chromadb.legacy-esm.js

@@ -2825,6 +2825,32 @@ var ChromaClient = class {
   }
 };
 
+// src/CloudClient.ts
+var CloudClient = class extends ChromaClient {
+  constructor({ apiKey, database, cloudHost, cloudPort }) {
+    if (!apiKey) {
+      apiKey = process.env.CHROMA_API_KEY;
+    }
+    if (!apiKey) {
+      throw new Error("No API key provided");
+    }
+    cloudHost = cloudHost || "https://api.trychroma.com";
+    cloudPort = cloudPort || "8000";
+    const path = `${cloudHost}:${cloudPort}`;
+    const auth = {
+      provider: "token",
+      credentials: apiKey,
+      providerOptions: { headerType: "X_CHROMA_TOKEN" }
+    };
+    return new ChromaClient({
+      path,
+      auth,
+      database
+    });
+    super();
+  }
+};
+
 // src/embeddings/OpenAIEmbeddingFunction.ts
 var OpenAIApi;
 var openAiVersion = null;
@@ -2965,6 +2991,55 @@ var CohereEmbeddingFunction = class _CohereEmbeddingFunction {
   }
 };
 
+// src/embeddings/GoogleGeminiEmbeddingFunction.ts
+var googleGenAiApi;
+var GoogleGenerativeAiEmbeddingFunction = class _GoogleGenerativeAiEmbeddingFunction {
+  constructor({ googleApiKey, model, taskType }) {
+    this.api_key = googleApiKey;
+    this.model = model || "embedding-001";
+    this.taskType = taskType || "RETRIEVAL_DOCUMENT";
+  }
+  async loadClient() {
+    if (this.googleGenAiApi)
+      return;
+    try {
+      const { googleGenAi } = await _GoogleGenerativeAiEmbeddingFunction.import();
+      googleGenAiApi = googleGenAi;
+      googleGenAiApi = new googleGenAiApi(this.api_key);
+    } catch (_a) {
+      if (_a.code === "MODULE_NOT_FOUND") {
+        throw new Error("Please install the @google/generative-ai package to use the GoogleGenerativeAiEmbeddingFunction, `npm install -S @google/generative-ai`");
+      }
+      throw _a;
+    }
+    this.googleGenAiApi = googleGenAiApi;
+  }
+  async generate(texts) {
+    await this.loadClient();
+    const model = this.googleGenAiApi.getGenerativeModel({ model: this.model });
+    const response = await model.batchEmbedContents({
+      requests: texts.map((t) => ({
+        content: { parts: [{ text: t }] },
+        taskType: this.taskType
+      }))
+    });
+    const embeddings = response.embeddings.map((e) => e.values);
+    return embeddings;
+  }
+  /** @ignore */
+  static async import() {
+    try {
+      const { GoogleGenerativeAI } = await import("@google/generative-ai");
+      const googleGenAi = GoogleGenerativeAI;
+      return { googleGenAi };
+    } catch (e) {
+      throw new Error(
+        "Please install @google/generative-ai as a dependency with, e.g. `yarn add @google/generative-ai`"
+      );
+    }
+  }
+};
+
 // src/types.ts
 var IncludeEnum = /* @__PURE__ */ ((IncludeEnum2) => {
   IncludeEnum2["Documents"] = "documents";
@@ -2994,13 +3069,54 @@ var HuggingFaceEmbeddingServerFunction = class {
     return data;
   }
 };
+
+// src/embeddings/JinaEmbeddingFunction.ts
+var JinaEmbeddingFunction = class {
+  constructor({ jinaai_api_key, model_name }) {
+    this.model_name = model_name || "jina-embeddings-v2-base-en";
+    this.api_url = "https://api.jina.ai/v1/embeddings";
+    this.headers = {
+      Authorization: `Bearer ${jinaai_api_key}`,
+      "Accept-Encoding": "identity",
+      "Content-Type": "application/json"
+    };
+  }
+  async generate(texts) {
+    try {
+      const response = await fetch(this.api_url, {
+        method: "POST",
+        headers: this.headers,
+        body: JSON.stringify({
+          input: texts,
+          model: this.model_name
+        })
+      });
+      const data = await response.json();
+      if (!data || !data.data) {
+        throw new Error(data.detail);
+      }
+      const embeddings = data.data;
+      const sortedEmbeddings = embeddings.sort((a, b) => a.index - b.index);
+      return sortedEmbeddings.map((result) => result.embedding);
+    } catch (error) {
+      if (error instanceof Error) {
+        throw new Error(`Error calling Jina AI API: ${error.message}`);
+      } else {
+        throw new Error(`Error calling Jina AI API: ${error}`);
+      }
+    }
+  }
+};
 export {
   AdminClient,
   ChromaClient,
+  CloudClient,
   CohereEmbeddingFunction,
   Collection,
+  GoogleGenerativeAiEmbeddingFunction,
   HuggingFaceEmbeddingServerFunction,
   IncludeEnum,
+  JinaEmbeddingFunction,
   OpenAIEmbeddingFunction
 };
 //# sourceMappingURL=chromadb.mjs.map

package/dist/chromadb.mjs

@@ -2825,6 +2825,32 @@ var ChromaClient = class {
   }
 };
 
+// src/CloudClient.ts
+var CloudClient = class extends ChromaClient {
+  constructor({ apiKey, database, cloudHost, cloudPort }) {
+    if (!apiKey) {
+      apiKey = process.env.CHROMA_API_KEY;
+    }
+    if (!apiKey) {
+      throw new Error("No API key provided");
+    }
+    cloudHost = cloudHost || "https://api.trychroma.com";
+    cloudPort = cloudPort || "8000";
+    const path = `${cloudHost}:${cloudPort}`;
+    const auth = {
+      provider: "token",
+      credentials: apiKey,
+      providerOptions: { headerType: "X_CHROMA_TOKEN" }
+    };
+    return new ChromaClient({
+      path,
+      auth,
+      database
+    });
+    super();
+  }
+};
+
 // src/embeddings/OpenAIEmbeddingFunction.ts
 var OpenAIApi;
 var openAiVersion = null;
@@ -2965,6 +2991,55 @@ var CohereEmbeddingFunction = class _CohereEmbeddingFunction {
   }
 };
 
+// src/embeddings/GoogleGeminiEmbeddingFunction.ts
+var googleGenAiApi;
+var GoogleGenerativeAiEmbeddingFunction = class _GoogleGenerativeAiEmbeddingFunction {
+  constructor({ googleApiKey, model, taskType }) {
+    this.api_key = googleApiKey;
+    this.model = model || "embedding-001";
+    this.taskType = taskType || "RETRIEVAL_DOCUMENT";
+  }
+  async loadClient() {
+    if (this.googleGenAiApi)
+      return;
+    try {
+      const { googleGenAi } = await _GoogleGenerativeAiEmbeddingFunction.import();
+      googleGenAiApi = googleGenAi;
+      googleGenAiApi = new googleGenAiApi(this.api_key);
+    } catch (_a) {
+      if (_a.code === "MODULE_NOT_FOUND") {
+        throw new Error("Please install the @google/generative-ai package to use the GoogleGenerativeAiEmbeddingFunction, `npm install -S @google/generative-ai`");
+      }
+      throw _a;
+    }
+    this.googleGenAiApi = googleGenAiApi;
+  }
+  async generate(texts) {
+    await this.loadClient();
+    const model = this.googleGenAiApi.getGenerativeModel({ model: this.model });
+    const response = await model.batchEmbedContents({
+      requests: texts.map((t) => ({
+        content: { parts: [{ text: t }] },
+        taskType: this.taskType
+      }))
+    });
+    const embeddings = response.embeddings.map((e) => e.values);
+    return embeddings;
+  }
+  /** @ignore */
+  static async import() {
+    try {
+      const { GoogleGenerativeAI } = await import("@google/generative-ai");
+      const googleGenAi = GoogleGenerativeAI;
+      return { googleGenAi };
+    } catch (e) {
+      throw new Error(
+        "Please install @google/generative-ai as a dependency with, e.g. `yarn add @google/generative-ai`"
+      );
+    }
+  }
+};
+
 // src/types.ts
 var IncludeEnum = /* @__PURE__ */ ((IncludeEnum2) => {
   IncludeEnum2["Documents"] = "documents";
@@ -2994,13 +3069,54 @@ var HuggingFaceEmbeddingServerFunction = class {
     return data;
   }
 };
+
+// src/embeddings/JinaEmbeddingFunction.ts
+var JinaEmbeddingFunction = class {
+  constructor({ jinaai_api_key, model_name }) {
+    this.model_name = model_name || "jina-embeddings-v2-base-en";
+    this.api_url = "https://api.jina.ai/v1/embeddings";
+    this.headers = {
+      Authorization: `Bearer ${jinaai_api_key}`,
+      "Accept-Encoding": "identity",
+      "Content-Type": "application/json"
+    };
+  }
+  async generate(texts) {
+    try {
+      const response = await fetch(this.api_url, {
+        method: "POST",
+        headers: this.headers,
+        body: JSON.stringify({
+          input: texts,
+          model: this.model_name
+        })
+      });
+      const data = await response.json();
+      if (!data || !data.data) {
+        throw new Error(data.detail);
+      }
+      const embeddings = data.data;
+      const sortedEmbeddings = embeddings.sort((a, b) => a.index - b.index);
+      return sortedEmbeddings.map((result) => result.embedding);
+    } catch (error) {
+      if (error instanceof Error) {
+        throw new Error(`Error calling Jina AI API: ${error.message}`);
+      } else {
+        throw new Error(`Error calling Jina AI API: ${error}`);
+      }
+    }
+  }
+};
 export {
   AdminClient,
   ChromaClient,
+  CloudClient,
   CohereEmbeddingFunction,
   Collection,
+  GoogleGenerativeAiEmbeddingFunction,
   HuggingFaceEmbeddingServerFunction,
   IncludeEnum,
+  JinaEmbeddingFunction,
   OpenAIEmbeddingFunction
 };
 //# sourceMappingURL=chromadb.mjs.map