diff options
author | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-05-21 20:39:31 +0200 |
---|---|---|
committer | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-05-21 20:39:31 +0200 |
commit | b7624b75a3ed6a48733ddaa0c6ea87227922d173 (patch) | |
tree | cecccb211d476342c0b329ea54c7412ca5ecb1de /g4f/client/client.py | |
parent | Improve readme (diff) | |
download | gpt4free-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 '')
-rw-r--r-- | g4f/client/client.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/g4f/client/client.py b/g4f/client/client.py index acf53c70..63bae4fe 100644 --- a/g4f/client/client.py +++ b/g4f/client/client.py @@ -6,6 +6,7 @@ import string from ..typing import Union, Iterator, Messages, ImageType from ..providers.types import BaseProvider, ProviderType, FinishReason +from ..providers.conversation import BaseConversation from ..image import ImageResponse as ImageProviderResponse from ..errors import NoImageResponseError from .stubs import ChatCompletion, ChatCompletionChunk, Image, ImagesResponse @@ -29,6 +30,9 @@ def iter_response( if isinstance(chunk, FinishReason): finish_reason = chunk.reason break + elif isinstance(chunk, BaseConversation): + yield chunk + continue content += str(chunk) if max_tokens is not None and idx + 1 >= max_tokens: finish_reason = "length" |