diff --git a/tests/lib/processors/llmproviders/gemini.py b/tests/lib/processors/llmproviders/gemini.py index d49189dc..11120daa 100644 --- a/tests/lib/processors/llmproviders/gemini.py +++ b/tests/lib/processors/llmproviders/gemini.py @@ -123,9 +123,7 @@ def test_generate_with_history(self, mock_gen_model, mock_gen_config): values={'GOOGLE_API_KEY': 'fake_env_key'}, clear=True ) - @mock.patch('google.generativeai.configure') - @mock.patch('google.generativeai.GenerativeModel', autospec=True) - def test_generate_from_template(self, mock_gen_model, mock_config): + def test_generate_from_template(self): """Tests the GenerateFromTemplate method.""" provider = gemini.GeminiLLMProvider() mock_generate = mock.Mock()