From 05f2f3dde78c75bbef381d3005bacfeae3013218 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 12 Sep 2023 11:48:42 -0700 Subject: [PATCH] fix get llm provider --- litellm/__pycache__/utils.cpython-311.pyc | Bin 98615 -> 99751 bytes litellm/utils.py | 3 +++ pyproject.toml | 2 +- 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc index d8e7e219ce043bca264b0ba17533b1c3c2e1072c..6e7724d5f9e6a18005cc957167751031d7a6ac7f 100644 GIT binary patch delta 1748 zcmZWpdr(wW7(d^=dzW2U*kzGjKoA#bm!~eEg7zdJQ3xN{7?iHSy&w;P?jo+Vh!Ksu zfS8_gEz-e9;zVQ8wJtMY<0GnRMrWZdXthEMr<|Iw{)lFp?A%qLsq^^GcfRlU{m!}P zeCLX7^1e=aP+wr6KO$vt0CB;e9}EhEhyl|QI-!&`wnND_wv3G@V<}tCCX%sCiDoOy zvPr~Jj`z|LG9_X;{1%-ZTEV7qg={*viYpA4Xnh6$NJm2@R>(|Hfz7gqVLPssg%j+W zb|`IP*S4dQU2Hy)Hns}CDSJ7pn%%;2>{iZEA!9A=Oo@}-M!X#?6rqF~VHxvq8{82Z z_Jsi>+EF8gg9}2Hh^JYm1?o=Yeun{G$5)AL$GSkLmCaUaD-Tm9rYkK82{pof*2go@epWnYKM;D5g>$DcRWS&QwVoM*Bre#m$IJIm1h^9)9puLnv)qYUIIz1m)_@#YvAG6xDLH!? z8Ac5LxoPxB5|x`mkEAd}F9%oZN?0JA>kb9DE_`wTQ!p)*9LfZ^C14!GfJ*rAh&K@8 zgoCGJbnv2Ze)vi##0s{tL_Zi5PJS(?;3NFb^*>=)ICaBH!8IZ5yC@325mwyn267Q2 zxBjBxH{tT_VFp5k)}PG`JSm*JI~oqvVp{+#4Tie!5GkHbg*yPpMKV!P z@_q$E)^gXebkKqrmaGP|3>LXM z^T{f^#KHoYrobpZz7C3HutLN|VDf|Y;!p|PW1v|a+6F(+&?ok?uvwy?;oxn63*xa# zhyu9k8r}{^VN5Kkf+7mO63^DaJvsa#p53HUiIn-Hb_g7i9SlT$m}vZBH0*f2UqVi zpmX^x25nf^8at5iA}NBEJ;+C$o$^8Odj(oe*P=%G{GO^6czx<3ot>^kr%Af?&*gjz zW2f+jV>;|i)V6kQvQtGUmLzN?DyNE35wg+6Xcr{N3;8ngDI|dr@+vG@>(^NFbBp;{ zVry1rHSgrPl!Ah%69V|%5~^7eDoJc0cz%PV{Uk*)I~y7}o*$NU?T&hutF>Cnc3T?w z`lc$D<6HR?l1I+j8o4?)LGD#?bxl=#eVwI-+imr#n@Ve|*g3S-tGaKvPvlk1(s)BT zTLV|#$jxIV0&mEJ6)Wobc@pW27W)csZk1NoRC0}$e-rc9$YG_Ef2Z|t;+Gz31f!kO zhribBYCY6Cs*m^R;~lDrh)9QeQXlS6O@&2!!cwM@Qkm)#j=V{I^tj$MrZ=6T&ukpk zn>~87BX3%P;>~?~PV5;qWq3>(?zk*>E9s=E3=`_OGb*=w8EKQ6=y8p4Ok?b|jcUvu zjoGa+&rF7yrV*oD=@X8;3AO%6#+W+Bt&W+{M2>4>$276M`Ezg*F=b5g7@rt7=8PG0 zEZ8gV(0MgxQx+sO{Pba>Grh=^+LB|;S6~&jCMjD>`+Q|W@g5!B8SkH z9h4o6t^~`n^)CNL=p;}0a3g$4!!=i>4Wj&SlM^GCj;tSu-8QIZlF~sljfgms?HkT* fAGDs|;Zy%Bs>SF%;7>2I!9vhZqyGy&l@R{}xiJ`1 delta 961 zcmZ8fZA_C_6u!^Lz{G^W@ct}`ezH7yRG4m=O)i{&dGDm zNzS<=M=T;@v5#bB*@(2^bHu!H^_V>m)-hk1TC$fGN71yG2I%XO1GJg0klbvLg-3&Q zrOX6bpSn~t2&L?#T3uq&RpvHYX>M!LXjN2i+Lw-76uLU9HMK63N_7i!YCf23qqdm`P1D?6io`f3~9ZJP8wGbZQkk=i&# zBPnU;wfcq#F7n9XJm_b7NteRr8l1KRq*JB(LF^x3<1r7SY*u23A-Q{=tbX> zu7ZUuA=KVW1_{k(r3p8dv-(649PGn{i}fdJ*>8zE$YJM?>|(k7SBwWBq>rSOayHlh zH5>eFntgn3#Q;P zpFF+}nC04uLLCP8eKnH>4_8JCT?q4{@hiDl&i}ZzViCUO&KZk>A6WjKe{g~4{vJ{= z$>Xzb1v7kP?kMn(uem?3;xAtL@S+X}PyJb;V>9>s{p)MkC!Q`wc@Esd>&CE#B5}=w z)jI0Lf5j+OK!v{qV+y*HcUR&GBFUS6gmEajwG=-gCSEDS6woj3m7`DrOWLZiOcbob zBjB`looHB(I^dERcoS9)SH%p;5tgzs@Esa-uXrvDkS`C-T3|o zw7bLH)Hx%+j65z04T(p+XxDueaHNQYFRL>&8qCnkK*c8!*i<`NhrCo8GR^-7;L#+) diff --git a/litellm/utils.py b/litellm/utils.py index 83c0e05f9d..1b8b1b4f52 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -947,6 +947,9 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None): ## openai - chatcompletion + text completion if model in litellm.open_ai_chat_completion_models or model in litellm.open_ai_text_completion_models: custom_llm_provider = "openai" + ## anthropic + elif model in litellm.anthropic_models: + custom_llm_provider = "anthropic" ## cohere elif model in litellm.cohere_models: custom_llm_provider = "cohere" diff --git a/pyproject.toml b/pyproject.toml index 876d9a605b..716b62615d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.603" +version = "0.1.604" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"