summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Llama2.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/Provider/Llama2.py')
-rw-r--r--g4f/Provider/Llama2.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/g4f/Provider/Llama2.py b/g4f/Provider/Llama2.py
index 6a94eea1..04b5aee0 100644
--- a/g4f/Provider/Llama2.py
+++ b/g4f/Provider/Llama2.py
@@ -3,6 +3,7 @@ from __future__ import annotations
from aiohttp import ClientSession
from ..typing import AsyncResult, Messages
+from ..requests.raise_for_status import raise_for_status
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
@@ -67,8 +68,10 @@ class Llama2(AsyncGeneratorProvider, ProviderModelMixin):
}
started = False
async with session.post(f"{cls.url}/api", json=data, proxy=proxy) as response:
- response.raise_for_status()
+ await raise_for_status(response)
async for chunk in response.content.iter_any():
+ if not chunk:
+ continue
if not started:
chunk = chunk.lstrip()
started = True