test: update test due to cohere ssl issues

This commit is contained in:
Krrish Dholakia 2025-05-19 20:07:57 -07:00
parent 38470d7164
commit 4e3c8ae94f
2 changed files with 6 additions and 6 deletions

View file

@ -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

View file

@ -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