chromadb 3.3.2 → 3.3.3

package/dist/cjs/chromadb.cjs

@@ -1635,7 +1635,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",
@@ -1674,32 +1676,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;
@@ -1715,42 +1696,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;
@@ -1763,25 +1726,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;
   }
 };
@@ -1907,7 +1861,6 @@ var processUpdateCollectionConfig = async ({
     );
   }
   const embeddingFunction = currentEmbeddingFunction || await getEmbeddingFunction({
-    collectionName,
     client: client2,
     efConfig: currentConfiguration.embeddingFunction ?? void 0
   });
@@ -4005,7 +3958,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
     });
@@ -4020,7 +3972,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;
@@ -4106,7 +4057,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) {
@@ -4167,10 +4118,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) {
@@ -5037,7 +4986,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;
@@ -5107,7 +5055,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;
@@ -5143,7 +5090,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;
@@ -5174,7 +5120,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;
@@ -5250,7 +5195,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;