diff --git a/pgml-sdks/pgml/src/queries.rs b/pgml-sdks/pgml/src/queries.rs
index 1ea7001bf..3fcdb9c31 100644
--- a/pgml-sdks/pgml/src/queries.rs
+++ b/pgml-sdks/pgml/src/queries.rs
@@ -217,9 +217,9 @@ ON CONFLICT (chunk_id) DO UPDATE SET ts = EXCLUDED.ts;
 pub const GENERATE_EMBEDDINGS_FOR_CHUNK_IDS: &str = r#"
 INSERT INTO %s (chunk_id, embedding) 
 SELECT 
-  id, 
+  unnest(array_agg(id)), 
   pgml.embed(
-    text => chunk, 
+    inputs => array_agg(chunk), 
     transformer => $1, 
     kwargs => $2 
   )