phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(ci/cd) openrouter unstable - use other model

Browse source
This commit is contained in:
ishaan-jaff 2024-01-15 17:43:56 -08:00
parent 813e686354
commit a8f2550c25
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_completion.py
View file

@ -105,7 +105,7 @@ def test_completion_mistral_api():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_completion_mistral_api() # test_completion_mistral_api()
def test_completion_claude2_1(): def test_completion_claude2_1():
@ -771,8 +771,9 @@ def test_completion_ollama_hosted():
def test_completion_openrouter1(): def test_completion_openrouter1():
try: try:
litellm.set_verbose = True
response = completion( response = completion(
model="openrouter/google/palm-2-chat-bison", model="openrouter/mistralai/mistral-tiny",
messages=messages, messages=messages,
max_tokens=5, max_tokens=5,
) )
@ -1637,7 +1638,7 @@ def test_azure_cloudflare_api():
pass pass
test_azure_cloudflare_api() # test_azure_cloudflare_api()
def test_completion_anyscale_2(): def test_completion_anyscale_2():
@ -1932,7 +1933,7 @@ def test_completion_cloudflare():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_completion_cloudflare() # test_completion_cloudflare()
def test_moderation(): def test_moderation():