from json import loads from re import match from time import time, sleep from typing import Generator, Optional from uuid import uuid4 from fake_useragent import UserAgent from requests import post from tls_client import Session from .mail import Mail from .models import ForeFrontResponse class Account: @staticmethod def create(proxy: Optional[str] = None, logging: bool = False): proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else False start = time() mail_client = Mail(proxies) mail_token = None mail_address = mail_client.get_mail() # print(mail_address) client = Session(client_identifier='chrome110') client.proxies = proxies client.headers = { 'origin': 'https://accounts.forefront.ai', 'user-agent': UserAgent().random, } response = client.post( 'https://clerk.forefront.ai/v1/client/sign_ups?_clerk_js_version=4.32.6', data={'email_address': mail_address}, ) trace_token = response.json()['response']['id'] if logging: print(trace_token) response = client.post( f'https://clerk.forefront.ai/v1/client/sign_ups/{trace_token}/prepare_verification?_clerk_js_version=4.32.6', data={ 'strategy': 'email_code', }, ) if logging: print(response.text) if 'sign_up_attempt' not in response.text: return 'Failed to create account!' while True: sleep(1) for _ in mail_client.fetch_inbox(): if logging: print(mail_client.get_message_content(_['id'])) mail_token = match(r'(\d){5,6}', mail_client.get_message_content(_['id'])).group(0) if mail_token: break if logging: print(mail_token) response = client.post( f'https://clerk.forefront.ai/v1/client/sign_ups/{trace_token}/attempt_verification?_clerk_js_version=4.38.4', data={'code': mail_token, 'strategy': 'email_code'}, ) if logging: print(response.json()) token = response.json()['client']['sessions'][0]['last_active_token']['jwt'] with open('accounts.txt', 'a') as f: f.write(f'{mail_address}:{token}\n') if logging: print(time() - start) return token class StreamingCompletion: @staticmethod def create( token=None, chat_id=None, prompt='', action_type='new', default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0', # default model='gpt-4', ) -> Generator[ForeFrontResponse, None, None]: if not token: raise Exception('Token is required!') if not chat_id: chat_id = str(uuid4()) headers = { 'authority': 'chat-server.tenant-forefront-default.knative.chi.coreweave.com', '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', 'authorization': 'Bearer ' + token, 'cache-control': 'no-cache', 'content-type': 'application/json', 'origin': 'https://chat.forefront.ai', 'pragma': 'no-cache', 'referer': 'https://chat.forefront.ai/', 'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"', 'sec-ch-ua-mobile': '?0', 'sec-ch-ua-platform': '"macOS"', 'sec-fetch-dest': 'empty', 'sec-fetch-mode': 'cors', 'sec-fetch-site': 'cross-site', 'user-agent': UserAgent().random, } json_data = { 'text': prompt, 'action': action_type, 'parentId': chat_id, 'workspaceId': chat_id, 'messagePersona': default_persona, 'model': model, } for chunk in post( 'https://chat-server.tenant-forefront-default.knative.chi.coreweave.com/chat', headers=headers, json=json_data, stream=True, ).iter_lines(): if b'finish_reason":null' in chunk: data = loads(chunk.decode('utf-8').split('data: ')[1]) token = data['choices'][0]['delta'].get('content') if token is not None: yield ForeFrontResponse( **{ 'id': chat_id, 'object': 'text_completion', 'created': int(time()), 'text': token, 'model': model, 'choices': [{'text': token, 'index': 0, 'logprobs': None, 'finish_reason': 'stop'}], 'usage': { 'prompt_tokens': len(prompt), 'completion_tokens': len(token), 'total_tokens': len(prompt) + len(token), }, } ) class Completion: @staticmethod def create( token=None, chat_id=None, prompt='', action_type='new', default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0', # default model='gpt-4', ) -> ForeFrontResponse: text = '' final_response = None res = list(StreamingCompletion.create(token=token, prompt=prompt)) for response in StreamingCompletion.create( token=token, chat_id=chat_id, prompt=prompt, action_type=action_type, default_persona=default_persona, model=model, ): if response: final_response = response text += response.text if final_response: final_response.text = text else: raise Exception('Unable to get the response, Please try again') return final_response