chromadb 3.3.2 → 3.3.3

package/dist/chromadb.d.ts

@@ -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/dist/chromadb.legacy-esm.js

@@ -1516,7 +1516,9 @@ var pythonEmbeddingFunctions = {
   default: "default-embed",
   together_ai: "together-ai",
   sentence_transformer: "sentence-transformer",
+  google_gemini: "google-gemini",
   google_genai: "google-gemini"
+  // Backward compatibility alias
 };
 var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "amazon_bedrock",
@@ -1555,32 +1557,11 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   knownSparseEmbeddingFunctions.set(name, fn);
 };
 var getEmbeddingFunction = async (args) => {
-  const { collectionName, client: client2, 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 void 0;
-  }
-  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 void 0;
-  }
-  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 void 0;
-  }
-  if (efConfig.type !== "known") {
+  const { client: client2, efConfig } = args;
+  if (efConfig?.type !== "known") {
     return void 0;
   }
   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 void 0;
   }
   const packageName = pythonEmbeddingFunctions[efConfig.name] || efConfig.name;
@@ -1596,42 +1577,24 @@ var getEmbeddingFunction = async (args) => {
     } catch (error) {
     }
     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 void 0;
     }
   }
-  let constructorConfig = efConfig.type === "known" ? efConfig.config : {};
+  const constructorConfig = efConfig.config ?? {};
   try {
     if (embeddingFunction.buildFromConfig) {
       return embeddingFunction.buildFromConfig(constructorConfig, client2);
     }
-    console.warn(
-      `Embedding function ${packageName} does not define a 'buildFromConfig' function. 'add' and 'query' will fail unless you provide them embeddings directly.`
-    );
     return void 0;
   } 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 void 0;
   }
 };
-var getSparseEmbeddingFunction = async (collectionName, client2, efConfig) => {
-  if (!efConfig) {
-    return void 0;
-  }
-  if (efConfig.type === "legacy") {
-    return void 0;
-  }
-  if (efConfig.type !== "known") {
+var getSparseEmbeddingFunction = async (client2, efConfig) => {
+  if (efConfig?.type !== "known") {
     return void 0;
   }
   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 void 0;
   }
   const packageName = pythonSparseEmbeddingFunctions[efConfig.name] || efConfig.name;
@@ -1644,25 +1607,16 @@ var getSparseEmbeddingFunction = async (collectionName, client2, efConfig) => {
     } catch (error) {
     }
     if (!sparseEmbeddingFunction) {
-      console.warn(
-        `Collection ${collectionName} was created with the ${packageName} sparse embedding function. However, the @chroma-core/${packageName} package is not installed.`
-      );
       return void 0;
     }
   }
-  let constructorConfig = efConfig.type === "known" ? efConfig.config : {};
+  const constructorConfig = efConfig.config ?? {};
   try {
     if (sparseEmbeddingFunction.buildFromConfig) {
       return sparseEmbeddingFunction.buildFromConfig(constructorConfig, client2);
     }
-    console.warn(
-      `Sparse embedding function ${packageName} does not define a 'buildFromConfig' function.`
-    );
     return void 0;
   } catch (e) {
-    console.warn(
-      `Sparse embedding function ${packageName} failed to build with config: ${constructorConfig}. Error: ${e}`
-    );
     return void 0;
   }
 };
@@ -1788,7 +1742,6 @@ var processUpdateCollectionConfig = async ({
     );
   }
   const embeddingFunction = currentEmbeddingFunction || await getEmbeddingFunction({
-    collectionName,
     client: client2,
     efConfig: currentConfiguration.embeddingFunction ?? void 0
   });
@@ -3886,7 +3839,6 @@ var Schema = class _Schema {
       spann: json.spann ? cloneObject(json.spann) : null
     });
     config.embeddingFunction = await getEmbeddingFunction({
-      collectionName: "schema deserialization",
       client: client2,
       efConfig: json.embedding_function
     });
@@ -3901,7 +3853,6 @@ var Schema = class _Schema {
       bm25: typeof json.bm25 === "boolean" ? json.bm25 : null
     });
     const embeddingFunction = await getSparseEmbeddingFunction(
-      "schema deserialization",
       client2,
       json.embedding_function
     ) ?? config.embeddingFunction ?? void 0;
