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

(test) fix replicate test

Browse source
This commit is contained in:
ishaan-jaff 2024-03-06 18:12:24 -08:00
parent 96a3c03a1f
commit 6123e349f2
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

@ -1453,9 +1453,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:13c3cdee13ee059ab779f0291d29054dab00a47dad8261375654de5540165fb0" model_name = "replicate/meta/llama-2-7b-chat"
litellm.register_prompt_template( litellm.register_prompt_template(
model="replicate/meta/llama-2-7b-chat:13c3cdee13ee059ab779f0291d29054dab00a47dad8261375654de5540165fb0", model="replicate/meta/llama-2-7b-chat",
initial_prompt_value="You are a good assistant", # [OPTIONAL] initial_prompt_value="You are a good assistant", # [OPTIONAL]
roles={ roles={
"system": { "system": {