From f7dd68803564779318f320fbfdfe4e5345ab89eb Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 18 Apr 2025 08:42:12 -0700 Subject: [PATCH] test: handle cohere rbac issue (verified happens on calling azure directly) --- tests/local_testing/test_completion.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index d56475e1a0..041a49cb99 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -131,15 +131,15 @@ def test_null_role_response(): assert response.choices[0].message.role == "assistant" - +@pytest.mark.skip(reason="Cohere having RBAC issues") def test_completion_azure_command_r(): try: - litellm.set_verbose = True + litellm._turn_on_debug() response = completion( model="azure/command-r-plus", - api_base=os.getenv("AZURE_COHERE_API_BASE"), - api_key=os.getenv("AZURE_COHERE_API_KEY"), + api_base="https://Cohere-command-r-plus-gylpd-serverless.eastus2.inference.ai.azure.com", + api_key="AO89xyvmOLLMgoMI7WaiEaP0t6M09itr", messages=[{"role": "user", "content": "What is the meaning of life?"}], )