chromadb 3.1.2 → 3.1.4

package/dist/chromadb.legacy-esm.js

@@ -1354,6 +1354,9 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     return void 0;
   }
   const packageName = pythonEmbeddingFunctions[efConfig.name] || efConfig.name;
+  if (packageName === "default-embed") {
+    await getDefaultEFConfig();
+  }
   let embeddingFunction = knownEmbeddingFunctions.get(packageName);
   if (!embeddingFunction) {
     try {
@@ -1364,7 +1367,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 +1462,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 +3303,10 @@ var Schema = class _Schema {
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    const embeddingFunction = await getEmbeddingFunction(
+    config.embeddingFunction = await getEmbeddingFunction(
       "schema deserialization",
       json.embedding_function
     );
-    config.embeddingFunction = embeddingFunction;
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
     }
@@ -4010,9 +4012,7 @@ var chromaFetch = async (input, init) => {
     case 429:
       throw new ChromaRateLimitError("Rate limit exceeded");
   }
-  throw new ChromaConnectionError(
-    `Unable to connect to the chromadb server (status: ${response.status}). Please try again later.`
-  );
+  throw new ChromaServerError(`${response.status}: ${response.statusText}`);
 };
 
 // src/admin-client.ts

package/dist/chromadb.mjs

@@ -1354,6 +1354,9 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     return void 0;
   }
   const packageName = pythonEmbeddingFunctions[efConfig.name] || efConfig.name;
+  if (packageName === "default-embed") {
+    await getDefaultEFConfig();
+  }
   let embeddingFunction = knownEmbeddingFunctions.get(packageName);
   if (!embeddingFunction) {
     try {
@@ -1364,7 +1367,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 +1462,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 +3303,10 @@ var Schema = class _Schema {
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    const embeddingFunction = await getEmbeddingFunction(
+    config.embeddingFunction = await getEmbeddingFunction(
       "schema deserialization",
       json.embedding_function
     );
-    config.embeddingFunction = embeddingFunction;
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
     }
@@ -4010,9 +4012,7 @@ var chromaFetch = async (input, init) => {
     case 429:
       throw new ChromaRateLimitError("Rate limit exceeded");
   }
-  throw new ChromaConnectionError(
-    `Unable to connect to the chromadb server (status: ${response.status}). Please try again later.`
-  );
+  throw new ChromaServerError(`${response.status}: ${response.statusText}`);
 };
 
 // src/admin-client.ts