tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-18 02:23:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(test_completion.py): fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2024-03-06 20:13:09 -08:00
parent d1d8adfb11
commit be6674f5e0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_completion.py
+2 -2
View File
@@ -1453,9 +1453,9 @@ def test_completion_replicate_vicuna():
def test_replicate_custom_prompt_dict():
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(
model="replicate/meta/llama-2-7b-chat:13c3cdee13ee059ab779f0291d29054dab00a47dad8261375654de5540165fb0",
model="replicate/meta/llama-2-7b-chat",
initial_prompt_value="You are a good assistant", # [OPTIONAL]
roles={
"system": {