refactor(azure.py): working azure completion calls with openai v1 sdk

This commit is contained in:
Krrish Dholakia 2023-11-11 16:44:39 -08:00
parent d0bd932b3c
commit 39c2597c33
9 changed files with 70 additions and 58 deletions

View file

@ -447,9 +447,7 @@ def completion(
logging_obj=logging,
acompletion=acompletion
)
if optional_params.get("stream", False) and acompletion is False:
response = CustomStreamWrapper(response, model, custom_llm_provider=custom_llm_provider, logging_obj=logging)
return response
## LOGGING
logging.post_call(
input=messages,