From eb33e1fa719dda41856747f2ebb83a5ea33f763f Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 30 Aug 2023 09:55:33 -0700
Subject: [PATCH] adding testing to make sure litedebugger is opt-in only

---
 litellm/__pycache__/__init__.cpython-311.pyc | Bin 7679 -> 7745 bytes
 litellm/__pycache__/utils.cpython-311.pyc    | Bin 75648 -> 75945 bytes
 litellm/tests/test_completion.py             |  19 ++++++++++---------
 litellm/utils.py                             |   2 +-
 4 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/litellm/__pycache__/__init__.cpython-311.pyc b/litellm/__pycache__/__init__.cpython-311.pyc
index 5c001e9abd456c48f9fa455fedbc8ac1f128951f..61bd76f5967023caceffc0483334eae34518abad 100644
GIT binary patch
delta 447
zcmexweb9z?IWI340}$vWyib|6k(ZyH@z-Wqc0VR@W1!$lh9VOnRb&Jr^uWaACQbu3
zu%O=L9h~8e29pK2#B~gSa+-{{cvH(#^Gf1VQcDsubBc=@fuagv&9^u*Q-Cs=X_ZAb
zlY_ZT4M8Soau=zC%+vr9Rb1%>CAucM2F8XaMVcT1K8S#^Q8rj`@?0(zk0LdYHnu8(
zlGKV4-IT<#%)I1GT>}H-D$eB6qRdnv)3Ay=DKRN2F&)BTPfSS!i5AI%Y*GYiXD!m1
z{FuvwO&cVlwb_unn{DzzK}k_J5Z?<#xPu4}5aGG`m0&d!H`wb|AOV}r1tOCeMSOwm
zTkIvN#U+_}>9<%*OMu>;EG=flR1`VcOH9Qg3Zym~M8trISP&5hBH}@W1&DwL>@N<R
w-29Z%oK(A_&dtljcCpCtDlvUvfDsoM<UYtSC{;6FP^<!?%{9{XOe`SH0FO{@0ssI2

delta 362
zcmX?T^WU0xIWI340}z<Ize_RP$ji^pShZP}-H%D!2q?Icp~x6W6&ZpET`)1ZiPL}$
zET}tq2WL2={$v3zaUFf2oF?Nf-qf<xyps5o)RM%^oZ@0epr`^^^DU0d6rfCIT4j;-
z<X|pSQIH9m97Sp%Gu1(a29TJ%lS@TOletQuB(<VMHzlzwGcP$)*TBHINEW0-5lCpV
z7HLmb<@R9H0*PpDF6QoLV{rhoC;t_c67U2Gx`7CH5aF>|N2r>K8|-jPkbw2(O(K&R
zC4GSGTkIvN#U+_}>9<%*OMs3o@&gJ^ju11N+%Klc5((n07n71Q2Wf+c!7mP*-29Z%
soK(A_j?FK`cCpBCDKULufDsoM<UWWoD1|UyPz(m5&HJS5nOH!Y0RzHTMgRZ+

diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc
index 36517095ba7e35980ddf5be1559d7713fb7e6f5b..b64c5c73240a804f9b690c45cf460523a8c32362 100644
GIT binary patch
delta 1941
zcmZ`%ZA?>F7(Va4{i0Y}T3SYJ*D9s8e26j;%KQMGo6Lv`bEr$Kq_-+6AG^IOvQC-0
z#KkRhJ9Ae>$@WJ=G)uOdrCDaOX=b)*L<H&tHF26PqT48HNW%bkP77!zp5{J1=RME!
zzVCU?X}T<HyCT!HYBXv@#)h^1+@Y^EA6x)N5--dHwq4NR;w5G&a&b8WB3{s50z0Ip
zb|bJ%?LB<t@*QnMkm1s|GE~i<;2ij?j^z-*!<{b3z<*tF;H6y}?7d_rtClOvW3jAM
z=NSLC;HJ9x;9S@~l!6y^ro#-pv2&Wy9fU2K@!e*0f(atdiH~<KlxLD1Egm@kA|CC`
zFg(K}tl~>H$if>gr9CIhhOlQ+T}M|cWaGPCvo$$E#SX-~xS2r(mxG<%wt`$bOB-(p
zjdhfLT7Hs6=oI<lV=r)Vt&^J-(+0b3`IBh5?9=*_#J5v1@2GY{>EdR`lpWo+l1Y@r
zhsmnu#X5EqW!{r6xtH*)a9ZkfvS8{%?*Dz}vWb5w_{VU8a?Wuks63sIN5d5`7nfgM
z1_ij~>MAIdB3IKGlX08Bc9XE#zOA~ZTCn>Ye11RY=kUUw@}x#hK&TS72kgE&&R--Q
z>Y2`He@;NC3Ei?eI%JjzvkRZSZU8?HUq1^gB=N>n8UFRwV%CssmNI`oC^N5iug*c?
z9ChUaHe9fXsr0bFdE{i_W5xFXQt+KWB9JR}|0ywsLIpyBH9)gp(@k$GjWEghSl=;7
zlJf6eU|_{Jf6rxLGcM?F)~zQ-i^z}G0f*KxoEG2iU!1JjkJcq|Nd&bVR3Nub%8bkd
zcn5D9*avO+{(wjKJ}L1~NyMat)^H#Ze?O3G<%dbl0Qr!e&m`EJ;PqBj3x3|~ZM2GI
zN-(94Ee{YmKT4Q<7UB6A{I(_quXxZ52c`QDx&S_wet5JV7?#1dLF)?&@+Nv^8>?$M
z{ue@bD|i?A<5l>L!H?i`+!wV-ql2rUge5#)MnO*D93>4=P?Jh-8Bg=@N(v!bx`#j;
zJC=(^)%Z|!4SXr}MJr>q7Y(g6#xxUjzMiHt;?|+p6{o1nKKxJAba*#tvE|=uDQP5&
z8)Bq3X1qzyhz+}j%Rs_AhR=Zodq!&DJE?WVz|ixZd2H8ES-eLTufTneXMpjr9#$&P
z&<a|>{ZZq$yTPmgn|M?K_W6mFAEuoc$gEf8_15|u1h1E$MUNFbv58U~6wDM_=!TWf
z?BbvT=737fQi2s+VyP0Q!Nt&4CA2creS|hKWMsCFd=FO=ScZVSlLUn8TP=GkP&yJw
zvhm>{cL7MCg0Zm|cA^G$E!qkErzEGF<(;(n3<ACS3ZbsPB(SU2S4y5nl~7$*JL$uT
zpQ+KhczZyotEmupe^pH-UsYc(dX*sGQx)GIq#hNLvUka^2}Kw-A>*bY^@!|~qR)yt
zrdXplwU!8igp4O}Q}5`Q;)t*$7>z~)Dk9#WWI$M-!Md5UGPjP;CLunD!b}Pd3M14)
zL{IS(UAkkJK>g=qUdL>XJ4|CS>s@q~Bfg~ui=vVCc}IL#4SC`hYG_V-6J&DI)UyzJ
zB@y~To=?Q1p+XICGV)HumSk{}`#+xyrJxYCdJqy+gwYFIc(?eW9_GtmB}-kXT@MW`
zx!%`IkS;GJp}bI?3EVPRCLT4z%kpv(@`QTKu!TvSOXhq5g~Cvo6{fN}T9l`Wcq-+~
z#a(vtN;ZW=J0vQKf-I1ZpTc650+v7oJ%jobFh$S=%oJ}p;l<g-4AoDL1F`s{r+1gi
J#4H!A{twd%`-}hp

delta 1785
zcmZ`%ZA?>F7(Va0w=GZ$g;KD5Nae#)zEKoVr=sqM0?rR+h(AQe?HDNN4)tT&IpUIK
zE;BKnZP(0A=aQ*JTw-pvCQG(BGdHJEDI!rE%C>BY$%Yqaz@d(v_kv6vp0>|D=RME!
zydSr1A-VOEY&&YRMG&$6+10O9owV&f2UVJSZZ1eeej8u#e1fK*ucUB_&%0O#hqTs<
zPLQN#2j73`*W|+is-&%_sAk{$z4#nmN_s%%ubzpeN(O(Yqmu9Fj5JLPKw}giI5UN}
zcV>WRU#J{BydqSI+jRa-$l}Tzv-;%;D{laVm6QJ!azhr6?VRe&7EZQDsSc2jX@Dr1
zd}-(WCv!Od#}ljt7g8XHue;!ShReE8>4dtSm*XLqe{i|5E-zr-LYPO%515ra*P}9s
zh$SoZb;%z+9wJJ?lbo?ybK6ZvBtnjh-{{%}%Bbr!r4SW7t1D&Z1X?AxJ?4lQ_i<z7
zs2x{&l<9_YXIDzu1WGZXiL%ZxdhrV7jQ=`OiujE#SNt<vI8LbG={KO{ipj?E<}1aP
znIF-BrF|A}zp@I7_jQ*;39sm00j1i(ZWo;u<zs9kb2s@~T59Wkv)$XB8+^_6{#V=^
z8k!pXZr`>#pHJ~A+iflWTL0!2cil!su=s_mE4939Q)$}Y;e@o3zLfM1IZ3BGBvVGO
z)!7l3c_KGtE$OnB=%k*r>)*l(t>~vjIlk8}E@vDI7a+bNdr_CYsKZ{wU+7!JZ~k1Y
zecs39gw-*t<4AX0M72lib}yvWsnXR{lQ`5#Jo?XAzU((03bpyaYt%8@Oh`*LU<p_4
zsjW+NYUid~AHpQ<lUwH~EZ5fmF^fV2|Fr*reWiFkD)EsuppZ3GvGK~=i%hVJcq6sL
zx90)`_znF{@bZ@iO6_l<uSE!iZ<OP0D^I@jCG6I^?py}gqaC}u7U&4@u)*ZXRJ{5g
z`Q?UYg`E|AubFwogUS54!9DPvc7Cu4%0@(ij0hqFgwseEMu-p>UYVf>lMsT)ye6Pc
z8uP)ZnZI_g8V>REgGpNdy*Wni?BV53LpLsAo6s_v?;TzWZQAu=3b9V1_JVQIMjsV6
zaK8yo@|p*yVKN`MpTf64bn?{?OChN3edwUr*JmSc8%js(3uBabj$}eKzdy1ZzWSm9
zW}0&n(K(itVajrZd{K0d-0a)tXT#Wp1@%|(CQ3bFhRF#r7;q!RAsj@K3wKHCO*0g^
zWwGxG_X~p>rPg2DA`7isES7~PX_SAhCs@$Grsl_Bl?9H{H@tM&GH)cyK|(G<9zr_8
z1GN7RcSa#mn8>KWNIseuV@5$~xJyNHHiD6G2n`T+;9miqTE{;+4Ko$%L_J`IDe74(
zyc<ruCby0dY?=hO!IUQ)2ZObdpvYpN>JKq64HDJ<7+3;SEpdQ9(r6bNSqA2QpU+#L
zq;@+X$uwV_!-7K&*d~FjZi<IA(?Su-3hs{wuMCUT(}^(0R4zh=L7D`O)S8bD1qg+~
zty3UT%ERmojTlN9lAmgCIBagyMrHE`-#oT1cxozG%`rPA$k54SxLED=fNQM-;&d_&
lISz=z;}}ez1lc+nhxDxgVh$7IfDym@`0W-+t;&QK{{=5C+DHHZ

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 6ffef16a6c..db85691254 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -72,16 +72,17 @@ def test_completion_claude_stream():
         pytest.fail(f"Error occurred: {e}")
 
 
-# def test_completion_hf_api():
-#     try:
-#         user_message = "write some code to find the sum of two numbers"
-#         messages = [{ "content": user_message,"role": "user"}]
-#         response = completion(model="stabilityai/stablecode-completion-alpha-3b-4k", messages=messages, custom_llm_provider="huggingface")
-#         # Add any assertions here to check the response
-#         print(response)
-#     except Exception as e:
-#         pytest.fail(f"Error occurred: {e}")
+def test_completion_hf_api():
+    try:
+        user_message = "write some code to find the sum of two numbers"
+        messages = [{ "content": user_message,"role": "user"}]
+        response = completion(model="stabilityai/stablecode-completion-alpha-3b-4k", messages=messages, custom_llm_provider="huggingface", logger_fn=logger_fn)
+        # Add any assertions here to check the response
+        print(response)
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
 
+test_completion_hf_api()
 # def test_completion_hf_deployed_api():
 #     try:
 #         user_message = "There's a llama in my garden 😱 What should I do?"
diff --git a/litellm/utils.py b/litellm/utils.py
index 2783de2ab0..64d4420c80 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -438,7 +438,7 @@ def client(original_function):
                 litellm.email is not None
                 or os.getenv("LITELLM_EMAIL", None) is not None
                 or litellm.token is not None
-                or os.getenv("LITELLM_TOKEN", None)
+                or os.getenv("LITELLM_TOKEN", None) is not None
             ):  # add to input, success and failure callbacks if user is using hosted product
                 get_all_keys()
                 if "lite_debugger" not in callback_list and litellm.logging: