chromadb 3.1.3 → 3.1.5

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, debug?: boolean) => Promise<EmbeddingFunction | undefined>;
+declare const getEmbeddingFunction: (collectionName: string, efConfig?: EmbeddingFunctionConfiguration) => Promise<EmbeddingFunction | undefined>;
 /**
  * Retrieves and instantiates a sparse embedding function from configuration.
  * @param collectionName - Name of the collection (for error messages)
@@ -1009,6 +1009,7 @@ declare class Schema {
     private enableAllIndexesForKey;
     private disableAllIndexesForKey;
     private validateSingleSparseVectorIndex;
+    private validateSparseVectorConfig;
     private initializeDefaults;
     private initializeKeys;
     private serializeValueTypes;

package/dist/chromadb.legacy-esm.js

@@ -1286,7 +1286,8 @@ var knownEmbeddingFunctions = /* @__PURE__ */ new Map();
 var pythonEmbeddingFunctions = {
   onnx_mini_lm_l6_v2: "default-embed",
   default: "default-embed",
-  together_ai: "together-ai"
+  together_ai: "together-ai",
+  sentence_transformer: "sentence-transformer"
 };
 var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "amazon_bedrock",
@@ -1297,7 +1298,6 @@ var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "instructor",
   "open_clip",
   "roboflow",
-  "sentence_transformer",
   "text2vec"
 ]);
 var knownSparseEmbeddingFunctions = /* @__PURE__ */ new Map();
@@ -1325,7 +1325,7 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   }
   knownSparseEmbeddingFunctions.set(name, fn);
 };
