From f134de1287dfaf0169f709b9e785efb5dbe97671 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 18 Sep 2023 12:54:56 -0700
Subject: [PATCH] bug fix for litellm proxy  implementation

---
 litellm/__pycache__/main.cpython-311.pyc | Bin 41221 -> 41230 bytes
 litellm/main.py                          |   5 +++--
 pyproject.toml                           |   2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc
index 847e9e656857d856f63430d446a53ab29f1ccbdc..64cdd8e27b1e60841d152113ae9e85281f13c61f 100644
GIT binary patch
delta 795
zcmYk2Ur19?9LImZd*?Rq-SzHnZg<no(G+y1W~h`>i+h(fO|PVpFA6~hz7$!YgIo(r
z@L`FeffCh={qv!S^CZ#3d`n1u36>B+y#>V?NJ0<QJ;#T-9M1Ri`Tfp6?##L{w=O8p
z6h#92+gWvAT6wK>>6jsDy-^5AW72eD60SMN<g9<plMPQNvKqM^?r6hTWUX)6fgcx!
z&b~o`sDmov42ca{h&$xUxwkI-vCuTqhU4VIh=@_Kyl!da`RI1E%N|Aj@NBvmZ`}68
ziox31)8xkJ(@-u1YdxA_g<wNY=`7#ZHdr3ml#$7%M;cJY&0zb0ByLG%HgBo*tg&^Z
zyN>0DNOv{Mouek-{mE-8%s<zd2UvcAbOXz8iIkV+_bxNel16!(=1w>!98ePZZ6${`
z5fDp(RE#az?J%7ze;#v8A?yKPvWMD~Pu%mR%314IB$Z_OyV~HV{z)2P<^>~U1UQ&2
zrpn=DvYS7o&|9Bs<rOO=8J5;57Oc3>u-ffxFEgx8(mJ(_SVlH71qYrY`K*pOL{@Iv
zMU^CP4Z3ld+?%~B;sYY&?cNhob@wIW3-WI6krN3Szi+rPv#|QGKbKKZqVR7n$`w@P
z>IBtKT^!xut)%4~C8*S?%+UirohlqFz@t%>qZiyV)j0Y<(Wswe5VRoGIfg)w&@jgc
z<g@^da;^dU<`~BYsEjP77LU^e*G8xa&R@)r(t})^z*n_YyBMX-Tw6fZ=db54(?eX-
z2(o{^E#FLgkQaIp^lEx$U-}UII_=+=^9asnKONq;|NcSAB?!VU*i%Ua!F?x4-TbIY
J5`X*^{{kmZ-nak&

delta 769
zcmYL_Pe>F|9LL{pXI*D!cXoGn))`mVOfCFh=4O`ohnZ<<H66)9FM4XDQ;;TZhip9<
zdDui)1r<tyC@%%|P7xl~TR|#Cumus(sUUbl7<8!S{UjQO`F=j{`~J+#{HE*Hrb~V#
z%Mx*}m-od>Gf(9X9VTf~Z*VDMO1e=$MJC)S-?W<YPe<wPXj>ef(2f2THxzOs=UxC%
zbtnihOKZ;C0PpF<g^gnPnrj)2!)2Np6`<7RZCoB*AKR(uaPESBY%W=-ZrJfx(%G?x
zr5TMp?+qHZMqYU>o#*SSI$I)JUf{ANNrSSm6*)dA3fp3l!`pH#FSgqfhj_j#OO*5c
zaKgmZ9|NY${Zq`W;`zBdS;zBRtrg<={iIpVla*u5GF|R*7uf|oZP%rYtHQ2oRq`o6
z;--h^UWa?ChwOFZl)WA$9TI+qEHAHqm08_9e@6}6^-re}<(@Y*qXfZhGV$y3r#ln)
zBKf|<Rtr|PCK<NUF68ZlVZ&~9vbD&t+v%zF3V@rwy`FbNf-X(#07LZgO=mDb&&&*o
zFicnHt_rY9eOae_Kx^(i0a&MR=I?pn8696TL`di8-GNM628ogX&w?+&1XK?Rs;n4M
zB%uiNB1$BpGasU#gmoq(28dr_LBtRdeM~_N6Io#@qDB;r>4>F7kFh9XjAWD&R)Ktg
zI44&k){#hT*;?#k^{5Rbpe<a?j<H76CK4`Nu34;L&8SC6P+z!~y~2*7CV}|Wg?P4^
r^#Ts`0ny9Z*?s8;qUvm5UxtBr6qSwa+rNL13$AG%TscV_*S^4C9X{C0

diff --git a/litellm/main.py b/litellm/main.py
index 135f81ce0..f79323236 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -234,10 +234,11 @@ def completion(
             custom_llm_provider = model.split("/", 1)[0]
             model = model.split("/", 1)[1]
         model, custom_llm_provider = get_llm_provider(model=model, custom_llm_provider=custom_llm_provider)
-        api_key = get_api_key(llm_provider=custom_llm_provider, dynamic_api_key=api_key) # get the api key from the environment if required for the model
-        if api_key and "sk-litellm" in api_key:
+        model_api_key = get_api_key(llm_provider=custom_llm_provider, dynamic_api_key=api_key) # get the api key from the environment if required for the model
+        if model_api_key and "sk-litellm" in model_api_key:
             api_base = "https://proxy.litellm.ai"
             custom_llm_provider = "openai" 
+            api_key = model_api_key
         # check if user passed in any of the OpenAI optional params
         optional_params = get_optional_params(
             functions=functions,
diff --git a/pyproject.toml b/pyproject.toml
index 012df37bf..20f9cffdc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.694"
+version = "0.1.695"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"