From 4b60e8cfb7bb55c8766da75d7a303c01f6294d34 Mon Sep 17 00:00:00 2001 From: Amna Mubashar Date: Thu, 17 Oct 2024 15:35:21 +0200 Subject: [PATCH] Fix --- integrations/google_vertex/tests/test_gemini.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/google_vertex/tests/test_gemini.py b/integrations/google_vertex/tests/test_gemini.py index 44e9bf1f9..b3d6dd5f5 100644 --- a/integrations/google_vertex/tests/test_gemini.py +++ b/integrations/google_vertex/tests/test_gemini.py @@ -319,7 +319,7 @@ def test_serialization_deserialization_pipeline(): pipeline = Pipeline() pipeline.add_component("prompt_builder", PromptBuilder(template=template)) - pipeline.add_component("gemini", VertexAIGeminiGenerator()) + pipeline.add_component("gemini", VertexAIGeminiGenerator(project_id="TestID123")) pipeline.connect("prompt_builder", "gemini") pipeline_dict = pipeline.to_dict()