9 files changed, 56 insertions, 77 deletions

diff --git a/testing/accounts.txt b/testing/accounts.txt
new file mode 100644
index 00000000..5d687ab7
--- /dev/null
+++ b/testing/accounts.txt
@@ -0,0 +1,3 @@
+idkaovprvbrjbb@bugfoo.com:eyJhbGciOiJSUzI1NiIsImtpZCI6Imluc18yTzZ3UTFYd3dxVFdXUWUyQ1VYZHZ2bnNaY2UiLCJ0eXAiOiJKV1QifQ.eyJhenAiOiJodHRwczovL2FjY291bnRzLmZvcmVmcm9udC5haSIsImV4cCI6MTY4MjYxNDUxOSwiaWF0IjoxNjgyNjE0NDU5LCJpc3MiOiJodHRwczovL2NsZXJrLmZvcmVmcm9udC5haSIsIm5iZiI6MTY4MjYxNDQ0OSwic2lkIjoic2Vzc18yUDFBeFNHNnFHdzhWSFNla2lhMVB0NWJYdW8iLCJzdWIiOiJ1c2VyXzJQMUF4TWlNRzFLb00xeFFUQVExcFFjd0d3dyJ9.Nw82ExUQMvJE8oVAmUYl81dS9J43YK0AQYwLJjE5fCozanqQi946q_6DRRQz-9kkm-Zd60zuFzl_ANgZ9lLOOSMd89oJArbu3VUOA8Q-BOtUE0INmwvPCP7KtLDEVRGEsF8oIddwE1Da1rEToBEHrwBuzOVbRvoKqbJ7KZrRB7tHzUVyik3n7K3bddyecIA5Gtj8Nmc_NvwVaBelHOfjH0eoU7y0oza-bOJPNCRosjV3kFdoYhHNtNvQkfIF7MSmqyeC-NHhDVdmv57ehkVdr_H0_MyoT2kNWEk1ETzST36Lhu06yaaWzAPhjpcThN8Q7OFNXSHUAhTJibMHER6UGw
+vpmysvdoydnumi@bugfoo.com:eyJhbGciOiJSUzI1NiIsImtpZCI6Imluc18yTzZ3UTFYd3dxVFdXUWUyQ1VYZHZ2bnNaY2UiLCJ0eXAiOiJKV1QifQ.eyJhenAiOiJodHRwczovL2FjY291bnRzLmZvcmVmcm9udC5haSIsImV4cCI6MTY4MjYyMDM4MSwiaWF0IjoxNjgyNjIwMzIxLCJpc3MiOiJodHRwczovL2NsZXJrLmZvcmVmcm9udC5haSIsIm5iZiI6MTY4MjYyMDMxMSwic2lkIjoic2Vzc18yUDFNcTM1Tmtld3lIa2NOYUJvNDcyV1haR0MiLCJzdWIiOiJ1c2VyXzJQMU1xNFRYcmJlYVhSYVdKUWFjZGRZTVNyMiJ9.PVQlpsxJxgTOaizJ34AGkP3yJzhDAUFBw29gEP0cHCuUZcAUk1MVKeXYoXr32nqQmUmPO5-kmbp3p639idzA3E7EYnOCJ6xOTs6qB4QvmtEfAJQLD84-h5cYXkLzpqcWZeszDkoAQAZKhmI5kGn_OJzblYVodSjz92klrGSASuqIwqywvGNKb2NwQ80yRrNHUcGeI6tRPtMdZ0ieOi04SphXpyB7-JLZuco8nwqivs3uX97Jtkw3EYsTjhp5MKwfG9hGjHk54IwxAJtbMCSMDxDcOYxO2vxVL4W4syCHCB3yUJANmuFjL7mIv76uy4w0YIJtDFAYn3oad7dbHWKtSQ
+xjwollilwtumif@bugfoo.com:eyJhbGciOiJSUzI1NiIsImtpZCI6Imluc18yTzZ3UTFYd3dxVFdXUWUyQ1VYZHZ2bnNaY2UiLCJ0eXAiOiJKV1QifQ.eyJhenAiOiJodHRwczovL2FjY291bnRzLmZvcmVmcm9udC5haSIsImV4cCI6MTY4MjYyMDcwNywiaWF0IjoxNjgyNjIwNjQ3LCJpc3MiOiJodHRwczovL2NsZXJrLmZvcmVmcm9udC5haSIsIm5iZiI6MTY4MjYyMDYzNywic2lkIjoic2Vzc18yUDFOVjBjMm5JdWVLak82SjVWUTVoTHVZTWQiLCJzdWIiOiJ1c2VyXzJQMU5VelpRWjZUZXRETkhnT0dWOHlaZ09pciJ9.O6GsP0QtekV5UMxvMEpUa65I5E_2ahihRbLGAfjSAnyWNY72fQgCgbi75n4L85ej9V9qeDxMqCPF0BA5hXN606ybdlBY_HvlvwRIgyKU2r5JyaltOow7bg6vu_qwj91YejvTDHR7w_cwq-FF4VRYysiJfoc-8jcO2HZl8HM0R8aaSf8xDHy0wCfpBYKN2Mw3-mP23Z08NxlehKXNdKoQq_zKQOJbl0DxfWaoFB6b-xocIl8RO2PR6r1yf0VzHaPNxa1PUrPcs94iaqXuOjpCzqEqO9kZf6WM0rWKRx6-u9A0BECaeirfYpNvdaNQHzIRSndzpTO-Kvnww6HtjDD-tg
diff --git a/testing/forefront_test.py b/testing/forefront_test.py
index 4d5f4bc1..092d1a74 100644
--- a/testing/forefront_test.py
+++ b/testing/forefront_test.py
@@ -1,10 +1,9 @@
-import forefront
+from openai_rev import forefront
 
 # create an account
 token = forefront.Account.create(logging=True)
 print(token)
 
 # get a response
