[ENH] A chroma-load that is limited to 4 connections. (#3394)

chroma-core · Jan 3, 2025 · bde75c8 · bde75c8
1 parent f488d7d
commit bde75c8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/rust/load/Cargo.toml b/rust/load/Cargo.toml
@@ -24,7 +24,7 @@ opentelemetry_sdk = { workspace = true }
 
 # Unlikely to be used in the workspace.
 axum = "0.7"
-chromadb = { git = "https://github.com/rescrv/chromadb-rs", rev = "70c79feb220c3f75c345916f6dfa073b88f8f16a" }
+chromadb = { git = "https://github.com/rescrv/chromadb-rs", rev = "4017daf7c5c601a095b5c509bfddabc6125e6616" }
 guacamole = { version = "0.9", default-features = false }
 tower-http = { version = "0.6.2", features = ["trace"] }
 reqwest = { version = "0.12", features = ["json"] }

diff --git a/rust/load/src/lib.rs b/rust/load/src/lib.rs
@@ -129,6 +129,7 @@ pub async fn client_for_url(url: String) -> ChromaClient {
                 header: ChromaTokenHeader::XChromaToken,
             },
             database: "hf-tiny-stories".to_string(),
+            connections: 4,
         })
         .await
         .unwrap()
@@ -137,6 +138,7 @@ pub async fn client_for_url(url: String) -> ChromaClient {
             url: Some(url),
             auth: ChromaAuthMethod::None,
             database: "hf-tiny-stories".to_string(),
+            connections: 4,
         })
         .await
         .unwrap()