0.2.2.0

3 files changed, 77 insertions, 1 deletions

diff --git a/g4f/Provider/FlowGpt.py b/g4f/Provider/FlowGpt.py
new file mode 100644
index 00000000..39192bf9
--- /dev/null
+++ b/g4f/Provider/FlowGpt.py
@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+import json
+from aiohttp import ClientSession
+
+from ..typing import AsyncResult, Messages
+from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
+
+class FlowGpt(AsyncGeneratorProvider, ProviderModelMixin):
+    url = "https://flowgpt.com/chat"
+    working = True
+    supports_gpt_35_turbo = True
+    supports_gpt_4 = True
+    supports_message_history = True
+    default_model = "gpt-3.5-turbo"
+    models = [
+        "gpt-4",
+        "gpt-3.5-turbo",
+        "gpt-3.5-long",
+        "google-gemini",
+        "claude-v2",
+        "llama2-13b"
+    ]
+    model_aliases = {
+        "gemini": "google-gemini",
+        "gemini-pro": "google-gemini"
+    }
+
+    @classmethod
+    async def create_async_generator(
+        cls,
+        model: str,
+        messages: Messages,
+        proxy: str = None,
+        **kwargs
+    ) -> AsyncResult:
+        model = cls.get_model(model)
+        headers = {
+            "User-Agent": "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:122.0) Gecko/20100101 Firefox/122.0",
+            "Accept": "*/*",
+            "Accept-Language": "en-US;q=0.7,en;q=0.3",
+            "Accept-Encoding": "gzip, deflate, br",
+            "Referer": "https://flowgpt.com/",
+            "Content-Type": "application/json",
+            "Authorization": "Bearer null",
+            "Origin": "https://flowgpt.com",
+            "Connection": "keep-alive",
+            "Sec-Fetch-Dest": "empty",
+            "Sec-Fetch-Mode": "cors",
+            "Sec-Fetch-Site": "same-site",
+            "TE": "trailers"
+        }
+        async with ClientSession(headers=headers) as session:
+            data = {
+                "model": model,
+                "nsfw": False,
+                "question": messages[-1]["content"],
+                "history": [{"role": "assistant", "content": "Hello, how can I help you today?"}, *messages[:-1]],
+                "system": kwargs.get("system_message", "You are helpful assistant. Follow the user's instructions carefully."),
+                "temperature": kwargs.get("temperature", 0.7),
+                "promptId": f"model-{model}",
+                "documentIds": [],
+                "chatFileDocumentIds": [],
+                "generateImage": False,
+                "generateAudio": False
+            }
+            async with session.post("https://backend-k8s.flowgpt.com/v2/chat-anonymous", json=data, proxy=proxy) as response:
+                response.raise_for_status()
+                async for chunk in response.content:
+                    if chunk.strip():
+                        message = json.loads(chunk)
+                        if "event" not in message:
+                            continue
+                        if message["event"] == "text":
+                            yield message["data"]
\ No newline at end of file
diff --git a/g4f/Provider/__init__.py b/g4f/Provider/__init__.py
index 270b6356..6cdc8806 100644
--- a/g4f/Provider/__init__.py
+++ b/g4f/Provider/__init__.py
@@ -31,6 +31,7 @@ from .ChatgptX        import ChatgptX
 from .Chatxyz         import Chatxyz
 from .DeepInfra       import DeepInfra
 from .FakeGpt         import FakeGpt
+from .FlowGpt         import FlowGpt
 from .FreeChatgpt     import FreeChatgpt
 from .FreeGpt         import FreeGpt
 from .GeekGpt         import GeekGpt
diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py
index 001f5a3c..556c3d9b 100644
--- a/g4f/Provider/needs_auth/OpenaiChat.py
+++ b/g4f/Provider/needs_auth/OpenaiChat.py
@@ -336,7 +336,7 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
         if cls._args is None and cookies is None:
             cookies = get_cookies("chat.openai.com", False)
         api_key = kwargs["access_token"] if "access_token" in kwargs else api_key
-        if api_key is None:
+        if api_key is None and cookies is not None:
             api_key = cookies["access_token"] if "access_token" in cookies else api_key
         if cls._args is None:
             cls._args = {