From eb0e2c6a93c3f21937457d13220ce2b7fca1f04a Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Mon, 2 Oct 2023 02:04:22 +0200
Subject: +Curl +Async +Stream Requests Update Model List

---
 g4f/Provider/FreeGpt.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'g4f/Provider/FreeGpt.py')

diff --git a/g4f/Provider/FreeGpt.py b/g4f/Provider/FreeGpt.py
index 534b69a5..092e1bb6 100644
--- a/g4f/Provider/FreeGpt.py
+++ b/g4f/Provider/FreeGpt.py
@@ -3,7 +3,7 @@ from __future__ import annotations
 import time, hashlib, random
 
 from ..typing import AsyncGenerator
-from g4f.requests import AsyncSession
+from ..requests import StreamSession
 from .base_provider import AsyncGeneratorProvider
 
 domains = [
@@ -23,7 +23,7 @@ class FreeGpt(AsyncGeneratorProvider):
         messages: list[dict[str, str]],
         **kwargs
     ) -> AsyncGenerator:
-        async with AsyncSession(impersonate="chrome107") as session:
+        async with StreamSession(impersonate="chrome107") as session:
             prompt = messages[-1]["content"]
             timestamp = int(time.time())
             data = {
@@ -35,7 +35,7 @@ class FreeGpt(AsyncGeneratorProvider):
             url = random.choice(domains)
             async with session.post(f"{url}/api/generate", 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
-- 
cgit v1.2.3