ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(ci/cd) test

Browse Source
This commit is contained in:
ishaan-jaff 2023-12-11 23:22:40 -08:00
parent 85a14d45e5
commit c89ed8f4c8
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/tests/test_completion.py
View File

@ -1024,6 +1024,9 @@ def test_customprompt_together_ai():
try:
litellm.set_verbose = False
litellm.num_retries = 0
print("in test_customprompt_together_ai")
print(litellm.success_callback)
print(litellm._async_success_callback)
response = completion(
model="together_ai/mistralai/Mistral-7B-Instruct-v0.1",
messages=messages,
@ -1032,7 +1035,6 @@ def test_customprompt_together_ai():
print(response)
except litellm.exceptions.Timeout as e:
print(f"Timeout Error")
litellm.num_retries = 3 # reset retries
pass
except Exception as e:
print(f"ERROR TYPE {type(e)}")