diff --git a/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/gemini.py b/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/gemini.py
index 281e33c67..7e012e315 100644
--- a/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/gemini.py
+++ b/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/gemini.py
@@ -194,7 +194,7 @@ def run(self, parts: Variadic[Union[str, ByteStream, Part]]):
 
         return {"replies": replies}
 
-    def get_response(self, response_body: List[str]) -> List[str]:
+    def get_response(self, response_body) -> List[str]:
         """
         Extracts the responses from the Vertex AI response.
 
@@ -215,7 +215,7 @@ def get_response(self, response_body: List[str]) -> List[str]:
                     replies.append(function_call)
         return replies
 
-    def get_stream_response(self, stream: List[str], streaming_callback: Callable[[StreamingChunk], None]) -> List[str]:
+    def get_stream_response(self, stream, streaming_callback: Callable[[StreamingChunk], None]) -> List[str]:
         """
         Extracts the responses from the Vertex AI streaming response.