diff options
Diffstat (limited to '')
| -rw-r--r-- | g4f/Provider/ChatgptLogin.py | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/g4f/Provider/ChatgptLogin.py b/g4f/Provider/ChatgptLogin.py new file mode 100644 index 00000000..05bd1262 --- /dev/null +++ b/g4f/Provider/ChatgptLogin.py @@ -0,0 +1,133 @@ +import base64 +import os +import re + +import requests + +from ..typing import Any, CreateResult +from .base_provider import BaseProvider + + +class ChatgptLogin(BaseProvider): + url = "https://chatgptlogin.ac" + supports_gpt_35_turbo = True + + @staticmethod + def create_completion( + model: str, + messages: list[dict[str, str]], + stream: bool, + **kwargs: Any, + ) -> CreateResult: + headers = { + "authority": "chatgptlogin.ac", + "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", + "content-type": "application/json", + "origin": "https://chatgptlogin.ac", + "referer": "https://chatgptlogin.ac/use-chatgpt-free/", + "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", + "x-wp-nonce": _get_nonce(), + } + + conversation = _transform(messages) + + json_data = { + "env": "chatbot", + "session": "N/A", + "prompt": "Converse as if you were an AI assistant. Be friendly, creative.", + "context": "Converse as if you were an AI assistant. Be friendly, creative.", + "messages": conversation, + "newMessage": messages[-1]["content"], + "userName": '<div class="mwai-name-text">User:</div>', + "aiName": '<div class="mwai-name-text">AI:</div>', + "model": "gpt-3.5-turbo", + "temperature": kwargs.get("temperature", 0.8), + "maxTokens": 1024, + "maxResults": 1, + "apiKey": "", + "service": "openai", + "embeddingsIndex": "", + "stop": "", + "clientId": os.urandom(6).hex(), + } + + response = requests.post( + "https://chatgptlogin.ac/wp-json/ai-chatbot/v1/chat", + headers=headers, + json=json_data, + ) + response.raise_for_status() + yield response.json()["reply"] + + @classmethod + @property + def params(cls): + params = [ + ("model", "str"), + ("messages", "list[dict[str, str]]"), + ("stream", "bool"), + ("temperature", "float"), + ] + param = ", ".join([": ".join(p) for p in params]) + return f"g4f.provider.{cls.__name__} supports: ({param})" + + +def _get_nonce() -> str: + res = requests.get( + "https://chatgptlogin.ac/use-chatgpt-free/", + headers={ + "Referer": "https://chatgptlogin.ac/use-chatgpt-free/", + "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", + }, + ) + + result = re.search( + r'class="mwai-chat mwai-chatgpt">.*<span>Send</span></button></div></div></div> <script defer src="(.*?)">', + res.text, + ) + if result is None: + return "" + + src = result.group(1) + decoded_string = base64.b64decode(src.split(",")[-1]).decode("utf-8") + result = re.search(r"let restNonce = '(.*?)';", decoded_string) + + return "" if result is None else result.group(1) + + +def _transform(messages: list[dict[str, str]]) -> list[dict[str, Any]]: + return [ + { + "id": os.urandom(6).hex(), + "role": message["role"], + "content": message["content"], + "who": "AI: " if message["role"] == "assistant" else "User: ", + "html": _html_encode(message["content"]), + } + for message in messages + ] + + +def _html_encode(string: str) -> str: + table = { + '"': """, + "'": "'", + "&": "&", + ">": ">", + "<": "<", + "\n": "<br>", + "\t": " ", + " ": " ", + } + + for key in table: + string = string.replace(key, table[key]) + + return string |