phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

adding support for nlp cloud

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-14 09:19:32 -07:00
parent 5b5835480c
commit f98da9f13c
10 changed files with 286 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_exceptions.py
View file

@ -75,6 +75,9 @@ def invalid_auth(model): # set the model key to an invalid key, depending on th
elif model in litellm.aleph_alpha_models:
temporary_key = os.environ["ALEPH_ALPHA_API_KEY"]
os.environ["ALEPH_ALPHA_API_KEY"] = "bad-key"
elif model in litellm.nlp_cloud_models:
temporary_key = os.environ["NLP_CLOUD_API_KEY"]
os.environ["NLP_CLOUD_API_KEY"] = "bad-key"
elif (
model
== "replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1"
@ -119,6 +122,8 @@ def invalid_auth(model): # set the model key to an invalid key, depending on th
os.environ["TOGETHERAI_API_KEY"] = temporary_key
elif model in litellm.aleph_alpha_models:
os.environ["ALEPH_ALPHA_API_KEY"] = temporary_key
elif model in litellm.nlp_cloud_models:
os.environ["NLP_CLOUD_API_KEY"] = temporary_key
return
# Test 3: Invalid Request Error