diff --git a/litellm/tests/test_async_fn.py b/litellm/tests/test_async_fn.py index 65db9a57a6f..ec322f6d5e0 100644 --- a/litellm/tests/test_async_fn.py +++ b/litellm/tests/test_async_fn.py @@ -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 diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index d6d0d93ae30..6309db673d8 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -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( diff --git a/litellm/tests/test_model_alias_map.py b/litellm/tests/test_model_alias_map.py index 31a7d34b8e1..1e69bcf95e2 100644 --- a/litellm/tests/test_model_alias_map.py +++ b/litellm/tests/test_model_alias_map.py @@ -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}")