From acccf17e44a0de579470c3cd8bbeaa9691a70ae5 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 24 Aug 2023 19:06:44 -0700 Subject: [PATCH] updates to get model list --- litellm/__pycache__/utils.cpython-311.pyc | Bin 66507 -> 66908 bytes litellm/utils.py | 9 ++++++++- pyproject.toml | 2 +- 3 files changed, 9 insertions(+), 2 deletions(-) diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc index 43dafb9b0e48e1f7720be12e03a9c73368ec1287..c2ebb09a4ce3847a032d48b2b4944a0e792802c3 100644 GIT binary patch delta 2341 zcmZXU3sh9c8OQH;AG<7Ec8RPj2+LE%6+umSsGvXq3G(m}gNnksI!n5^JaTunr7p#u zaN?tS1gANY#+;CR1Q}F^XBCurQnS5Ab7SH8#Xi^i0g00c* z0>>Sv@7PtA?X&5rZSqfSPa{JUSC)gOSm#&rD5Oa&%f-!n=-fOThX6sPfWn%fY%vO_ zUirvK`HY5RH_Atm)KY++LDBSg3HI`l!G+IrC<+O_I|c#IS&OhNn4lP^3XzQ^DA8I{ zDWjw&scb2lO>-?nb10Xixs)r=la$Y*rzlsVWXk7I3S}uOM5#?+S%uPQZ#9}nxds)X z`Sh!0XaVi5Md_65PzGhWuhxaM)bRj8PbGO2_4@$`#P4o$Eg<V7gOo`<8PFYkc_Z;=lQU*#d6*h*}?JPaJ?XrRSy z<8I-WV~wPI`3)W(lC+h{JOmSa=}Q0+9^uN9HXCE`(o9xv%w6D&FD+{P~?yQC@^&j}6ZT@r3N8>dpaR46bM zLwzEibxJcs-Qef)b51N2$5m;^V_oVD+T1;v252V&j2d5nYTSti=*HVF4e2;_5+>{I z7KhVZX?1S=nH8BWPV**fy#v>f9Pr0)lDHtSf@|H{?_TbpH%(R6)Q)-_{Rv%g0d+M3g&i*)NEug9i!Cilc< zx??jhW>DOXnSRLhM#7R^9W;q~Dct3hxWz^sbY==%#52z95RM#Ao9mR?EjDCT99}1U3-%OpYM!>!}!C_={B#lxvtKtD7n<2 zTxKrsGWeA3W-T^dWiy3It-Q>f+8JJ9IL2_C!mF>gIUKf{=goF|^~A@$qRnZocBpAA zT{oQ2HFZ)4b^bkDt5(|B#*K&}cvV*#Ou~D+jy2wjh;hdz_eP}bSn7$5cgsS|WyE}N+*z?xYr&oe_d^HS(03GokN{5wz-?T0BO>Hu>gp9MZ8b<~Wvir8V%Xg7;kFxl;2cTvRzvn} zmh%q75W_y^@DamjEJ9RR4&^RG8^b>-Ov2+lY+xAQ{BSi~#!23Avi#;W0nXwrgYf|a z%=RyaFYwVpWBQjYPV~ijwHs@z>+Dvit+vKoyP-)A|di8Cjf`ABpg2->qo4 zN@8xk1WyirpOv*(QsohYklB3RUb1LsLk1-8!>Trd|H*2U;zQ(T& zFO-65zHiCkaHF8|9fErpeHtqn=x8+A{b{H`SEPL;IpqOOV)9qibX@Ucc%5a9V&B!U zu1{@$)|yaT3BbBh!{k6VxSx4zSxd(dgcpp?g|D%FG+hd(1154}RIP$`eEMN3MBq;z zrb}^jcqvJEWaEPb79ui8%omYE-%JBIkKgYXG)lx)03h9-Qg!qg>o3-e!z_V I{XBg6e-+TcasU7T delta 2050 zcmZ9N3s96*6oCKp|NCGU*j-TxgykvNW}%X1fT_5E`2e+iqPVWyHDT72pS8)vo{VYP zjCE>Eqtfw#qM3LtQORj)D4m)@a>5od(KOSB2$~uiv(8z<9(U*e{&UYg=bn4cf9~x0 z7Yv_0Z?M%yMp_jb3ryF<(GSlJ$^1D%sNWw{2dTB0#`0O~h40alVKr&jR5j zPPQog6O%*S;)Q&KiBEQNq8{je7_P}ZtHsP#^ zr(iP<$t{BovO4!3mCO&jhpA92gZaN2;fOpl?F|(QdG{{8p5I~*OeZg=zCkrczVjx-8Z0wAN3-ZP7W6dy6eq4JQ zpag68eFO_-dEJvLl*=CuJP+^$RvavWsrco=Ly(M5zqD zw=&>a<_nZA_X&?TfGeaMELbno@Oyb@iiPZx`iGuIGadW!tU*yK&tv8h@6?3tLtE_W z&Gz*6#MGKKTi4dC-5hbw>3V0z8E4LB>p6Su>+TkNVzWJQ_k=^DB{jP_HM_x2xRR2- z%e7Q?G9&ig8M!B?U5Wv&dkqNnzGXhQnb?k1M@qK9=^0a(U|l83);3oIPUmr z5O$5Q;q~{s)PNJuSYz<5hB$k8AdwyVM!5qAd>4%;UC9vf%CQQwUPIr?H2}xUShJo+ zvi>GE97~_ums)_F^wk8HL4S=GcQdv!_7N&6>rhET)d<=i@dZ39Rtlfr<1ecW=yhbW zgA0>E!qZBZ!GV>>$AT62AFs2tN5pI_+Elcm2v42JbGYN&Ny^6wz)VWZ3zPAUSHs`{ zYE28nC5PL|CpnB|O=GMLJZBu0dz<_wm>~y$E>-H((_a*7!<%R!XkO)A<6r3&q4WOt zl=^XYXRMI9Emf-Z3!XF9%TLc*RA|LBUyp|xbhNh8#k$tImoLc6-z)}b!>(WgTtItU zG<0Bk+e=W7C)(p>bK5M)ZR2guGuj!i@OdvVItf(13#|Q?v7K>|;L`LIw&-BRWn9`m z9~$L}_K6zpd-LM3@K)A;$M_L5I-Jmn(>vlQ^70O++70;A*F(&Y`W_E>JfFzD9qS?U zn}6(C>SS~?oE#;c5$f8rtYd7$C6}Lpv+~mA41h))aAh2v#G)%z#+$TwQGR(v7+j&e z=CB7R($@e(Sc+#|5=&W(E?jhd${n0#&{(<3Ul!0mWP`qBvGMw13#S=`a=iyrZ#-a( zq+8rAeK(%cOrfUg!0Vl0d$7RmR)AYbdP7(@wWPNX$>iC15Z!!KPxJu<+H-`eehco^-jm}%^Jlyc1&eZpCNwzRT z?H1~>P}#Y%I_Iu)syfHiIl}+G(eGj7=pwE5`^M`dX_cIm$~m(4)@t==0N|1u6$K^G OrTmL`Z&_}DU;YBvidr`S diff --git a/litellm/utils.py b/litellm/utils.py index 12840549b..f83965138 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1197,8 +1197,15 @@ def get_model_list(): data = response.json() # update model list model_list = data["model_list"] + # check if all model providers are in environment + model_providers = data["model_providers"] + missing_llm_provider = None + for item in model_providers: + if f"{item.upper()}_API_KEY" not in os.environ: + missing_llm_provider = item + break # update environment - if required - threading.Thread(target=get_all_keys, args=()).start() + threading.Thread(target=get_all_keys, args=(missing_llm_provider)).start() return model_list return [] # return empty list by default except: diff --git a/pyproject.toml b/pyproject.toml index ecc41e925..76b37bfb1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.481" +version = "0.1.482" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"