From d721bace91154c40a620e4c610abbc4d22698e7a Mon Sep 17 00:00:00 2001 From: Stefano Fiorucci Date: Fri, 12 Jan 2024 11:08:27 +0100 Subject: [PATCH] improve test (#191) --- 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 f4e361c7d..07d75a226 100644 --- a/integrations/ollama/tests/test_chat_generator.py +++ b/integrations/ollama/tests/test_chat_generator.py @@ -115,7 +115,7 @@ def test_run_with_chat_history(self): assert isinstance(response, dict) assert isinstance(response["replies"], list) - assert "Manchester" in response["replies"][-1].content + assert "Manchester" in response["replies"][-1].content or "Glasgow" in response["replies"][-1].content @pytest.mark.integration def test_run_model_unavailable(self):