diff --git a/python/packages/ai/teams/ai/embeddings/embeddings_response.py b/python/packages/ai/teams/ai/embeddings/embeddings_response.py
index 7928269de..d5223c472 100644
--- a/python/packages/ai/teams/ai/embeddings/embeddings_response.py
+++ b/python/packages/ai/teams/ai/embeddings/embeddings_response.py
@@ -24,7 +24,7 @@ class EmbeddingsResponse:
     status: EmbeddingsResponseStatus
     "Status of the embeddings response."
 
-    output: Optional[Union[List[List[int]], str]] = None
+    output: Optional[Union[List[List[float]], str]] = None
     "Optional. Embeddings for the given inputs or the error string."
 
     message: Optional[str] = None