File size: 2,853 Bytes
51e07f2 96d1023 51e07f2 96d1023 1ee4f76 96d1023 51e07f2 1ee4f76 51e07f2 96d1023 1ee4f76 96d1023 1ee4f76 96d1023 1ee4f76 51e07f2 1ee4f76 51e07f2 96d1023 51e07f2 1ee4f76 51e07f2 1ee4f76 51e07f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import uuid
from pathlib import Path
from curl_cffi import requests
from tclogger import logger, OSEnver
secrets_path = Path(__file__).parents[1] / "secrets.json"
ENVER = OSEnver(secrets_path)
class OpenaiAPI:
def __init__(self):
self.init_requests_params()
def init_requests_params(self):
self.api_base = "https://chat.openai.com/backend-anon"
self.api_me = f"{self.api_base}/me"
self.api_models = f"{self.api_base}/models"
self.api_chat_requirements = f"{self.api_base}/sentinel/chat-requirements"
self.uuid = str(uuid.uuid4())
self.requests_headers = {
"Accept": "*/*",
"Accept-Encoding": "gzip, deflate, br, zstd",
"Accept-Language": "en-US,en;q=0.9",
"Cache-Control": "no-cache",
"Oai-Device-Id": self.uuid,
"Oai-Language": "en-US",
"Pragma": "no-cache",
"Referer": "https://chat.openai.com/",
"Sec-Ch-Ua": 'Google Chrome";v="123", "Not:A-Brand";v="8", "Chromium";v="123"',
"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/123.0.0.0 Safari/537.36",
}
http_proxy = ENVER["http_proxy"]
self.requests_proxies = {
"http": http_proxy,
"https": http_proxy,
}
def log_request(self, url, method="GET"):
if ENVER["http_proxy"]:
logger.note(f"> Using Proxy: {ENVER['http_proxy']}")
logger.note(f"> {method}: {url}", end=" ")
def log_response(self, res: requests.Response):
status_code = res.status_code
status_code_str = f"[{status_code}]"
if status_code == 200:
logger.success(status_code_str)
else:
logger.warn(f"uuid: {self.uuid}")
logger.warn(status_code_str)
logger.mesg(res.json())
def get_models(self):
self.log_request(self.api_models)
res = requests.get(
self.api_models,
headers=self.requests_headers,
proxies=self.requests_proxies,
timeout=10,
impersonate="chrome120",
)
self.log_response(res)
def auth(self):
self.log_request(self.api_models, method="POST")
res = requests.post(
self.api_chat_requirements,
headers=self.requests_headers,
proxies=self.requests_proxies,
timeout=10,
impersonate="chrome120",
)
self.log_response(res)
if __name__ == "__main__":
api = OpenaiAPI()
api.auth()
# python -m tests.openai
|