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

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():
litellm.set_verbose = True
model_name = "replicate/meta/llama-2-7b-chat"
model_name = "replicate/meta/llama-2-70b-chat"
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]
roles={
"system": {
@ -1660,6 +1660,7 @@ def test_replicate_custom_prompt_dict():
"content": "what is yc write 1 paragraph",
}
],
repetition_penalty=0.1,
num_retries=3,
)
print(f"response: {response}")