chromadb 3.3.2 → 3.3.3

package/dist/cjs/chromadb.d.cts

@@ -757,7 +757,6 @@ declare const registerSparseEmbeddingFunction: (name: string, fn: SparseEmbeddin
  * @returns EmbeddingFunction instance or undefined if it cannot be constructed
  */
 declare const getEmbeddingFunction: (args: {
-    collectionName: string;
     client: ChromaClient;
     efConfig?: EmbeddingFunctionConfiguration;
 }) => Promise<EmbeddingFunction | undefined>;
@@ -765,7 +764,7 @@ declare const getEmbeddingFunction: (args: {
  * Retrieves and instantiates a sparse embedding function from configuration.
  * @returns SparseEmbeddingFunction instance or undefined if it cannot be constructed
  */
-declare const getSparseEmbeddingFunction: (collectionName: string, client: ChromaClient, efConfig?: EmbeddingFunctionConfiguration) => Promise<SparseEmbeddingFunction | undefined>;
+declare const getSparseEmbeddingFunction: (client: ChromaClient, efConfig?: EmbeddingFunctionConfiguration) => Promise<SparseEmbeddingFunction | undefined>;
 /**
  * Serializes an embedding function to configuration format.
  * @param embeddingFunction - User provided embedding function
@@ -1722,6 +1721,7 @@ interface Collection {
     /**
      * Performs hybrid search on the collection using expression builders.
      * @param searches - Single search payload or array of payloads
+     * @param options
      * @returns Promise resolving to column-major search results
      */
     search(searches: SearchLike | SearchLike[], options?: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chromadb",
-  "version": "3.3.2",
+  "version": "3.3.3",
   "description": "A JavaScript interface for chroma",
   "keywords": [
     "chroma",

package/src/chroma-client.ts

@@ -259,7 +259,6 @@ export class ChromaClient {
         const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
         const resolvedEmbeddingFunction =
           (await getEmbeddingFunction({
-            collectionName: collection.name,
             client: this,
             efConfig:
               collection.configuration_json.embedding_function ?? undefined,
@@ -346,7 +345,6 @@ export class ChromaClient {
     const resolvedEmbeddingFunction =
       embeddingFunction ??
       (await getEmbeddingFunction({
-        collectionName: data.name,
         client: this,
         efConfig: data.configuration_json.embedding_function ?? undefined,
       })) ??
@@ -391,7 +389,6 @@ export class ChromaClient {
     const resolvedEmbeddingFunction =
       embeddingFunction ??
       (await getEmbeddingFunction({
-        collectionName: data.name,
         client: this,
         efConfig: data.configuration_json.embedding_function ?? undefined,
       })) ??
@@ -426,7 +423,6 @@ export class ChromaClient {
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
     const resolvedEmbeddingFunction =
       (await getEmbeddingFunction({
-        collectionName: data.name,
         efConfig: data.configuration_json.embedding_function ?? undefined,
         client: this,
       })) ?? schemaEmbeddingFunction;
@@ -523,7 +519,6 @@ export class ChromaClient {
     const resolvedEmbeddingFunction =
       embeddingFunction ??
       (await getEmbeddingFunction({
-        collectionName: name,
         efConfig: data.configuration_json.embedding_function ?? undefined,
         client: this,
       })) ??

package/src/collection-configuration.ts

@@ -207,7 +207,6 @@ export const processUpdateCollectionConfig = async ({
   const embeddingFunction =
     currentEmbeddingFunction ||
     (await getEmbeddingFunction({
-      collectionName: collectionName,
       client,
       efConfig: currentConfiguration.embeddingFunction ?? undefined,
     }));

package/src/collection.ts

@@ -45,7 +45,7 @@ import {
   processUpdateCollectionConfig,
   UpdateCollectionConfiguration,
 } from "./collection-configuration";
-import { SearchLike, SearchResult, toSearch } from "./execution/expression";
+import { SearchLike, SearchResult, toSearch } from "./execution";
 import { isPlainObject } from "./execution/expression/common";
 import { Schema, EMBEDDING_KEY, DOCUMENT_KEY } from "./schema";
 import type { SparseVectorIndexConfig } from "./schema";
@@ -216,6 +216,7 @@ export interface Collection {
   /**
    * Performs hybrid search on the collection using expression builders.
    * @param searches - Single search payload or array of payloads
+   * @param options
    * @returns Promise resolving to column-major search results
    */
   search(
@@ -366,7 +367,9 @@ export class CollectionImpl implements Collection {
 
     if (!embeddingFunction) {
       throw new ChromaValueError(
-        "Embedding function must be defined for operations requiring embeddings.",
+        `No embedding function found for collection '${this._name}'. ` +
+          "You can either provide embeddings directly, or ensure the appropriate " +
+          "embedding function package (e.g. @chroma-core/default-embed) is installed.",
       );
     }
 
@@ -459,10 +462,8 @@ export class CollectionImpl implements Collection {
           // Get document at this position
           if (index < documentsList.length) {
             const doc = documentsList[index];
-            if (typeof doc === "string") {
-              inputs.push(doc);
-              positions.push(index);
-            }
+            inputs.push(doc);
+            positions.push(index);
           }
         });
 

package/src/embedding-function.ts

@@ -1,6 +1,5 @@
 import { EmbeddingFunctionConfiguration, SparseVector } from "./api";
 import { ChromaValueError } from "./errors";
-import { DefaultEmbeddingFunction } from "@chroma-core/default-embed";
 import { ChromaClient } from "./chroma-client";
 
 /**
@@ -140,7 +139,8 @@ const pythonEmbeddingFunctions: Record<string, string> = {
   default: "default-embed",
   together_ai: "together-ai",
   sentence_transformer: "sentence-transformer",
-  google_genai: "google-gemini",
+  google_gemini: "google-gemini",
+  google_genai: "google-gemini", // Backward compatibility alias
 };
 
 const unsupportedEmbeddingFunctions: Set<string> = new Set([
@@ -225,41 +225,16 @@ export const registerSparseEmbeddingFunction = (
  * @returns EmbeddingFunction instance or undefined if it cannot be constructed
  */
 export const getEmbeddingFunction = async (args: {
-  collectionName: string;
   client: ChromaClient;
   efConfig?: EmbeddingFunctionConfiguration;
 }) => {
-  const { collectionName, client, efConfig } = args;
+  const { client, efConfig } = args;
 
-  if (!efConfig) {
-    console.warn(
-      `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`,
-    );
-    return undefined;
-  }
-
-  if (efConfig.type === "legacy") {
-    console.warn(
-      `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`,
-    );
-    return undefined;
-  }
-
-  if (efConfig.type === "unknown") {
-    console.warn(
-      `Unknown embedding function configuration for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`,
-    );
-    return undefined;
-  }
-
-  if (efConfig.type !== "known") {
+  if (efConfig?.type !== "known") {
     return undefined;
   }
 
   if (unsupportedEmbeddingFunctions.has(efConfig.name)) {
-    console.warn(
-      `Embedding function ${efConfig.name} is not supported in the JS/TS SDK. 'add' and 'query' will fail unless you provide them embeddings directly.`,
-    );
     return undefined;
   }
 
@@ -276,33 +251,23 @@ export const getEmbeddingFunction = async (args: {
       await import(fullPackageName);
       embeddingFunction = knownEmbeddingFunctions.get(packageName);
     } catch (error) {
-      // Dynamic loading failed, proceed with warning
+      // Dynamic loading failed
     }
 
     if (!embeddingFunction) {
-      console.warn(
-        `Collection ${collectionName} was created with the ${packageName} embedding function. However, the @chroma-core/${packageName} package is not installed. 'add' and 'query' will fail unless you provide them embeddings directly, or install the @chroma-core/${packageName} package.`,
-      );
       return undefined;
     }
   }
 
-  let constructorConfig: Record<string, any> =
-    efConfig.type === "known" ? (efConfig.config as Record<string, any>) : {};
+  const constructorConfig: Record<string, any> =
+    (efConfig.config as Record<string, any>) ?? {};
 
   try {
     if (embeddingFunction.buildFromConfig) {
       return embeddingFunction.buildFromConfig(constructorConfig, client);
     }
-
-    console.warn(
-      `Embedding function ${packageName} does not define a 'buildFromConfig' function. 'add' and 'query' will fail unless you provide them embeddings directly.`,
-    );
     return undefined;
   } catch (e) {
-    console.warn(
-      `Embedding function ${packageName} failed to build with config: ${constructorConfig}. 'add' and 'query' will fail unless you provide them embeddings directly. Error: ${e}`,
-    );
     return undefined;
   }
 };
@@ -312,26 +277,14 @@ export const getEmbeddingFunction = async (args: {
  * @returns SparseEmbeddingFunction instance or undefined if it cannot be constructed
  */
 export const getSparseEmbeddingFunction = async (
-  collectionName: string,
   client: ChromaClient,
   efConfig?: EmbeddingFunctionConfiguration,
 ) => {
-  if (!efConfig) {
-    return undefined;
-  }
-
-  if (efConfig.type === "legacy") {
-    return undefined;
-  }
-
-  if (efConfig.type !== "known") {
+  if (efConfig?.type !== "known") {
     return undefined;
   }
 
   if (unsupportedSparseEmbeddingFunctions.has(efConfig.name)) {
-    console.warn(
-      "Embedding function ${efConfig.name} is not supported in the JS/TS SDK. 'add' and 'query' will fail unless you provide them embeddings directly.",
-    );
     return undefined;
   }
 
@@ -345,33 +298,23 @@ export const getSparseEmbeddingFunction = async (
       await import(fullPackageName);
       sparseEmbeddingFunction = knownSparseEmbeddingFunctions.get(packageName);
     } catch (error) {
-      // Dynamic loading failed, proceed with warning
+      // Dynamic loading failed
     }
 
     if (!sparseEmbeddingFunction) {
-      console.warn(
-        `Collection ${collectionName} was created with the ${packageName} sparse embedding function. However, the @chroma-core/${packageName} package is not installed.`,
-      );
       return undefined;
     }
   }
 
-  let constructorConfig: Record<string, any> =
-    efConfig.type === "known" ? (efConfig.config as Record<string, any>) : {};
+  const constructorConfig: Record<string, any> =
+    (efConfig.config as Record<string, any>) ?? {};
 
   try {
     if (sparseEmbeddingFunction.buildFromConfig) {
       return sparseEmbeddingFunction.buildFromConfig(constructorConfig, client);
     }
-
-    console.warn(
-      `Sparse embedding function ${packageName} does not define a 'buildFromConfig' function.`,
-    );
     return undefined;
   } catch (e) {
-    console.warn(
-      `Sparse embedding function ${packageName} failed to build with config: ${constructorConfig}. Error: ${e}`,
-    );
     return undefined;
   }
 };

package/src/schema.ts

@@ -1285,7 +1285,6 @@ export class Schema {
     });
 
     config.embeddingFunction = await getEmbeddingFunction({
-      collectionName: "schema deserialization",
       client,
       efConfig: json.embedding_function as EmbeddingFunctionConfiguration,
     });
@@ -1307,7 +1306,6 @@ export class Schema {
 
     const embeddingFunction =
       (await getSparseEmbeddingFunction(
-        "schema deserialization",
         client,
         json.embedding_function as EmbeddingFunctionConfiguration,
       )) ??