From efd75a11b871d61ac31b0e274acdfb33daba361d Mon Sep 17 00:00:00 2001
From: abc <98614666+xtekky@users.noreply.github.com>
Date: Sun, 27 Aug 2023 17:37:44 +0200
Subject: ~ | code styling

---
 g4f/Provider/ChatgptAi.py | 66 ++++++++++++++++++++++-------------------------
 1 file changed, 31 insertions(+), 35 deletions(-)

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

diff --git a/g4f/Provider/ChatgptAi.py b/g4f/Provider/ChatgptAi.py
index 53518f65..13c591cb 100644
--- a/g4f/Provider/ChatgptAi.py
+++ b/g4f/Provider/ChatgptAi.py
@@ -1,23 +1,20 @@
-import re
+import re, requests
 
-import requests
-
-from ..typing import Any, CreateResult
+from ..typing       import Any, CreateResult
 from .base_provider import BaseProvider
 
 
 class ChatgptAi(BaseProvider):
-    url = "https://chatgpt.ai/gpt-4/"
-    working = True
-    supports_gpt_4 = True
+    url: str        = "https://chatgpt.ai/gpt-4/"
+    working         = True
+    supports_gpt_4  = True
 
     @staticmethod
     def create_completion(
         model: str,
         messages: list[dict[str, str]],
-        stream: bool,
-        **kwargs: Any,
-    ) -> CreateResult:
+        stream: bool, **kwargs: Any) -> CreateResult:
+        
         chat = ""
         for message in messages:
             chat += "%s: %s\n" % (message["role"], message["content"])
@@ -26,36 +23,35 @@ class ChatgptAi(BaseProvider):
         response = requests.get("https://chatgpt.ai/")
         nonce, post_id, _, bot_id = re.findall(
             r'data-nonce="(.*)"\n     data-post-id="(.*)"\n     data-url="(.*)"\n     data-bot-id="(.*)"\n     data-width',
-            response.text,
-        )[0]
+            response.text)[0]
 
         headers = {
-            "authority": "chatgpt.ai",
-            "accept": "*/*",
-            "accept-language": "en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3",
-            "cache-control": "no-cache",
-            "origin": "https://chatgpt.ai",
-            "pragma": "no-cache",
-            "referer": "https://chatgpt.ai/gpt-4/",
-            "sec-ch-ua": '"Not.A/Brand";v="8", "Chromium";v="114", "Google Chrome";v="114"',
-            "sec-ch-ua-mobile": "?0",
-            "sec-ch-ua-platform": '"Windows"',
-            "sec-fetch-dest": "empty",
-            "sec-fetch-mode": "cors",
-            "sec-fetch-site": "same-origin",
-            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36",
+            "authority"          : "chatgpt.ai",
+            "accept"             : "*/*",
+            "accept-language"    : "en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3",
+            "cache-control"      : "no-cache",
+            "origin"             : "https://chatgpt.ai",
+            "pragma"             : "no-cache",
+            "referer"            : "https://chatgpt.ai/gpt-4/",
+            "sec-ch-ua"          : '"Not.A/Brand";v="8", "Chromium";v="114", "Google Chrome";v="114"',
+            "sec-ch-ua-mobile"   : "?0",
+            "sec-ch-ua-platform" : '"Windows"',
+            "sec-fetch-dest"     : "empty",
+            "sec-fetch-mode"     : "cors",
+            "sec-fetch-site"     : "same-origin",
+            "user-agent"         : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36",
         }
         data = {
-            "_wpnonce": nonce,
-            "post_id": post_id,
-            "url": "https://chatgpt.ai/gpt-4",
-            "action": "wpaicg_chat_shortcode_message",
-            "message": chat,
-            "bot_id": bot_id,
+            "_wpnonce" : nonce,
+            "post_id"  : post_id,
+            "url"      : "https://chatgpt.ai/gpt-4",
+            "action"   : "wpaicg_chat_shortcode_message",
+            "message"  : chat,
+            "bot_id"   : bot_id,
         }
 
         response = requests.post(
-            "https://chatgpt.ai/wp-admin/admin-ajax.php", headers=headers, data=data
-        )
+            "https://chatgpt.ai/wp-admin/admin-ajax.php", headers=headers, data=data)
+        
         response.raise_for_status()
-        yield response.json()["data"]
+        yield response.json()["data"]
\ No newline at end of file
-- 
cgit v1.2.3