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

test(test_completion.py): change tgai model

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-17 16:06:34 -08:00
parent 30a95a7823
commit 985583023a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_completion.py
View file

@ -847,7 +847,7 @@ def test_customprompt_together_ai():
litellm.set_verbose = False litellm.set_verbose = False
litellm.num_retries = 0 litellm.num_retries = 0
response = completion( response = completion(
model="together_ai/OpenAssistant/llama2-70b-oasst-sft-v10", model="together_ai/togethercomputer/llama-2-70b-chat",
messages=messages, messages=messages,
roles={"system":{"pre_message":"<|im_start|>system\n", "post_message":"<|im_end|>"}, "assistant":{"pre_message":"<|im_start|>assistant\n","post_message":"<|im_end|>"}, "user":{"pre_message":"<|im_start|>user\n","post_message":"<|im_end|>"}} roles={"system":{"pre_message":"<|im_start|>system\n", "post_message":"<|im_end|>"}, "assistant":{"pre_message":"<|im_start|>assistant\n","post_message":"<|im_end|>"}, "user":{"pre_message":"<|im_start|>user\n","post_message":"<|im_end|>"}}
) )
@ -860,7 +860,7 @@ def test_customprompt_together_ai():
print(f"ERROR TYPE {type(e)}") print(f"ERROR TYPE {type(e)}")
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_customprompt_together_ai() test_customprompt_together_ai()
def test_completion_sagemaker(): def test_completion_sagemaker():
try: try: