From faa78ad5434bcc2558bc3d06956744e42eefff11 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 5 Sep 2023 14:30:03 -0700
Subject: [PATCH] bump version

---
 litellm/__pycache__/main.cpython-311.pyc | Bin 28982 -> 29001 bytes
 litellm/main.py                          |   2 ++
 litellm/tests/test_completion.py         |  15 +++++++++++++--
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc
index adb45149c93427b0a44a8cd3be029af9e0f8d8bf..a9e7a21749e306b4c572b5e36ac659eaa73f8fa0 100644
GIT binary patch
delta 459
zcmYk#Pbh<N7zgm@dFS1`&E8>~jVSrIMMD0**@n%wviW=QCl1O=(zN724ZEt5`Xv|Q
zAj(a@lIwfJfjOCrGD$AH2lMB|emi*%&*yo*&*3>s;bjWNV^P#Wrd)epp^a1VS<swU
z1MI*elEiosF=}L!5aL#>QNsckHLSq=LKF^of})8mYt59`KH9O4zBG0s&Qe>`1&2I+
zZ`ni)k=3?cjFFP|drXmZ`!5=3TH00@#;pJ`NRJ@sjq6feig^{zNA|nsta@!nvZ5xT
zFH7c?Du7~D(o(s_e^X00XI+!vu(!e;kl;%2ip=Xm`+y6sotRx*x*<^A2rYvS)s2JC
z%n7iuL&ENNGViMV<;(}Q513Da$IN^#?X@u9ntct-k4le*`B@hzV9MQ{J#452*_-dF
z&PM(=Z}zDikaX@~u}ThdMt9Jh(adLdDv#D9wo+y6o5McxI`M-}`ZT#D;4qyGO{j32
zp3j^h?$hbnJH#XEn7hrvFcBjzTR0bW3jFJR2x^nkkN+YFlSss(&L&eN9y$2~ac_#$

delta 429
zcmX^4h;iE^M&9MTyj%=G5D@z<#d{;~Q+BQmstgR%8A=!#7$!TYicJ>cILOMJ!c;51
z`2h!?2BX&I_Xc*1jOm;8j7~5yGH?EF(#Ob{G`Y=e4rAS9W%HYi4<`RL|If&?`LD%P
zHql0L28ITPuVIV~JVKooJr)fvADEmcFSpO;;o`Kqp`>|%M;?eaYdSEA*PAdb@sxA)
zGGREy?7|D8&k3sgSu&inU<7k4x%_RI&oi6&n=_v`X9cls)Ibs!l-+DV^mPR{MG$?D
z$&~>_KW26k0ntw-JjFotJAGGu5dB5oi33D`)9?}m(FYx@oIwmNhJ!9VPO4xbHzOCY
zr3bxCxE#GjnGbr|2|0QTFdy_5U<L9wUkTS^WHjE)9QlKZ(QERb=%0*Mo4>_WurvB^
z&QFPEW}LkFbower#<`ozGOsf-F4^pteUXJRd$K}-t$H>$Bi9E8B+`eGflc^>FC&r=
SP<rym0#P=wP|W1{1*-sjM}U6-

diff --git a/litellm/main.py b/litellm/main.py
index 6042a5eaa1..73d8c7c472 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -118,6 +118,8 @@ def completion(
         model_response = ModelResponse()
         if azure:  # this flag is deprecated, remove once notebooks are also updated.
             custom_llm_provider = "azure"
+            if deployment_id:
+                model=deployment_id
         elif (
             model.split("/", 1)[0] in litellm.provider_list
         ):  # allow custom provider to be passed in via the model name "azure/chatgpt-test"
diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index caabe467e2..ae59af2044 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -337,6 +337,19 @@ def test_completion_azure():
         pytest.fail(f"Error occurred: {e}")
 
 
+def test_completion_azure_deployment_id():
+    try:
+        response = completion(
+            model="chatgpt-3.5-turbo",
+            deployment_id="chatgpt-v-2",
+            messages=messages,
+            azure=True,
+        )
+        # Add any assertions here to check the response
+        print(response)
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+
 # Replicate API endpoints are unstable -> throw random CUDA errors -> this means our tests can fail even if our tests weren't incorrect.
 def test_completion_replicate_llama_stream():
     model_name = "replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1"
@@ -410,8 +423,6 @@ def test_customprompt_together_ai():
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
 
-test_customprompt_together_ai()
-
 def test_completion_sagemaker():
     try:
         response = completion(