fix: mypy linting

2025-10-11 09:02:42 -07:00 · 2025-10-11 09:02:42 -07:00 · 97f0daa22b
commit 97f0daa22b
parent 8dc5960332
1 changed files with 2 additions and 0 deletions
--- 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