feature: add timings to all search routes in chunk_handler

Author: skeptrunedev

server/src/bin/ingestion-microservice.rs

@@ -15,7 +15,7 @@ use trieve_server::operators::chunk_operator::{
     update_chunk_metadata_query,
 };
 use trieve_server::operators::event_operator::create_event_query;
-use trieve_server::operators::model_operator::{create_embeddings, get_splade_embedding};
+use trieve_server::operators::model_operator::{create_embeddings, get_sparse_vector};
 use trieve_server::operators::parse_operator::{
     average_embeddings, coarse_doc_chunker, convert_html_to_text,
 };
@@ -420,7 +420,7 @@ async fn upload_chunk(
     };
 
     let splade_vector = if dataset_config.FULLTEXT_ENABLED {
-        match get_splade_embedding(&content.clone(), "doc").await {
+        match get_sparse_vector(&content.clone(), "doc").await {
             Ok(v) => v,
             Err(_) => vec![(0, 0.0)],
         }
@@ -606,7 +606,7 @@ async fn update_chunk(
         .map_err(|_| ServiceError::BadRequest("chunk not found".into()))?;
 
     let splade_vector = if server_dataset_config.FULLTEXT_ENABLED {
-        match get_splade_embedding(&content, "doc").await {
+        match get_sparse_vector(&content, "doc").await {
             Ok(v) => v,
             Err(_) => vec![(0, 0.0)],
         }

server/src/handlers/chunk_handler.rs

@@ -1286,6 +1286,7 @@ pub async fn search_chunk(
                 pool,
                 dataset_org_plan_sub.dataset,
                 server_dataset_config,
+                &mut timer,
             )
             .await?
         }
@@ -1297,6 +1298,7 @@ pub async fn search_chunk(
                 pool,
                 dataset_org_plan_sub.dataset,
                 server_dataset_config,
+                &mut timer,
             )
             .await?
         }
@@ -1307,8 +1309,8 @@ pub async fn search_chunk(
                 page,
                 pool,
                 dataset_org_plan_sub.dataset,
-                &mut timer,
                 server_dataset_config,
+                &mut timer,
             )
             .await?
         }

server/src/operators/model_operator.rs

@@ -164,7 +164,7 @@ pub struct CustomSparseEmbedData {
 }
 
 #[tracing::instrument]
-pub async fn get_splade_embedding(
+pub async fn get_sparse_vector(
     message: &str,
     embed_type: &str,
 ) -> Result<Vec<(u32, f32)>, ServiceError> {