diff options
Diffstat (limited to '')
| -rw-r--r-- | gpt4free/hpgptai/__init__.py | 37 |
1 files changed, 28 insertions, 9 deletions
diff --git a/gpt4free/hpgptai/__init__.py b/gpt4free/hpgptai/__init__.py index c8772a19..f5d1f0ed 100644 --- a/gpt4free/hpgptai/__init__.py +++ b/gpt4free/hpgptai/__init__.py @@ -5,20 +5,26 @@ @File ：__init__.py.py @IDE ：PyCharm """ +import re import json -import requests +import base64 import random import string +import requests +from fake_useragent import UserAgent + class ChatCompletion: @staticmethod def create( messages: list, - context: str="Converse as if you were an AI assistant. Be friendly, creative.", - restNonce:str="9d6d743bd3", - proxy:str=None + context: str = "Converse as if you were an AI assistant. Be friendly, creative.", + restNonce: str = None, + proxy: str = None ): url = "https://chatgptlogin.ac/wp-json/ai-chatbot/v1/chat" + if not restNonce: + restNonce = ChatCompletion.get_restNonce(proxy) headers = { "Content-Type": "application/json", "X-Wp-Nonce": restNonce @@ -27,7 +33,7 @@ class ChatCompletion: data = { "env": "chatbot", "session": "N/A", - "prompt": ChatCompletion.__build_prompt(context,messages), + "prompt": ChatCompletion.__build_prompt(context, messages), "context": context, "messages": messages, "newMessage": messages[-1]["content"], @@ -48,7 +54,6 @@ class ChatCompletion: return res.json() return res.text - @staticmethod def randomStr(): return ''.join(random.choices(string.ascii_lowercase + string.digits, k=34))[:11] @@ -66,12 +71,26 @@ class ChatCompletion: prompt += '\n' + "AI: " return prompt - + @classmethod + def get_restNonce(cls, proxy: str = None): + url = "https://chatgptlogin.ac/" + headers = { + "Referer": "https://chatgptlogin.ac/", + "User-Agent": UserAgent().random + } + proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else None + res = requests.get(url, headers=headers, proxies=proxies) + src = re.search( + 'class="mwai-chat mwai-chatgpt">.*<span>Send</span></button></div></div></div> <script defer src="(.*?)">', + res.text).group(1) + decoded_string = base64.b64decode(src.split(",")[-1]).decode('utf-8') + restNonce = re.search(r"let restNonce = '(.*?)';", decoded_string).group(1) + return restNonce class Completion: @staticmethod - def create(prompt: str,proxy:str): + def create(prompt: str, proxy: str): messages = [ { "content": prompt, @@ -81,4 +100,4 @@ class Completion: "who": "User: ", }, ] - return ChatCompletion.create(messages=messages,proxy=proxy)
\ No newline at end of file + return ChatCompletion.create(messages=messages, proxy=proxy)
\ No newline at end of file |