improve test (#191)

deepset-ai · Jan 12, 2024 · d721bac · d721bac
1 parent c974c8a
commit d721bac
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
@@ -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):