-for response in forefront.StreamingCompletion.create(token=token,
-                                                     prompt='hello world', model='gpt-4'):
-    print(response.completion.choices[0].text, end='')
+for response in forefront.StreamingCompletion.create(token=token, prompt='hello world', model='gpt-4'):
+    print(response)
diff --git a/testing/poe_account_create_test.py b/testing/poe_account_create_test.py
index 5d435b1f..24751c4a 100644
--- a/testing/poe_account_create_test.py
+++ b/testing/poe_account_create_test.py
@@ -2,11 +2,10 @@ from hashlib import md5
 from json import dumps
 from re import findall
 
-from tls_client import Session as TLS
-from twocaptcha import TwoCaptcha
-
 from quora import extract_formkey
 from quora.mail import Emailnator
+from tls_client import Session as TLS
+from twocaptcha import TwoCaptcha
 
 solver = TwoCaptcha('72747bf24a9d89b4dcc1b24875efd358')
 
@@ -14,14 +13,13 @@ solver = TwoCaptcha('72747bf24a9d89b4dcc1b24875efd358')
 class Account:
     def create(proxy: None or str = None, logging: bool = False, enable_bot_creation: bool = False):
         client = TLS(client_identifier='chrome110')
-        client.proxies = {
-            'http': f'http://{proxy}',
-            'https': f'http://{proxy}'} if proxy else None
+        client.proxies = {'http': f'http://{proxy}', 'https': f'http://{proxy}'} if proxy else None
 
         mail_client = Emailnator()
         mail_address = mail_client.get_mail()
 
-        if logging: print('email', mail_address)
+        if logging:
+            print('email', mail_address)
 
         client.headers = {
             'authority': 'poe.com',
@@ -39,29 +37,30 @@ class Account:
             'sec-fetch-dest': 'empty',
             'sec-fetch-mode': 'cors',
             'sec-fetch-site': 'same-origin',
-            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
         }
 
         client.headers["poe-formkey"] = extract_formkey(client.get('https://poe.com/login').text)
         client.headers["poe-tchannel"] = client.get('https://poe.com/api/settings').json()['tchannelData']['channel']
 
         # token = reCaptchaV3('https://www.recaptcha.net/recaptcha/enterprise/anchor?ar=1&k=6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG&co=aHR0cHM6Ly9wb2UuY29tOjQ0Mw..&hl=en&v=4PnKmGB9wRHh1i04o7YUICeI&size=invisible&cb=bi6ivxoskyal')
