kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: avoid JSONDecodeError for empty line stream response

Browse Source
This commit is contained in:
johnny0120 2024-12-25 13:15:34 +08:00
parent 7d1e002be9
commit 22d6d2ac53
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ollama/_client.py
View File

@ -167,6 +167,7 @@ class Client(BaseClient):
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 @@ class AsyncClient(BaseClient):
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)