Skip to content

Commit

Permalink
Type fixing
Browse files Browse the repository at this point in the history
  • Loading branch information
Amnah199 committed Aug 21, 2024
1 parent 9ec32a5 commit 7acaf5a
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -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.
Expand Down

0 comments on commit 7acaf5a

Please sign in to comment.