mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
added model openrouter/mistralai/mistral-7b-instruct with test
This commit is contained in:
parent
905a3f08ee
commit
42cc4e2464
2 changed files with 19 additions and 0 deletions
|
@ -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():
|
||||
|
|
|
@ -382,6 +382,13 @@
|
|||
"litellm_provider": "openrouter",
|
||||
"mode": "chat"
|
||||
},
|
||||
"openrouter/mistralai/mistral-7b-instruct": {
|
||||
"max_tokens": 4096,
|
||||
"input_cost_per_token": 0.0,
|
||||
"output_cost_per_token": 0.0,
|
||||
"litellm_provider": "openrouter",
|
||||
"mode": "chat"
|
||||
},
|
||||
"j2-ultra": {
|
||||
"max_tokens": 8192,
|
||||
"input_cost_per_token": 0.000015,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue