fix: avoid JSONDecodeError for empty line stream response

ollama · Dec 25, 2024 · 22d6d2a · 22d6d2a
1 parent 7d1e002
commit 22d6d2a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)