Update router/src/http/server.rs

Co-authored-by: OlivierDehaene <Olivier.dehaene@gmail.com>
huggingface · Mar 5, 2024 · da4a2d1 · da4a2d1
1 parent 4f9a392
commit da4a2d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/router/src/http/server.rs b/router/src/http/server.rs
@@ -1052,7 +1052,7 @@ async fn vertex_compatibility(
     let results = join_all(futures)
         .await
         .into_iter()
-        .collect::<Result<Vec<AllEmbeddingsInferResponse>, TextEmbeddingsError>>()
+        .collect::<Result<Vec<PooledEmbeddingsInferResponse>, TextEmbeddingsError>>()
         .map_err(ErrorResponse::from)?;
 
     let mut embeddings = Vec::with_capacity(batch_size);