From 4e3c8ae94f199fd4003658d01862e27b01c40020 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 19 May 2025 20:07:57 -0700 Subject: [PATCH] test: update test due to cohere ssl issues --- tests/llm_translation/test_azure_ai.py | 2 +- tests/local_testing/test_completion.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/llm_translation/test_azure_ai.py b/tests/llm_translation/test_azure_ai.py index 62f68b02d35..09596e147fd 100644 --- a/tests/llm_translation/test_azure_ai.py +++ b/tests/llm_translation/test_azure_ai.py @@ -184,7 +184,7 @@ def test_azure_ai_services_with_api_version(): == "https://litellm8397336933.services.ai.azure.com/models/chat/completions?api-version=2024-05-01-preview" ) - +@pytest.mark.skip(reason="Skipping due to cohere ssl issues") def test_completion_azure_ai_command_r(): try: import os diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index ef7086ac11f..98cc93ce394 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -457,11 +457,11 @@ def test_completion_claude_3_function_call(model): ("claude-3-opus-20240229", None, None), ("command-r", None, None), ("anthropic.claude-3-sonnet-20240229-v1:0", None, None), - ( - "azure_ai/command-r-plus", - os.getenv("AZURE_COHERE_API_KEY"), - os.getenv("AZURE_COHERE_API_BASE"), - ), + # ( + # "azure_ai/command-r-plus", + # os.getenv("AZURE_COHERE_API_KEY"), + # os.getenv("AZURE_COHERE_API_BASE"), + # ), ], ) @pytest.mark.asyncio