From 126496d3cacd06a4fa8cbb4e5bde417ce6bb5b4a Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner.lohaus@netformic.com>
Date: Fri, 25 Aug 2023 06:41:32 +0200
Subject: Add OpenaiChat and Hugchat Provider Add tests for providers with auth
 Improve async support / 2x faster Shared get_cookies by domain  function

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

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

diff --git a/g4f/Provider/OpenaiChat.py b/g4f/Provider/OpenaiChat.py
new file mode 100644
index 00000000..cca258b3
--- /dev/null
+++ b/g4f/Provider/OpenaiChat.py
@@ -0,0 +1,74 @@
+has_module = True
+try:
+    from revChatGPT.V1 import AsyncChatbot
+except ImportError:
+    has_module = False
+from .base_provider import AsyncGeneratorProvider, get_cookies
+from ..typing import AsyncGenerator
+
+class OpenaiChat(AsyncGeneratorProvider):
+    url = "https://chat.openai.com"
+    needs_auth = True
+    working = has_module
+    supports_gpt_35_turbo = True
+    supports_gpt_4 = True
+    supports_stream = True
+
+    @classmethod
+    async def create_async_generator(
+        cls,
+        model: str,
+        messages: list[dict[str, str]],
+        proxy: str = None,
+        access_token: str = None,
+        cookies: dict = None,
+        **kwargs
+    ) -> AsyncGenerator:
+        
+        config = {"access_token": access_token, "model": model}
+        if proxy:
+            if "://" not in proxy:
+                proxy = f"http://{proxy}"
+            config["proxy"] = proxy
+
+        bot = AsyncChatbot(
+            config=config
+        )
+
+        if not access_token:
+            cookies = cookies if cookies else get_cookies("chat.openai.com")
+            response = await bot.session.get("https://chat.openai.com/api/auth/session", cookies=cookies)
+            access_token = response.json()["accessToken"]
+            bot.set_access_token(access_token)
+
+        if len(messages) > 1:
+            formatted = "\n".join(
+                ["%s: %s" % ((message["role"]).capitalize(), message["content"]) for message in messages]
+            )
+            prompt = f"{formatted}\nAssistant:"
+        else:
+            prompt = messages.pop()["content"]
+
+        returned = None
+        async for message in bot.ask(prompt):
+            message = message["message"]
+            if returned:
+                if message.startswith(returned):
+                    new = message[len(returned):]
+                    if new:
+                        yield new
+            else:
+                yield message
+            returned = message
+
+    @classmethod
+    @property
+    def params(cls):
+        params = [
+            ("model", "str"),
+            ("messages", "list[dict[str, str]]"),
+            ("stream", "bool"),
+            ("proxy", "str"),
+        ]
+        param = ", ".join([": ".join(p) for p in params])
+        return f"g4f.provider.{cls.__name__} supports: ({param})"
-- 
cgit v1.2.3