ci/cd - anyscale discontinued their API endoints - skip test

This commit is contained in:
Ishaan Jaff 2024-08-01 17:58:48 -07:00
parent fc3463e178
commit 80831ce73f
3 changed files with 14 additions and 5 deletions

View file

@ -18,6 +18,7 @@ from litellm import acompletion, acreate, completion
litellm.num_retries = 3
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_sync_response_anyscale():
litellm.set_verbose = False
user_message = "Hello, how are you?"
@ -119,6 +120,7 @@ def test_async_response_azure():
# test_async_response_azure()
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_async_anyscale_response():
import asyncio
@ -301,6 +303,7 @@ def test_get_response_streaming():
# test_get_response_streaming()
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_get_response_non_openai_streaming():
import asyncio

View file

@ -2574,6 +2574,7 @@ def test_completion_hf_model_no_provider():
# test_completion_hf_model_no_provider()
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_completion_anyscale_with_functions():
function1 = [
{
@ -3594,6 +3595,7 @@ def test_completion_anthropic_hanging():
assert msg["role"] != converted_messages[i + 1]["role"]
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_completion_anyscale_api():
try:
# litellm.set_verbose=True
@ -3711,6 +3713,7 @@ def test_azure_cloudflare_api():
# test_azure_cloudflare_api()
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_completion_anyscale_2():
try:
# litellm.set_verbose = True
@ -3733,6 +3736,7 @@ def test_completion_anyscale_2():
pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="anyscale stopped serving public api endpoints")
def test_mistral_anyscale_stream():
litellm.set_verbose = False
response = completion(

View file

@ -1,19 +1,21 @@
#### What this tests ####
# This tests the model alias mapping - if user passes in an alias, and has set an alias, set it to the actual value
import sys, os
import os
import sys
import traceback
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
import pytest
import litellm
from litellm import completion, embedding
litellm.set_verbose = True
model_alias_map = {"good-model": "anyscale/meta-llama/Llama-2-7b-chat-hf"}
model_alias_map = {"good-model": "groq/llama3-8b-8192"}
def test_model_alias_map(caplog):
@ -33,7 +35,7 @@ def test_model_alias_map(caplog):
for log in captured_logs:
assert "ERROR" not in log
assert "Llama-2-7b-chat-hf" in response.model
assert "groq/llama3-8b-8192" in response.model
except Exception as e:
pytest.fail(f"Error occurred: {e}")