From 7f3b90db3c71e0f8f839b330e9a4ba0b11f987d7 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 21 Aug 2023 14:05:24 -0700 Subject: [PATCH] fixing azure bug --- litellm/main.py | 2 +- litellm/tests/test_completion.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 51371302e..60ed6fc3c 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -164,7 +164,7 @@ def completion( ) else: response = openai.ChatCompletion.create( - model=model, messages=messages, **optional_params + engine=model, messages=messages, **optional_params ) ## LOGGING logging.post_call(input=messages, api_key=openai.api_key, original_response=response, additional_args={"headers": litellm.headers, "api_version": openai.api_version, "api_base": openai.api_base}) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index fc9954459..971726982 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -254,8 +254,7 @@ def test_completion_openai_with_functions(): def test_completion_azure(): try: response = completion( - model="gpt-3.5-turbo", - deployment_id="chatgpt-test", + model="chatgpt-test", messages=messages, custom_llm_provider="azure", )