Fix test

deepset-ai · May 30, 2024 · 8c034ce · 8c034ce
1 parent 30a7f3f
commit 8c034ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/ollama/tests/test_chat_generator.py b/integrations/ollama/tests/test_chat_generator.py
@@ -41,7 +41,7 @@ def test_init(self):
         assert component.timeout == 5
 
     def test_create_json_payload(self, chat_messages):
-        observed = OllamaChatGenerator(model="some_model")._create_json_payload(chat_messages, {"temperature": 0.1})
+        observed = OllamaChatGenerator(model="some_model")._create_json_payload(chat_messages, False, {"temperature": 0.1})
         expected = {
             "messages": [
                 {"role": "user", "content": "Tell me about why Super Mario is the greatest superhero"},