From 88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Thu, 5 Oct 2023 05:13:37 +0200
Subject: Add AiAsk, Chatgpt4Online, ChatgptDemo  and ChatgptX Provider Fix
 Bing, Liaobots and ChatgptAi Provider Add "gpt_35_long" model and custom
 timeout

---
 g4f/Provider/AiAsk.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 g4f/Provider/AiAsk.py

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

diff --git a/g4f/Provider/AiAsk.py b/g4f/Provider/AiAsk.py
new file mode 100644
index 00000000..906b5575
--- /dev/null
+++ b/g4f/Provider/AiAsk.py
@@ -0,0 +1,43 @@
+from aiohttp import ClientSession
+from ..typing import AsyncGenerator
+from .base_provider import AsyncGeneratorProvider
+
+class AiAsk(AsyncGeneratorProvider):
+    url = "https://e.aiask.me"
+    supports_gpt_35_turbo = True
+    working = True
+
+    @classmethod
+    async def create_async_generator(
+        cls,
+        model: str,
+        messages: list[dict[str, str]],
+        timeout: int = 30,
+        **kwargs
+    ) -> AsyncGenerator:
+        headers = {
+            "accept": "application/json, text/plain, */*",
+            "origin": cls.url,
+            "referer": f"{cls.url}/chat",
+        }
+        async with ClientSession(headers=headers, timeout=timeout) as session:
+            data = {
+                "continuous": True,
+                "id": "fRMSQtuHl91A4De9cCvKD",
+                "list": messages,
+                "models": "0",
+                "prompt": "",
+                "temperature": kwargs.get("temperature", 0.5),
+                "title": "",
+            }
+            buffer = ""
+            rate_limit = "您的免费额度不够使用这个模型啦，请点击右上角登录继续使用！"
+            async with session.post(f"{cls.url}/v1/chat/gpt/", json=data) as response:
+                response.raise_for_status()
+                async for chunk in response.content.iter_any():
+                    buffer += chunk.decode()
+                    if not rate_limit.startswith(buffer):
+                        yield buffer
+                        buffer = ""
+                    elif buffer == rate_limit:
+                        raise RuntimeError("Rate limit reached")
\ No newline at end of file
-- 
cgit v1.2.3