Fix the unit tests for ollama FIM

Signed-off-by: Radoslav Dimitrov <[email protected]>
stacklok · Jan 30, 2025 · 3df7c6c · 3df7c6c
1 parent dc7bcc9
commit 3df7c6c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/providers/ollama/test_ollama_completion_handler.py b/tests/providers/ollama/test_ollama_completion_handler.py
@@ -40,6 +40,7 @@ async def test_execute_completion_is_fim_request(handler, chat_request):
         prompt="FIM prompt",
         stream=False,
         options=chat_request["options"],
+        suffix="",
     )