embeddings: Fix OpenAI embeddings dimensions parameter type

moawnallah · moawnallah · commit 0ebfdf8034f4 · 2026-01-19T02:38:56.000Z
diff --git a/sdks/python/apache_beam/ml/transforms/embeddings/open_ai.py b/sdks/python/apache_beam/ml/transforms/embeddings/open_ai.py
@@ -109,7 +109,7 @@ def request(
         "input": batch,
     }
     if self.dimensions:
-      kwargs["dimensions"] = [str(self.dimensions)]
+      kwargs["dimensions"] = self.dimensions
     if self.user:
       kwargs["user"] = self.user
 

Original file line number	Diff line number	Diff line change
`@@ -109,7 +109,7 @@ def request(`
`109`	`109`	`"input": batch,`
`110`	`110`	`}`
`111`	`111`	`if self.dimensions:`
`112`		`- kwargs["dimensions"] = [str(self.dimensions)]`
	`112`	`+ kwargs["dimensions"] = self.dimensions`
`113`	`113`	`if self.user:`
`114`	`114`	`kwargs["user"] = self.user`
`115`	`115`