fix: azure flag check

This commit is contained in:
Krrish Dholakia 2023-10-05 22:44:41 -07:00
parent 97900b6b68
commit 6a572cdf1c
2 changed files with 1 additions and 1 deletions

View file

@ -239,7 +239,7 @@ def completion(
] # update the model to the actual value if an alias has been passed in
model_response = ModelResponse()
if kwargs['azure'] == True: # don't remove flag check, to remain backwards compatible for repos like Codium
if kwargs.get('azure', False) == True: # don't remove flag check, to remain backwards compatible for repos like Codium
custom_llm_provider="azure"
if deployment_id != None: # azure llms
model=deployment_id