phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(feat) exception mapping for cohere_chat

Browse source
This commit is contained in:
ishaan-jaff 2024-03-12 10:45:42 -07:00
parent 8fabaed543
commit e5bb65669d
2 changed files with 58 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/tests/test_completion.py
View file

@ -1980,6 +1980,29 @@ def test_chat_completion_cohere():
pytest.fail(f"Error occurred: {e}")
def test_chat_completion_cohere_stream():
try:
litellm.set_verbose = False
messages = [
{"role": "system", "content": "You're a good bot"},
{
"role": "user",
"content": "Hey",
},
]
response = completion(
model="cohere_chat/command-r",
messages=messages,
max_tokens=10,
stream=True,
)
print(response)
for chunk in response:
print(chunk)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_azure_cloudflare_api():
litellm.set_verbose = True
try: