From da4a2d1085bc088a375d8570554a06f16c672656 Mon Sep 17 00:00:00 2001
From: drbh <david.richard.holtz@gmail.com>
Date: Tue, 5 Mar 2024 18:07:41 -0500
Subject: [PATCH] Update router/src/http/server.rs

Co-authored-by: OlivierDehaene <Olivier.dehaene@gmail.com>
---
 router/src/http/server.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/router/src/http/server.rs b/router/src/http/server.rs
index b9bbf5de..e569a8ed 100644
--- 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);