From 8c034ce013385a97fc3e74cc38e962948984be8b Mon Sep 17 00:00:00 2001 From: Etienne Date: Thu, 30 May 2024 15:40:44 -0400 Subject: [PATCH] Fix test --- integrations/ollama/tests/test_chat_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/ollama/tests/test_chat_generator.py b/integrations/ollama/tests/test_chat_generator.py index ff6a840b8..5da3c691c 100644 --- 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"},