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

added model openrouter/mistralai/mistral-7b-instruct with test

Browse source
This commit is contained in:
Lucas Hofer 2023-09-30 16:49:18 +01:00
parent 905a3f08ee
commit 42cc4e2464
2 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_completion.py
View file

@ -461,6 +461,18 @@ def test_completion_openrouter2():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_openrouter3():
try:
response = completion(
model="openrouter/mistralai/mistral-7b-instruct",
messages=messages,
max_tokens=5,
)
# Add any assertions here to check the response
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openrouter()
def test_completion_hf_model_no_provider():