@@ -3987,7 +3938,7 @@ var CollectionImpl = class _CollectionImpl {
     const embeddingFunction = this._embeddingFunction ?? this.getSchemaEmbeddingFunction();
     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.`
       );
     }
     if (isQuery && embeddingFunction.generateForQueries) {
@@ -4048,10 +3999,8 @@ var CollectionImpl = class _CollectionImpl {
           }
           if (index < documentsList.length) {
             const doc = documentsList[index];
-            if (typeof doc === "string") {
-              inputs.push(doc);
-              positions.push(index);
-            }
+            inputs.push(doc);
+            positions.push(index);
           }
         });
         if (inputs.length === 0) {
@@ -4918,7 +4867,6 @@ var ChromaClient = class {
         );
         const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
         const resolvedEmbeddingFunction = await getEmbeddingFunction({
-          collectionName: collection.name,
           client: this,
           efConfig: collection.configuration_json.embedding_function ?? void 0
         }) ?? schemaEmbeddingFunction;
@@ -4988,7 +4936,6 @@ var ChromaClient = class {
     );
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(serverSchema);
     const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
-      collectionName: data.name,
       client: this,
       efConfig: data.configuration_json.embedding_function ?? void 0
     }) ?? schemaEmbeddingFunction;
@@ -5024,7 +4971,6 @@ var ChromaClient = class {
     const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
     const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
-      collectionName: data.name,
       client: this,
       efConfig: data.configuration_json.embedding_function ?? void 0
     }) ?? schemaEmbeddingFunction;
@@ -5055,7 +5001,6 @@ var ChromaClient = class {
     const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
     const resolvedEmbeddingFunction = await getEmbeddingFunction({
-      collectionName: data.name,
       efConfig: data.configuration_json.embedding_function ?? void 0,
       client: this
     }) ?? schemaEmbeddingFunction;
@@ -5131,7 +5076,6 @@ var ChromaClient = class {
     );
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(serverSchema);
     const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
-      collectionName: name,
       efConfig: data.configuration_json.embedding_function ?? void 0,
       client: this
     }) ?? schemaEmbeddingFunction;

package/dist/chromadb.mjs

@@ -1516,7 +1516,9 @@ var pythonEmbeddingFunctions = {
   default: "default-embed",
   together_ai: "together-ai",
   sentence_transformer: "sentence-transformer",
+  google_gemini: "google-gemini",
   google_genai: "google-gemini"
+  // Backward compatibility alias
 };
 var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "amazon_bedrock",
@@ -1555,32 +1557,11 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   knownSparseEmbeddingFunctions.set(name, fn);
 };
 var getEmbeddingFunction = async (args) => {
-  const { collectionName, client: client2, 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 void 0;
-  }
-  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 void 0;
-  }
-  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 void 0;
-  }
-  if (efConfig.type !== "known") {
+  const { client: client2, efConfig } = args;
+  if (efConfig?.type !== "known") {
     return void 0;
   }
   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 void 0;
   }
   const packageName = pythonEmbeddingFunctions[efConfig.name] || efConfig.name;
@@ -1596,42 +1577,24 @@ var getEmbeddingFunction = async (args) => {
     } catch (error) {
     }
     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 void 0;
     }
   }
-  let constructorConfig = efConfig.type === "known" ? efConfig.config : {};
+  const constructorConfig = efConfig.config ?? {};
   try {
     if (embeddingFunction.buildFromConfig) {
       return embeddingFunction.buildFromConfig(constructorConfig, client2);
     }
-    console.warn(
-      `Embedding function ${packageName} does not define a 'buildFromConfig' function. 'add' and 'query' will fail unless you provide them embeddings directly.`
-    );
     return void 0;
   } 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 void 0;
   }
 };
-var getSparseEmbeddingFunction = async (collectionName, client2, efConfig) => {
-  if (!efConfig) {
-    return void 0;
-  }
-  if (efConfig.type === "legacy") {
-    return void 0;
-  }
-  if (efConfig.type !== "known") {
+var getSparseEmbeddingFunction = async (client2, efConfig) => {
+  if (efConfig?.type !== "known") {
     return void 0;
   }
   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 void 0;
   }
   const packageName = pythonSparseEmbeddingFunctions[efConfig.name] || efConfig.name;
@@ -1644,25 +1607,16 @@ var getSparseEmbeddingFunction = async (collectionName, client2, efConfig) => {
     } catch (error) {
     }
     if (!sparseEmbeddingFunction) {
-      console.warn(
-        `Collection ${collectionName} was created with the ${packageName} sparse embedding function. However, the @chroma-core/${packageName} package is not installed.`
-      );
       return void 0;
     }
   }
-  let constructorConfig = efConfig.type === "known" ? efConfig.config : {};
+  const constructorConfig = efConfig.config ?? {};
   try {
     if (sparseEmbeddingFunction.buildFromConfig) {
       return sparseEmbeddingFunction.buildFromConfig(constructorConfig, client2);
     }
-    console.warn(
-      `Sparse embedding function ${packageName} does not define a 'buildFromConfig' function.`
-    );
     return void 0;
   } catch (e) {
-    console.warn(
-      `Sparse embedding function ${packageName} failed to build with config: ${constructorConfig}. Error: ${e}`
-    );
     return void 0;
   }
 };
@@ -1788,7 +1742,6 @@ var processUpdateCollectionConfig = async ({
     );
   }
   const embeddingFunction = currentEmbeddingFunction || await getEmbeddingFunction({
-    collectionName,
     client: client2,
     efConfig: currentConfiguration.embeddingFunction ?? void 0
   });
@@ -3886,7 +3839,6 @@ var Schema = class _Schema {
       spann: json.spann ? cloneObject(json.spann) : null
     });
     config.embeddingFunction = await getEmbeddingFunction({
-      collectionName: "schema deserialization",
       client: client2,
       efConfig: json.embedding_function
     });
@@ -3901,7 +3853,6 @@ var Schema = class _Schema {
       bm25: typeof json.bm25 === "boolean" ? json.bm25 : null
     });
     const embeddingFunction = await getSparseEmbeddingFunction(
-      "schema deserialization",
       client2,
       json.embedding_function
     ) ?? config.embeddingFunction ?? void 0;
@@ -3987,7 +3938,7 @@ var CollectionImpl = class _CollectionImpl {
     const embeddingFunction = this._embeddingFunction ?? this.getSchemaEmbeddingFunction();
     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.`
       );
     }
     if (isQuery && embeddingFunction.generateForQueries) {
@@ -4048,10 +3999,8 @@ var CollectionImpl = class _CollectionImpl {
           }
           if (index < documentsList.length) {
             const doc = documentsList[index];
-            if (typeof doc === "string") {
-              inputs.push(doc);
-              positions.push(index);
-            }
+            inputs.push(doc);
+            positions.push(index);
           }
         });
         if (inputs.length === 0) {
@@ -4918,7 +4867,6 @@ var ChromaClient = class {
         );
         const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
         const resolvedEmbeddingFunction = await getEmbeddingFunction({
-          collectionName: collection.name,
           client: this,
           efConfig: collection.configuration_json.embedding_function ?? void 0
         }) ?? schemaEmbeddingFunction;
@@ -4988,7 +4936,6 @@ var ChromaClient = class {
     );
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(serverSchema);
     const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
-      collectionName: data.name,
       client: this,
       efConfig: data.configuration_json.embedding_function ?? void 0
     }) ?? schemaEmbeddingFunction;
@@ -5024,7 +4971,6 @@ var ChromaClient = class {
     const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
     const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
-      collectionName: data.name,
       client: this,
       efConfig: data.configuration_json.embedding_function ?? void 0
     }) ?? schemaEmbeddingFunction;
@@ -5055,7 +5001,6 @@ var ChromaClient = class {
     const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
     const resolvedEmbeddingFunction = await getEmbeddingFunction({
-      collectionName: data.name,
       efConfig: data.configuration_json.embedding_function ?? void 0,
       client: this
     }) ?? schemaEmbeddingFunction;
@@ -5131,7 +5076,6 @@ var ChromaClient = class {
     );
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(serverSchema);
     const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
-      collectionName: name,
       efConfig: data.configuration_json.embedding_function ?? void 0,
       client: this
     }) ?? schemaEmbeddingFunction;