diff --git a/litellm/llms/azure_ai/embed/handler.py b/litellm/llms/azure_ai/embed/handler.py
index da39c5f3b8..13b8cc4cf2 100644
--- a/litellm/llms/azure_ai/embed/handler.py
+++ b/litellm/llms/azure_ai/embed/handler.py
@@ -210,6 +210,7 @@ class AzureAIEmbedding(OpenAIChatCompletion):
         client=None,
         aembedding=None,
         max_retries: Optional[int] = None,
+        shared_session=None,
     ) -> EmbeddingResponse:
         """
         - Separate image url from text
@@ -275,6 +276,7 @@ class AzureAIEmbedding(OpenAIChatCompletion):
                     else None
                 ),
                 aembedding=aembedding,
+                shared_session=shared_session,
             )
 
             text_embedding_responses = response.data