From 0b712c2bde77fe7e3e347d71d5de3c9e62f26738 Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Fri, 12 Apr 2024 18:01:54 +0200 Subject: Add unittests for async client (#1830) * Add unittests for async client * Add pollyfill for anext * Update integration tests --- g4f/requests/curl_cffi.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) (limited to 'g4f/requests') diff --git a/g4f/requests/curl_cffi.py b/g4f/requests/curl_cffi.py index 000448fe..e955d640 100644 --- a/g4f/requests/curl_cffi.py +++ b/g4f/requests/curl_cffi.py @@ -34,15 +34,13 @@ class StreamResponse: """Asynchronously parse the JSON response content.""" return json.loads(await self.inner.acontent(), **kwargs) - async def iter_lines(self) -> AsyncGenerator[bytes, None]: + def iter_lines(self) -> AsyncGenerator[bytes, None]: """Asynchronously iterate over the lines of the response.""" - async for line in self.inner.aiter_lines(): - yield line + return self.inner.aiter_lines() - async def iter_content(self) -> AsyncGenerator[bytes, None]: + def iter_content(self) -> AsyncGenerator[bytes, None]: """Asynchronously iterate over the response content.""" - async for chunk in self.inner.aiter_content(): - yield chunk + return self.inner.aiter_content() async def __aenter__(self): """Asynchronously enter the runtime context for the response object.""" -- cgit v1.2.3