Fix GeminiGenerator input type

deepset-ai · Dec 15, 2023 · 212c2e8 · 212c2e8
1 parent 27f55ff
commit 212c2e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/google-vertex/src/google_vertex_haystack/generators/gemini.py b/integrations/google-vertex/src/google_vertex_haystack/generators/gemini.py
@@ -131,7 +131,7 @@ def _convert_part(self, part: Union[str, ByteStream, Part]) -> Part:
             raise ValueError(msg)
 
     @component.output_types(answers=List[Union[str, Dict[str, str]]])
-    def run(self, parts: Variadic[List[Union[str, ByteStream, Part]]]):
+    def run(self, parts: Variadic[Union[str, ByteStream, Part]]):
         converted_parts = [self._convert_part(p) for p in parts]
 
         contents = [Content(parts=converted_parts, role="user")]