diff --git a/litellm/llms/azure.py b/litellm/llms/azure.py
index 5285aaebe0f..b1a3df7383a 100644
--- a/litellm/llms/azure.py
+++ b/litellm/llms/azure.py
@@ -181,9 +181,9 @@ class AzureChatCompletion(BaseLLM):
     def embedding(self,
                 model: str,
                 input: list,
-                api_key: Optional[str] = None,
-                api_base: Optional[str] = None,
-                api_version: Optional[str] = None,
+                api_key: str,
+                api_base: str,
+                api_version: str,
                 logging_obj=None,
                 model_response=None,
                 optional_params=None,):
diff --git a/litellm/llms/base.py b/litellm/llms/base.py
index 300b3f88e31..cc03be91972 100644
--- a/litellm/llms/base.py
+++ b/litellm/llms/base.py
@@ -16,10 +16,14 @@ class BaseLLM:
 
     def completion(
         self,
+        *args, 
+        **kwargs
     ):  # logic for parsing in - calling - parsing out model completion calls
         pass
 
     def embedding(
         self,
+        *args, 
+        **kwargs
     ):  # logic for parsing in - calling - parsing out model embedding calls
         pass