From 7c5fd5824b8b30eb85aa579c8a9a06c82fad02e3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 8 Nov 2023 19:24:53 -0800
Subject: [PATCH] refactor(azure.py): fix linting errors

---
 litellm/llms/azure.py | 6 +++---
 litellm/llms/base.py  | 4 ++++
 2 files changed, 7 insertions(+), 3 deletions(-)

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