From 22d6d2ac53ba7feda250313690346bd4cc258644 Mon Sep 17 00:00:00 2001 From: johnny0120 Date: Wed, 25 Dec 2024 13:15:34 +0800 Subject: [PATCH] fix: avoid JSONDecodeError for empty line stream response --- ollama/_client.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ollama/_client.py b/ollama/_client.py index 87fa881..3b8cfe3 100644 --- a/ollama/_client.py +++ b/ollama/_client.py @@ -167,6 +167,7 @@ def inner(): raise ResponseError(e.response.text, e.response.status_code) from None for line in r.iter_lines(): + if not line: continue part = json.loads(line) if err := part.get('error'): raise ResponseError(err) @@ -669,6 +670,7 @@ async def inner(): raise ResponseError(e.response.text, e.response.status_code) from None async for line in r.aiter_lines(): + if not line: continue part = json.loads(line) if err := part.get('error'): raise ResponseError(err)