diff --git a/integrations/cohere/tests/test_cohere_generators.py b/integrations/cohere/tests/test_cohere_generators.py index c3dd59535..32fdfce50 100644 --- 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"