feature: Quantize Vectors env flag

devflowinc · Apr 5, 2024 · e6b9f8a · e6b9f8a
1 parent afde22f
commit e6b9f8a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/.env.server b/.env.server
@@ -36,3 +36,4 @@ BASE_SERVER_URL="http://localhost:8090"
 UNLIMITED="true"
 REDIS_CONNECTIONS=30
 SENTRY_URL="http://********************.ingest.sentry.io/******",
+QUANTIZE_VECTORS="false"
diff --git a/server/src/lib.rs b/server/src/lib.rs
@@ -400,7 +400,12 @@ pub async fn main() -> std::io::Result<()> {
 
     let oidc_client = build_oidc_client().await;
 
-    let _ = create_new_qdrant_collection_query(None, None, None, false)
+    let quantize_vectors = std::env::var("QUANTIZE_VECTORS")
+        .unwrap_or("false".to_string())
+        .parse()
+        .unwrap_or(false);
+
+    let _ = create_new_qdrant_collection_query(None, None, None, quantize_vectors)
         .await
         .map_err(|err| {
             log::error!("Failed to create qdrant group: {:?}", err);