From f492f2e05fbf24401046d0bca7b501f7121583e8 Mon Sep 17 00:00:00 2001 From: nex Date: Fri, 10 Nov 2023 22:43:10 +0000 Subject: [PATCH] Fix stream reader --- cogs/other.py | 1 + 1 file changed, 1 insertion(+) diff --git a/cogs/other.py b/cogs/other.py index 41b7bb2..29b1b10 100644 --- a/cogs/other.py +++ b/cogs/other.py @@ -99,6 +99,7 @@ async def ollama_stream_reader(response: httpx.Response) -> typing.AsyncGenerato while not _buffer.endswith(b"}\n"): async for char in stream: _buffer += char + _buffer = _buffer.rstrip() log.debug("Resolving %r", _buffer.decode()) chunk = json.loads(_buffer.decode("utf-8", "replace")) log.debug("%r -> %r", _buffer.decode(), chunk)