-var getEmbeddingFunction = async (collectionName, efConfig, debug) => {
+var getEmbeddingFunction = async (collectionName, efConfig) => {
   if (!efConfig) {
     console.warn(
       `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`
@@ -1354,15 +1354,7 @@ var getEmbeddingFunction = async (collectionName, efConfig, debug) => {
     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);
@@ -2762,8 +2754,9 @@ var Schema = class _Schema {
       );
     }
     if (!configProvided && keyProvided && key) {
-      this.enableAllIndexesForKey(key);
-      return this;
+      throw new Error(
+        `Cannot enable all index types for key '${key}'. Please specify a specific index configuration.`
+      );
     }
     if (configProvided && !keyProvided) {
       this.setIndexInDefaults(config, true);
@@ -2797,8 +2790,9 @@ var Schema = class _Schema {
       );
     }
     if (keyProvided && !configProvided && key) {
-      this.disableAllIndexesForKey(key);
-      return this;
+      throw new Error(
+        `Cannot disable all index types for key '${key}'. Please specify a specific index configuration.`
+      );
     }
     if (keyProvided && configProvided && key) {
       this.setIndexForKey(key, config, false);
@@ -2915,6 +2909,7 @@ var Schema = class _Schema {
   setIndexForKey(key, config, enabled) {
     if (config instanceof SparseVectorIndexConfig && enabled) {
       this.validateSingleSparseVectorIndex(key);
+      this.validateSparseVectorConfig(config);
     }
     const current = this.keys[key] = ensureValueTypes(this.keys[key]);
     if (config instanceof StringInvertedIndexConfig) {
@@ -2961,7 +2956,7 @@ var Schema = class _Schema {
       new VectorIndexType(true, new VectorIndexConfig())
     );
     current.sparseVector = new SparseVectorValueType(
-      new SparseVectorIndexType(true, new SparseVectorIndexConfig())
+      new SparseVectorIndexType(false, new SparseVectorIndexConfig())
     );
     current.intValue = new IntValueType(
       new IntInvertedIndexType(true, new IntInvertedIndexConfig())
@@ -3011,6 +3006,13 @@ var Schema = class _Schema {
       }
     }
   }
+  validateSparseVectorConfig(config) {
+    if (config.sourceKey !== null && config.sourceKey !== void 0 && !config.embeddingFunction) {
+      throw new Error(
+        `If sourceKey is provided then embeddingFunction must also be provided since there is no default embedding function. Config: ${JSON.stringify(config)}`
+      );
+    }
+  }
   initializeDefaults() {
     this.defaults.string = new StringValueType(
       new FtsIndexType(false, new FtsIndexConfig()),
@@ -3311,11 +3313,9 @@ var Schema = class _Schema {
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    console.log("SCHEMA THING 1");
     config.embeddingFunction = await getEmbeddingFunction(
       "schema deserialization",
-      json.embedding_function,
-      true
+      json.embedding_function
     );
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
@@ -4022,9 +4022,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

@@ -1286,7 +1286,8 @@ var knownEmbeddingFunctions = /* @__PURE__ */ new Map();
 var pythonEmbeddingFunctions = {
   onnx_mini_lm_l6_v2: "default-embed",
   default: "default-embed",
-  together_ai: "together-ai"
+  together_ai: "together-ai",
+  sentence_transformer: "sentence-transformer"
 };
 var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "amazon_bedrock",
@@ -1297,7 +1298,6 @@ var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "instructor",
   "open_clip",
   "roboflow",
-  "sentence_transformer",
   "text2vec"
 ]);
 var knownSparseEmbeddingFunctions = /* @__PURE__ */ new Map();
@@ -1325,7 +1325,7 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   }
   knownSparseEmbeddingFunctions.set(name, fn);
 };
-var getEmbeddingFunction = async (collectionName, efConfig, debug) => {
+var getEmbeddingFunction = async (collectionName, efConfig) => {
   if (!efConfig) {
     console.warn(
       `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`
@@ -1354,15 +1354,7 @@ var getEmbeddingFunction = async (collectionName, efConfig, debug) => {
     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);
@@ -2762,8 +2754,9 @@ var Schema = class _Schema {
       );
     }
     if (!configProvided && keyProvided && key) {
-      this.enableAllIndexesForKey(key);
-      return this;
+      throw new Error(
+        `Cannot enable all index types for key '${key}'. Please specify a specific index configuration.`
+      );
     }
     if (configProvided && !keyProvided) {
       this.setIndexInDefaults(config, true);
@@ -2797,8 +2790,9 @@ var Schema = class _Schema {
       );
     }
     if (keyProvided && !configProvided && key) {
-      this.disableAllIndexesForKey(key);
-      return this;
+      throw new Error(
+        `Cannot disable all index types for key '${key}'. Please specify a specific index configuration.`
+      );
     }
     if (keyProvided && configProvided && key) {
       this.setIndexForKey(key, config, false);
@@ -2915,6 +2909,7 @@ var Schema = class _Schema {
   setIndexForKey(key, config, enabled) {
     if (config instanceof SparseVectorIndexConfig && enabled) {
       this.validateSingleSparseVectorIndex(key);
+      this.validateSparseVectorConfig(config);
     }
     const current = this.keys[key] = ensureValueTypes(this.keys[key]);
     if (config instanceof StringInvertedIndexConfig) {
@@ -2961,7 +2956,7 @@ var Schema = class _Schema {
       new VectorIndexType(true, new VectorIndexConfig())
     );
     current.sparseVector = new SparseVectorValueType(
-      new SparseVectorIndexType(true, new SparseVectorIndexConfig())
+      new SparseVectorIndexType(false, new SparseVectorIndexConfig())
     );
     current.intValue = new IntValueType(
       new IntInvertedIndexType(true, new IntInvertedIndexConfig())
@@ -3011,6 +3006,13 @@ var Schema = class _Schema {
       }
     }
   }
+  validateSparseVectorConfig(config) {
+    if (config.sourceKey !== null && config.sourceKey !== void 0 && !config.embeddingFunction) {
+      throw new Error(
+        `If sourceKey is provided then embeddingFunction must also be provided since there is no default embedding function. Config: ${JSON.stringify(config)}`
+      );
+    }
+  }
   initializeDefaults() {
     this.defaults.string = new StringValueType(
       new FtsIndexType(false, new FtsIndexConfig()),
@@ -3311,11 +3313,9 @@ var Schema = class _Schema {
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    console.log("SCHEMA THING 1");
     config.embeddingFunction = await getEmbeddingFunction(
       "schema deserialization",
-      json.embedding_function,
-      true
+      json.embedding_function
     );
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
@@ -4022,9 +4022,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