Update test_gemini.py

deepset-ai · Aug 21, 2024 · 9ec32a5 · 9ec32a5
1 parent e09fa09
commit 9ec32a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/google_vertex/tests/test_gemini.py b/integrations/google_vertex/tests/test_gemini.py
@@ -171,7 +171,7 @@ def test_from_dict(_mock_vertexai, _mock_generative_model):
     assert gemini._project_id == "TestID123"
     assert gemini._safety_settings == {HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_ONLY_HIGH}
 
-    #assert gemini._tools == [Tool(function_declarations=[GET_CURRENT_WEATHER_FUNC])]
+    # assert gemini._tools == [Tool(function_declarations=[GET_CURRENT_WEATHER_FUNC])]
     assert isinstance(gemini._generation_config, GenerationConfig)