From c6df8d2c7a7ca10098627abbc2466a6b91de4ee2 Mon Sep 17 00:00:00 2001 From: Stefano Fiorucci Date: Fri, 4 Apr 2025 13:33:22 +0200 Subject: [PATCH] test: monkeypatch OpenAI API key in some unit tests (#9173) --- test/components/evaluators/test_context_relevance_evaluator.py | 3 ++- test/components/evaluators/test_faithfulness_evaluator.py | 3 ++- test/utils/test_deserialization.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/test/components/evaluators/test_context_relevance_evaluator.py b/test/components/evaluators/test_context_relevance_evaluator.py index e62314440..c381aadba 100644 --- a/test/components/evaluators/test_context_relevance_evaluator.py +++ b/test/components/evaluators/test_context_relevance_evaluator.py @@ -80,7 +80,8 @@ class TestContextRelevanceEvaluator: assert component._chat_generator.client.api_key == "test-api-key" assert component._chat_generator.generation_kwargs == {"response_format": {"type": "json_object"}, "seed": 42} - def test_init_with_chat_generator(self): + def test_init_with_chat_generator(self, monkeypatch): + monkeypatch.setenv("OPENAI_API_KEY", "test-api-key") chat_generator = OpenAIChatGenerator(generation_kwargs={"response_format": {"type": "json_object"}, "seed": 42}) component = ContextRelevanceEvaluator(chat_generator=chat_generator) diff --git a/test/components/evaluators/test_faithfulness_evaluator.py b/test/components/evaluators/test_faithfulness_evaluator.py index a2756c48d..f8edbe0ba 100644 --- a/test/components/evaluators/test_faithfulness_evaluator.py +++ b/test/components/evaluators/test_faithfulness_evaluator.py @@ -100,7 +100,8 @@ class TestFaithfulnessEvaluator: assert component._chat_generator.client.api_key == "test-api-key" assert component._chat_generator.generation_kwargs == {"response_format": {"type": "json_object"}, "seed": 42} - def test_init_with_chat_generator(self): + def test_init_with_chat_generator(self, monkeypatch): + monkeypatch.setenv("OPENAI_API_KEY", "test-api-key") chat_generator = OpenAIChatGenerator(generation_kwargs={"response_format": {"type": "json_object"}, "seed": 42}) component = FaithfulnessEvaluator(chat_generator=chat_generator) diff --git a/test/utils/test_deserialization.py b/test/utils/test_deserialization.py index cf5a5b39e..9f58079be 100644 --- a/test/utils/test_deserialization.py +++ b/test/utils/test_deserialization.py @@ -98,7 +98,8 @@ class TestDeserializeDocumentStoreInInitParamsInplace: class TestDeserializeChatGeneratorInplace: - def test_deserialize_chatgenerator_inplace(self): + def test_deserialize_chatgenerator_inplace(self, monkeypatch): + monkeypatch.setenv("OPENAI_API_KEY", "test-api-key") chat_generator = OpenAIChatGenerator() data = {"chat_generator": chat_generator.to_dict()}