Skip to content

Commit 22d6d2a

Browse files
committed
fix: avoid JSONDecodeError for empty line stream response
1 parent 7d1e002 commit 22d6d2a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ollama/_client.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,7 @@ def inner():
167167
raise ResponseError(e.response.text, e.response.status_code) from None
168168

169169
for line in r.iter_lines():
170+
if not line: continue
170171
part = json.loads(line)
171172
if err := part.get('error'):
172173
raise ResponseError(err)
@@ -669,6 +670,7 @@ async def inner():
669670
raise ResponseError(e.response.text, e.response.status_code) from None
670671

671672
async for line in r.aiter_lines():
673+
if not line: continue
672674
part = json.loads(line)
673675
if err := part.get('error'):
674676
raise ResponseError(err)

0 commit comments

Comments
 (0)