Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Amnah199 committed Oct 17, 2024
1 parent c10853b commit 4b60e8c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion integrations/google_vertex/tests/test_gemini.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit 4b60e8c

Please sign in to comment.