From 733ccec70b451b58867a36fa5570e25103813291 Mon Sep 17 00:00:00 2001
From: Jack McCluskey <thejackmccluskey@gmail.com>
Date: Fri, 27 Jun 2025 10:41:59 -0400
Subject: [PATCH] Fix Vertex text embedding model type

---
 sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py b/sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py
index 91450e4d8d9a..a75e1c3c9e1a 100644
--- a/sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py
+++ b/sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py
@@ -115,7 +115,7 @@ def __init__(
   def get_request(
       self,
       text_batch: Sequence[TextEmbeddingInput],
-      model: MultiModalEmbeddingModel,
+      model: TextEmbeddingModel,
       throttle_delay_secs: int):
     while self.throttler.throttle_request(time.time() * _MSEC_TO_SEC):
       LOGGER.info(