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

test(test_completion.py): change model

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-09 21:38:17 -07:00
parent 0408b3217f
commit a6b004f10b
2 changed files with 5317 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

5320
litellm/tests/log.txt
View file

File diff suppressed because it is too large Load diff

5
litellm/tests/test_completion.py
View file

@ -1632,9 +1632,9 @@ def test_completion_replicate_vicuna():
def test_replicate_custom_prompt_dict(): def test_replicate_custom_prompt_dict():
litellm.set_verbose = True litellm.set_verbose = True
model_name = "replicate/meta/llama-2-7b-chat" model_name = "replicate/meta/llama-2-70b-chat"
litellm.register_prompt_template( litellm.register_prompt_template(
model="replicate/meta/llama-2-7b-chat", model="replicate/meta/llama-2-70b-chat",
initial_prompt_value="You are a good assistant", # [OPTIONAL] initial_prompt_value="You are a good assistant", # [OPTIONAL]
roles={ roles={
"system": { "system": {
@ -1660,6 +1660,7 @@ def test_replicate_custom_prompt_dict():
"content": "what is yc write 1 paragraph", "content": "what is yc write 1 paragraph",
} }
], ],
repetition_penalty=0.1,
num_retries=3, num_retries=3,
) )
print(f"response: {response}") print(f"response: {response}")