summaryrefslogtreecommitdiffstats
path: root/g4f/client/async_client.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-05-21 20:39:31 +0200
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-05-21 20:39:31 +0200
commitb7624b75a3ed6a48733ddaa0c6ea87227922d173 (patch)
treececccb211d476342c0b329ea54c7412ca5ecb1de /g4f/client/async_client.py
parentImprove readme (diff)
downloadgpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.tar
gpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.tar.gz
gpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.tar.bz2
gpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.tar.lz
gpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.tar.xz
gpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.tar.zst
gpt4free-b7624b75a3ed6a48733ddaa0c6ea87227922d173.zip
Diffstat (limited to 'g4f/client/async_client.py')
-rw-r--r--g4f/client/async_client.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/g4f/client/async_client.py b/g4f/client/async_client.py
index dbfa6b70..2fe4640b 100644
--- a/g4f/client/async_client.py
+++ b/g4f/client/async_client.py
@@ -18,6 +18,7 @@ from ..Provider import ProviderUtils
from ..typing import Union, Messages, AsyncIterator, ImageType
from ..errors import NoImageResponseError, ProviderNotFoundError
from ..requests.aiohttp import get_connector
+from ..providers.conversation import BaseConversation
from ..image import ImageResponse as ImageProviderResponse, ImageDataResponse
try:
@@ -42,6 +43,9 @@ async def iter_response(
if isinstance(chunk, FinishReason):
finish_reason = chunk.reason
break
+ elif isinstance(chunk, BaseConversation):
+ yield chunk
+ continue
content += str(chunk)
count += 1
if max_tokens is not None and count >= max_tokens: