test(test_completion.py): fix replicate test which was hanging

2026-08-07 00:24:03 +00:00 · 2024-06-06 22:17:30 -07:00
parent 09a471da10
commit ddc82774b0
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,
        )