:gem: [Feature] New api for chat_requiements, and moduralize logs

tests/openai.py

@@ -18,12 +18,13 @@ class OpenaiAPI:
         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": str(uuid.uuid4()),
+            "Oai-Device-Id": self.uuid,
             "Oai-Language": "en-US",
             "Pragma": "no-cache",
             "Referer": "https://chat.openai.com/",
@@ -42,12 +43,24 @@ class OpenaiAPI:
             "https": http_proxy,
         }
 
-    def get_models(self):
+    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.note(f"> Get: {self.api_models}")
+        logger.mesg(res.json())
 
+    def get_models(self):
+        self.log_request(self.api_models)
         res = requests.get(
             self.api_models,
             headers=self.requests_headers,
@@ -56,12 +69,23 @@ class OpenaiAPI:
             impersonate="chrome120",
         )
 
-        logger.warn(res.status_code)
-        logger.mesg(res.json())
+        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.get_models()
+    api.auth()
 
     # python -m tests.openai