mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 11:43:54 +00:00
test(test_completion.py): fix test to use mock response - prevent hang
This commit is contained in:
parent
48e743a81a
commit
a4c20294e0
1 changed files with 1 additions and 0 deletions
|
@ -2538,6 +2538,7 @@ def test_replicate_custom_prompt_dict():
|
|||
"content": "what is yc write 1 paragraph",
|
||||
}
|
||||
],
|
||||
mock_response="hello world",
|
||||
repetition_penalty=0.1,
|
||||
num_retries=3,
|
||||
)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue