From aaca89e0fe83956a265d7afefafdad3dbbf6b2f9 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 5 Aug 2023 22:43:25 -0700
Subject: [PATCH] fix test

---
 litellm/tests/test_secrets.py | 2 +-
 pyproject.toml                | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/tests/test_secrets.py b/litellm/tests/test_secrets.py
index 3123c4745..2a4141bd4 100644
--- a/litellm/tests/test_secrets.py
+++ b/litellm/tests/test_secrets.py
@@ -19,7 +19,7 @@ messages = [{ "content": user_message,"role": "user"}]
 
 def test_completion_azure():
     try:
-        response = completion(model="chatgpt-test", messages=messages, azure=True)
+        response = completion(model="gpt-3.5-turbo", deployment_id="chatgpt-test", messages=messages, azure=True)
         # Add any assertions here to check the response
         print(response)
     except Exception as e:
diff --git a/pyproject.toml b/pyproject.toml
index 9bf71df72..a83be2b3b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.350"
+version = "0.1.351"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"