From 7acaf5a5b078d5a2b771038675ee0d662e2002f7 Mon Sep 17 00:00:00 2001 From: Amna Mubashar Date: Wed, 21 Aug 2024 13:45:04 +0200 Subject: [PATCH] Type fixing --- .../components/generators/google_vertex/gemini.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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.