chromadb 3.1.2 → 3.1.3

package/dist/chromadb.d.ts

@@ -688,7 +688,7 @@ declare const registerSparseEmbeddingFunction: (name: string, fn: SparseEmbeddin
  * @param efConfig - Configuration for the embedding function
  * @returns EmbeddingFunction instance or undefined if it cannot be constructed
  */
-declare const getEmbeddingFunction: (collectionName: string, efConfig?: EmbeddingFunctionConfiguration) => Promise<EmbeddingFunction | undefined>;
+declare const getEmbeddingFunction: (collectionName: string, efConfig?: EmbeddingFunctionConfiguration, debug?: boolean) => Promise<EmbeddingFunction | undefined>;
 /**
  * Retrieves and instantiates a sparse embedding function from configuration.
  * @param collectionName - Name of the collection (for error messages)

package/dist/chromadb.legacy-esm.js

@@ -1325,7 +1325,7 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   }
   knownSparseEmbeddingFunctions.set(name, fn);
 };
-var getEmbeddingFunction = async (collectionName, efConfig) => {
+var getEmbeddingFunction = async (collectionName, efConfig, debug) => {
   if (!efConfig) {
     console.warn(
       `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`
@@ -1354,6 +1354,17 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     return void 0;
   }
   const packageName = pythonEmbeddingFunctions[efConfig.name] || efConfig.name;
+  if (debug) {
+    console.log("DEBUFFING");
+    console.log(packageName);
+  }
+  if (packageName === "default-embed") {
+    console.log("INSIDE HIT");
+    if (debug) {
+      console.log("FROM DEBUG");
+    }
+    await getDefaultEFConfig();
+  }
   let embeddingFunction = knownEmbeddingFunctions.get(packageName);
   if (!embeddingFunction) {
     try {
@@ -1364,7 +1375,7 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     }
     if (!embeddingFunction) {
       console.warn(
-        `Collection ${collectionName} was created with the ${packageName} embedding function. However, the @chroma-core/${packageName} package is not install. 'add' and 'query' will fail unless you provide them embeddings directly, or install the @chroma-core/${packageName} package.`
+        `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;
     }
@@ -1459,8 +1470,8 @@ var serializeEmbeddingFunction = ({
 var getDefaultEFConfig = async () => {
   try {
     const { DefaultEmbeddingFunction } = await import("@chroma-core/default-embed");
-    if (!knownEmbeddingFunctions.has(new DefaultEmbeddingFunction().name)) {
-      registerEmbeddingFunction("default", DefaultEmbeddingFunction);
+    if (!knownEmbeddingFunctions.has("default-embed")) {
+      registerEmbeddingFunction("default-embed", DefaultEmbeddingFunction);
     }
   } catch (e) {
     console.warn(
@@ -3300,11 +3311,12 @@ var Schema = class _Schema {
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    const embeddingFunction = await getEmbeddingFunction(
+    console.log("SCHEMA THING 1");
+    config.embeddingFunction = await getEmbeddingFunction(
       "schema deserialization",
-      json.embedding_function
+      json.embedding_function,
+      true
     );
-    config.embeddingFunction = embeddingFunction;
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
     }

package/dist/chromadb.mjs

@@ -1325,7 +1325,7 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   }
   knownSparseEmbeddingFunctions.set(name, fn);
 };
-var getEmbeddingFunction = async (collectionName, efConfig) => {
+var getEmbeddingFunction = async (collectionName, efConfig, debug) => {
   if (!efConfig) {
     console.warn(
       `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`
@@ -1354,6 +1354,17 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     return void 0;
   }
   const packageName = pythonEmbeddingFunctions[efConfig.name] || efConfig.name;
+  if (debug) {
+    console.log("DEBUFFING");
+    console.log(packageName);
+  }
+  if (packageName === "default-embed") {
+    console.log("INSIDE HIT");
+    if (debug) {
+      console.log("FROM DEBUG");
+    }
+    await getDefaultEFConfig();
+  }
   let embeddingFunction = knownEmbeddingFunctions.get(packageName);
   if (!embeddingFunction) {
     try {
@@ -1364,7 +1375,7 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     }
     if (!embeddingFunction) {
       console.warn(
-        `Collection ${collectionName} was created with the ${packageName} embedding function. However, the @chroma-core/${packageName} package is not install. 'add' and 'query' will fail unless you provide them embeddings directly, or install the @chroma-core/${packageName} package.`
+        `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;
     }
@@ -1459,8 +1470,8 @@ var serializeEmbeddingFunction = ({
 var getDefaultEFConfig = async () => {
   try {
     const { DefaultEmbeddingFunction } = await import("@chroma-core/default-embed");
-    if (!knownEmbeddingFunctions.has(new DefaultEmbeddingFunction().name)) {
-      registerEmbeddingFunction("default", DefaultEmbeddingFunction);
+    if (!knownEmbeddingFunctions.has("default-embed")) {
+      registerEmbeddingFunction("default-embed", DefaultEmbeddingFunction);
     }
   } catch (e) {
     console.warn(
@@ -3300,11 +3311,12 @@ var Schema = class _Schema {
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    const embeddingFunction = await getEmbeddingFunction(
+    console.log("SCHEMA THING 1");
+    config.embeddingFunction = await getEmbeddingFunction(
       "schema deserialization",
-      json.embedding_function
+      json.embedding_function,
+      true
     );
-    config.embeddingFunction = embeddingFunction;
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
     }