From 35c854443c868c7a1568ffb38400c814edc89ecb Mon Sep 17 00:00:00 2001 From: Mattt Zmuda Date: Fri, 19 Apr 2024 14:42:00 -0700 Subject: [PATCH] Fix initialization of stream decoder Signed-off-by: Mattt Zmuda --- replicate/stream.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/replicate/stream.py b/replicate/stream.py index 9aebf94d..844973d4 100644 --- a/replicate/stream.py +++ b/replicate/stream.py @@ -79,10 +79,16 @@ class Decoder: A decoder for server-sent events. """ - event: Optional["ServerSentEvent.EventType"] = None - data: List[str] = [] - last_event_id: Optional[str] = None - retry: Optional[int] = None + event: Optional["ServerSentEvent.EventType"] + data: List[str] + last_event_id: Optional[str] + retry: Optional[int] + + def __init__(self) -> None: + self.event = None + self.data = [] + self.last_event_id = None + self.retry = None def decode(self, line: str) -> Optional[ServerSentEvent]: """ @@ -134,6 +140,7 @@ def decode(self, line: str) -> Optional[ServerSentEvent]: def __iter__(self) -> Iterator[ServerSentEvent]: decoder = EventSource.Decoder() + for line in self.response.iter_lines(): line = line.rstrip("\n") sse = decoder.decode(line)