-        token = solver.recaptcha(sitekey='6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG',
-                                 url='https://poe.com/login?redirect_url=%2F',
-                                 version='v3',
-                                 enterprise=1,
-                                 invisible=1,
-                                 action='login', )['code']
-
-        payload = dumps(separators=(',', ':'), obj={
-            'queryName': 'MainSignupLoginSection_sendVerificationCodeMutation_Mutation',
-            'variables': {
-                'emailAddress': mail_address,
-                'phoneNumber': None,
-                'recaptchaToken': token
+        token = solver.recaptcha(
+            sitekey='6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG',
+            url='https://poe.com/login?redirect_url=%2F',
+            version='v3',
+            enterprise=1,
+            invisible=1,
+            action='login',
+        )['code']
+
+        payload = dumps(
+            separators=(',', ':'),
+            obj={
+                'queryName': 'MainSignupLoginSection_sendVerificationCodeMutation_Mutation',
+                'variables': {'emailAddress': mail_address, 'phoneNumber': None, 'recaptchaToken': token},
+                'query': 'mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n  $emailAddress: String\n  $phoneNumber: String\n  $recaptchaToken: String\n) {\n  sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n    status\n    errorMessage\n  }\n}\n',
             },
-            'query': 'mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n  $emailAddress: String\n  $phoneNumber: String\n  $recaptchaToken: String\n) {\n  sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n    status\n    errorMessage\n  }\n}\n',
-        })
+        )
 
         base_string = payload + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
         client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
@@ -74,31 +73,34 @@ class Account:
             print('please try using a proxy / wait for fix')
 
         if 'Bad Request' in response.text:
-            if logging: print('bad request, retrying...', response.json())
+            if logging:
+                print('bad request, retrying...', response.json())
             quit()
 
-        if logging: print('send_code', response.json())
+        if logging:
+            print('send_code', response.json())
 
         mail_content = mail_client.get_message()
         mail_token = findall(r';">(\d{6,7})</div>', mail_content)[0]
 
-        if logging: print('code', mail_token)
+        if logging:
+            print('code', mail_token)
 
-        payload = dumps(separators=(',', ':'), obj={
-            "queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
-            "variables": {
-                "verificationCode": str(mail_token),
-                "emailAddress": mail_address,
-                "phoneNumber": None
+        payload = dumps(
+            separators=(',', ':'),
+            obj={
+                "queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
+                "variables": {"verificationCode": str(mail_token), "emailAddress": mail_address, "phoneNumber": None},
+                "query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n  $verificationCode: String!\n  $emailAddress: String\n  $phoneNumber: String\n) {\n  signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n    status\n    errorMessage\n  }\n}\n",
             },
-            "query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n  $verificationCode: String!\n  $emailAddress: String\n  $phoneNumber: String\n) {\n  signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n    status\n    errorMessage\n  }\n}\n"
-        })
+        )
 
         base_string = payload + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
         client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
 
         response = client.post('https://poe.com/api/gql_POST', data=payload)
-        if logging: print('verify_code', response.json())
+        if logging:
+            print('verify_code', response.json())
 
 
 Account.create(proxy='xtekky:wegwgwegwed_streaming-1@geo.iproyal.com:12321', logging=True)
diff --git a/testing/poe_test.py b/testing/poe_test.py
index 8d527879..5cab067c 100644
--- a/testing/poe_test.py
+++ b/testing/poe_test.py
@@ -7,7 +7,5 @@ print('token', token)
 
 sleep(2)
 
-for response in quora.StreamingCompletion.create(model='gpt-3.5-turbo',
-                                                 prompt='hello world',
-                                                 token=token):
-    print(response.completion.choices[0].text, end="", flush=True)
+for response in quora.StreamingCompletion.create(model='ChatGPT', prompt='hello world', token=token):
+    print(response.text, flush=True)
diff --git a/testing/quora_test_2.py b/testing/quora_test_2.py
index d5316946..5d06f9ed 100644
--- a/testing/quora_test_2.py
+++ b/testing/quora_test_2.py
@@ -1,17 +1,12 @@
-import quora
+from openai_rev import quora
 
 token = quora.Account.create(logging=True, enable_bot_creation=True)
 
 model = quora.Model.create(
-    token=token,
-    model='gpt-3.5-turbo',  # or claude-instant-v1.0
-    system_prompt='you are ChatGPT a large language model ...'
+    token=token, model='ChatGPT', system_prompt='you are ChatGPT a large language model ...'  # or claude-instant-v1.0
 )
 
 print(model.name)
 
-for response in quora.StreamingCompletion.create(
-        custom_model=model.name,
-        prompt='hello world',
-        token=token):
-    print(response.completion.choices[0].text)
+for response in quora.StreamingCompletion.create(custom_model=model.name, prompt='hello world', token=token):
+    print(response.text)
diff --git a/testing/sqlchat_test.py b/testing/sqlchat_test.py
index 577d85a6..1db71be2 100644
--- a/testing/sqlchat_test.py
+++ b/testing/sqlchat_test.py
@@ -1,6 +1,4 @@
 import sqlchat
 
-for response in sqlchat.StreamCompletion.create(
-        prompt='write python code to reverse a string',
-        messages=[]):
+for response in sqlchat.StreamCompletion.create(prompt='write python code to reverse a string', messages=[]):
     print(response.completion.choices[0].text, end='')
diff --git a/testing/t3nsor_test.py b/testing/t3nsor_test.py
index 1506a1b4..6d36400d 100644
--- a/testing/t3nsor_test.py
+++ b/testing/t3nsor_test.py
@@ -1,6 +1,4 @@
 import t3nsor
 
-for response in t3nsor.StreamCompletion.create(
-        prompt='write python code to reverse a string',
-        messages=[]):
+for response in t3nsor.StreamCompletion.create(prompt='write python code to reverse a string', messages=[]):
     print(response.completion.choices[0].text)
diff --git a/testing/writesonic_test.py b/testing/writesonic_test.py
index 5c68bbe0..aff847f0 100644
--- a/testing/writesonic_test.py
+++ b/testing/writesonic_test.py
@@ -4,17 +4,14 @@ import writesonic
 # create account (3-4s)
 account = writesonic.Account.create(logging=True)
 
-# with loging: 
+# with loging:
 # 2023-04-06 21:50:25 INFO __main__ -> register success : '{"id":"51aa0809-3053-44f7-922a...' (2s)
 # 2023-04-06 21:50:25 INFO __main__ -> id : '51aa0809-3053-44f7-922a-2b85d8d07edf'
 # 2023-04-06 21:50:25 INFO __main__ -> token : 'eyJhbGciOiJIUzI1NiIsInR5cCI6Ik...'
 # 2023-04-06 21:50:28 INFO __main__ -> got key : '194158c4-d249-4be0-82c6-5049e869533c' (2s)
 
 # simple completion
-response = writesonic.Completion.create(
-    api_key=account.key,
-    prompt='hello world'
-)
+response = writesonic.Completion.create(api_key=account.key, prompt='hello world')
 
 print(response.completion.choices[0].text)  # Hello! How may I assist you today?
 
@@ -24,16 +21,7 @@ response = writesonic.Completion.create(
     api_key=account.key,
     prompt='what is my name ?',
     enable_memory=True,
-    history_data=[
-        {
-            'is_sent': True,
-            'message': 'my name is Tekky'
-        },
-        {
-            'is_sent': False,
-            'message': 'hello Tekky'
-        }
-    ]
+    history_data=[{'is_sent': True, 'message': 'my name is Tekky'}, {'is_sent': False, 'message': 'hello Tekky'}],
 )
 
 print(response.completion.choices[0].text)  # Your name is Tekky.
@@ -41,9 +29,7 @@ print(response.completion.choices[0].text)  # Your name is Tekky.
 # enable internet
 
 response = writesonic.Completion.create(
-    api_key=account.key,
-    prompt='who won the quatar world cup ?',
-    enable_google_results=True
+    api_key=account.key, prompt='who won the quatar world cup ?', enable_google_results=True
 )
 
 print(response.completion.choices[0].text)  # Argentina won the 2022 FIFA World Cup tournament held in Qatar ...
diff --git a/testing/you_test.py b/testing/you_test.py
index 6ba12f57..34800301 100644
--- a/testing/you_test.py
+++ b/testing/you_test.py
@@ -1,4 +1,4 @@
-import you
+from openai_rev import you
 
 # simple request with links and details
 response = you.Completion.create(prompt="hello world", detailed=True, include_links=True)