diff options
author | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-02 02:04:22 +0200 |
---|---|---|
committer | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-02 02:04:22 +0200 |
commit | eb0e2c6a93c3f21937457d13220ce2b7fca1f04a (patch) | |
tree | e0139cd399052db29326d502f9d738d1943c7bce /g4f/Provider/ChatForAi.py | |
parent | Change event loop policy on windows (diff) | |
download | gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.tar gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.tar.gz gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.tar.bz2 gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.tar.lz gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.tar.xz gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.tar.zst gpt4free-eb0e2c6a93c3f21937457d13220ce2b7fca1f04a.zip |
Diffstat (limited to 'g4f/Provider/ChatForAi.py')
-rw-r--r-- | g4f/Provider/ChatForAi.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/g4f/Provider/ChatForAi.py b/g4f/Provider/ChatForAi.py index efb5478e..779799cf 100644 --- a/g4f/Provider/ChatForAi.py +++ b/g4f/Provider/ChatForAi.py @@ -3,7 +3,7 @@ from __future__ import annotations import time, hashlib from ..typing import AsyncGenerator -from g4f.requests import AsyncSession +from ..requests import StreamSession from .base_provider import AsyncGeneratorProvider @@ -19,7 +19,7 @@ class ChatForAi(AsyncGeneratorProvider): messages: list[dict[str, str]], **kwargs ) -> AsyncGenerator: - async with AsyncSession(impersonate="chrome107") as session: + async with StreamSession(impersonate="chrome107") as session: conversation_id = f"id_{int(time.time())}" prompt = messages[-1]["content"] timestamp = int(time.time()) @@ -43,7 +43,7 @@ class ChatForAi(AsyncGeneratorProvider): } async with session.post(f"{cls.url}/api/handle/provider-openai", json=data) as response: response.raise_for_status() - async for chunk in response.content.iter_any(): + async for chunk in response.iter_content(): yield chunk.decode() @classmethod |