refactor(azure.py): fix linting errors

This commit is contained in:
Krrish Dholakia 2023-11-08 19:24:53 -08:00
parent 6f4707bbb3
commit 7c5fd5824b
2 changed files with 7 additions and 3 deletions

View file

@ -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,):

View file

@ -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