relax test constraints (#591)

deepset-ai · Mar 18, 2024 · 058783b · 058783b
1 parent 430c080
commit 058783b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/cohere/tests/test_cohere_chat_generator.py b/integrations/cohere/tests/test_cohere_chat_generator.py
@@ -308,7 +308,7 @@ def test_live_run_with_connector(self):
         message: ChatMessage = results["replies"][0]
         assert "Paris" in message.content
         assert message.meta["documents"] is not None
-        assert message.meta["citations"] is not None
+        assert "citations" in message.meta  # Citations might be None
 
     @pytest.mark.skipif(
         not os.environ.get("COHERE_API_KEY", None) and not os.environ.get("CO_API_KEY", None),