From 65f8015221d8425369b0139449a779e080280bb0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 15 Apr 2025 21:08:48 -0700 Subject: [PATCH] test fix - azure deprecated azure ai mistral --- tests/local_testing/test_completion_cost.py | 23 --------------------- 1 file changed, 23 deletions(-) diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index af89c38789..3e30041489 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -864,29 +864,6 @@ def test_vertex_ai_embedding_completion_cost(caplog): # assert False - -def test_completion_azure_ai(): - try: - os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True" - litellm.model_cost = litellm.get_model_cost_map(url="") - - litellm.set_verbose = True - response = litellm.completion( - model="azure_ai/Mistral-large-nmefg", - messages=[{"content": "what llm are you", "role": "user"}], - max_tokens=15, - num_retries=3, - api_base=os.getenv("AZURE_AI_MISTRAL_API_BASE"), - api_key=os.getenv("AZURE_AI_MISTRAL_API_KEY"), - ) - print(response) - - assert "response_cost" in response._hidden_params - assert isinstance(response._hidden_params["response_cost"], float) - except Exception as e: - pytest.fail(f"Error occurred: {e}") - - @pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.asyncio async def test_completion_cost_hidden_params(sync_mode):