mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
refactor: move all testing to top-level of repo
Closes https://github.com/BerriAI/litellm/issues/486
This commit is contained in:
parent
5403c5828c
commit
3560f0ef2c
213 changed files with 74 additions and 217 deletions
54
tests/local_testing/test_multiple_deployments.py
Normal file
54
tests/local_testing/test_multiple_deployments.py
Normal file
|
@ -0,0 +1,54 @@
|
|||
#### What this tests ####
|
||||
# This tests error handling + logging (esp. for sentry breadcrumbs)
|
||||
|
||||
import sys, os
|
||||
import traceback
|
||||
|
||||
sys.path.insert(
|
||||
0, os.path.abspath("../..")
|
||||
) # Adds the parent directory to the system path
|
||||
import pytest
|
||||
import litellm
|
||||
from litellm import completion
|
||||
|
||||
messages = [{"role": "user", "content": "Hey, how's it going?"}]
|
||||
|
||||
## All your mistral deployments ##
|
||||
model_list = [
|
||||
{
|
||||
"model_name": "mistral-7b-instruct",
|
||||
"litellm_params": { # params for litellm completion/embedding call
|
||||
"model": "replicate/mistralai/mistral-7b-instruct-v0.1:83b6a56e7c828e667f21fd596c338fd4f0039b46bcfa18d973e8e70e455fda70",
|
||||
"api_key": os.getenv("REPLICATE_API_KEY"),
|
||||
},
|
||||
},
|
||||
{
|
||||
"model_name": "mistral-7b-instruct",
|
||||
"litellm_params": { # params for litellm completion/embedding call
|
||||
"model": "together_ai/mistralai/Mixtral-8x7B-Instruct-v0.1",
|
||||
"api_key": os.getenv("TOGETHERAI_API_KEY"),
|
||||
},
|
||||
},
|
||||
{
|
||||
"model_name": "mistral-7b-instruct",
|
||||
"litellm_params": {
|
||||
"model": "deepinfra/mistralai/Mistral-7B-Instruct-v0.1",
|
||||
"api_key": os.getenv("DEEPINFRA_API_KEY"),
|
||||
},
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
def test_multiple_deployments():
|
||||
try:
|
||||
## LiteLLM completion call ## returns first response
|
||||
response = completion(
|
||||
model="mistral-7b-instruct", messages=messages, model_list=model_list
|
||||
)
|
||||
print(f"response: {response}")
|
||||
except Exception as e:
|
||||
traceback.print_exc()
|
||||
pytest.fail(f"An exception occurred: {e}")
|
||||
|
||||
|
||||
test_multiple_deployments()
|
Loading…
Add table
Add a link
Reference in a new issue