fix unit tests

deepset-ai · Mar 8, 2024 · 564b5a7 · 564b5a7
1 parent 6e00364
commit 564b5a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/integrations/cohere/tests/test_cohere_generators.py b/integrations/cohere/tests/test_cohere_generators.py
@@ -13,7 +13,8 @@
 
 
 class TestCohereGenerator:
-    def test_init_default(self):
+    def test_init_default(self, monkeypatch):
+        monkeypatch.setenv("COHERE_API_KEY", "foo")
         component = CohereGenerator()
         assert component.api_key == Secret.from_env_var(["COHERE_API_KEY", "CO_API_KEY"])
         assert component.model == "command"