phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test - azure/command-r-plus

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-05 08:56:05 -07:00
parent 71352b1b36
commit 5d196ff300
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
litellm/tests/test_completion.py
View file

@ -53,6 +53,25 @@ def test_completion_custom_provider_model_name():
# test_completion_custom_provider_model_name() # test_completion_custom_provider_model_name()
def test_completion_azure_command_r():
try:
litellm.set_verbose = True
response = completion(
model="azure/command-r-plus",
model_id="Cohere-command-r-plus-gylpd",
api_base=os.getenv("AZURE_COHERE_API_BASE"),
api_key=os.getenv("AZURE_COHERE_API_KEY"),
messages=[{"role": "user", "content": "What is the meaning of life?"}],
)
print(response)
except litellm.Timeout as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_claude(): def test_completion_claude():
litellm.set_verbose = True litellm.set_verbose = True
litellm.cache = None litellm.cache = None