From 058783baf966295630eb990a44326fad0b81947f Mon Sep 17 00:00:00 2001 From: Massimiliano Pippi Date: Mon, 18 Mar 2024 12:47:06 +0100 Subject: [PATCH] relax test constraints (#591) --- integrations/cohere/tests/test_cohere_chat_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/cohere/tests/test_cohere_chat_generator.py b/integrations/cohere/tests/test_cohere_chat_generator.py index 9a822856e..1684046eb 100644 --- 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),