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

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

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():

View file

@ -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,