feature: add ignore get total page count

devflowinc · May 7, 2024 · e401d48 · e401d48
1 parent f8297ab
commit e401d48
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 2 deletions.
diff --git a/server/kubectl b/server/kubectl
diff --git a/server/src/handlers/chunk_handler.rs b/server/src/handlers/chunk_handler.rs
@@ -1169,6 +1169,7 @@ impl Default for SearchChunkData {
             get_collisions: None,
             highlight_results: None,
             highlight_delimiters: None,
+            quote_negated_words: None,
             score_threshold: None,
             slim_chunks: None,
         }

diff --git a/server/src/handlers/message_handler.rs b/server/src/handlers/message_handler.rs
@@ -666,6 +666,7 @@ pub async fn stream_response(
     let search_chunk_query_results = retrieve_qdrant_points_query(
         VectorType::Dense(embedding_vector),
         1,
+        false,
         n_retrievals_to_include.try_into().unwrap(),
         None,
         None,
@@ -675,7 +676,6 @@ pub async fn stream_response(
             negated_words: None,
         },
         dataset.id,
-        false,
         pool.clone(),
         config,
     )

diff --git a/server/src/operators/search_operator.rs b/server/src/operators/search_operator.rs
@@ -1320,12 +1320,12 @@ pub async fn search_hybrid_chunks(
     let search_chunk_query_results = retrieve_qdrant_points_query(
         VectorType::Dense(embedding_vector),
         page,
+        get_total_pages,
         data.page_size.unwrap_or(10),
         data.score_threshold,
         data.filters.clone(),
         parsed_query.clone(),
         dataset.id,
-        get_total_pages,
         pool.clone(),
         config.clone(),
     );