1
0
Fork 0

refactored code

This commit is contained in:
Raju Komati 2023-05-01 20:04:45 +05:30
parent 1eefa094e3
commit 05494ebbb1
No known key found for this signature in database
GPG key ID: A581A5D67A8EB090
9 changed files with 42 additions and 39 deletions

View file

@ -4,8 +4,8 @@ from gpt4free import cocalc
from gpt4free import forefront
from gpt4free import quora
from gpt4free import theb
from gpt4free import you
from gpt4free import usesless
from gpt4free import you
class Provider(Enum):
@ -24,7 +24,6 @@ class Completion:
@staticmethod
def create(provider: Provider, prompt: str, **kwargs) -> str:
"""
Invokes the given provider with given prompt and addition arguments and returns the string response
@ -50,7 +49,7 @@ class Completion:
@staticmethod
def __useless_service(prompt: str, **kwargs) -> str:
return usesless.Completion.create(prompt = prompt, **kwargs)
return usesless.Completion.create(prompt=prompt, **kwargs)
@staticmethod
def __you_service(prompt: str, **kwargs) -> str:

View file

@ -5,8 +5,8 @@ from typing import Generator, Optional
from uuid import uuid4
from fake_useragent import UserAgent
from requests import post
from pymailtm import MailTm, Message
from requests import post
from tls_client import Session
from .typing import ForeFrontResponse
@ -43,10 +43,7 @@ class Account:
response = client.post(
f'https://clerk.forefront.ai/v1/client/sign_ups/{trace_token}/prepare_verification?_clerk_js_version=4.38.4',
data={
'strategy': 'email_link',
'redirect_url': 'https://accounts.forefront.ai/sign-up/verify'
},
data={'strategy': 'email_link', 'redirect_url': 'https://accounts.forefront.ai/sign-up/verify'},
)
if logging:
@ -93,14 +90,14 @@ class StreamingCompletion:
action_type='new',
default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0', # default
model='gpt-4',
proxy=None
proxy=None,
) -> Generator[ForeFrontResponse, None, None]:
if not token:
raise Exception('Token is required!')
if not chat_id:
chat_id = str(uuid4())
proxies = { 'http': 'http://' + proxy, 'https': 'http://' + proxy } if proxy else None
proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else None
headers = {
'authority': 'chat-server.tenant-forefront-default.knative.chi.coreweave.com',
@ -168,7 +165,7 @@ class Completion:
action_type='new',
default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0', # default
model='gpt-4',
proxy=None
proxy=None,
) -> ForeFrontResponse:
text = ''
final_response = None
@ -179,7 +176,7 @@ class Completion:
action_type=action_type,
default_persona=default_persona,
model=model,
proxy=proxy
proxy=proxy,
):
if response:
final_response = response
@ -191,4 +188,3 @@ class Completion:
raise Exception('Unable to get the response, Please try again')
return final_response

View file

@ -1,4 +1,5 @@
from typing import Any, List
from pydantic import BaseModel

View file

@ -1,7 +1,10 @@
from requests import Session
from time import sleep
from json import loads
from re import findall
from time import sleep
from requests import Session
class Mail:
def __init__(self) -> None:
self.client = Session()
@ -9,27 +12,32 @@ class Mail:
self.cookies = {'acceptcookie': 'true'}
self.cookies["ci_session"] = self.client.cookies.get_dict()["ci_session"]
self.email = None
def get_mail(self):
respone=self.client.post("https://etempmail.com/getEmailAddress")
#cookies
respone = self.client.post("https://etempmail.com/getEmailAddress")
# cookies
self.cookies["lisansimo"] = eval(respone.text)["recover_key"]
self.email = eval(respone.text)["address"]
return self.email
def get_message(self):
print("Waiting for message...")
while True:
sleep(5)
respone=self.client.post("https://etempmail.com/getInbox")
mail_token=loads(respone.text)
respone = self.client.post("https://etempmail.com/getInbox")
mail_token = loads(respone.text)
print(self.client.cookies.get_dict())
if len(mail_token) == 1:
break
params = {'id': '1',}
self.mail_context = self.client.post("https://etempmail.com/getInbox",params=params)
params = {
'id': '1',
}
self.mail_context = self.client.post("https://etempmail.com/getInbox", params=params)
self.mail_context = eval(self.mail_context.text)[0]["body"]
return self.mail_context
#,cookies=self.cookies
# ,cookies=self.cookies
def get_verification_code(self):
message = self.mail_context
code = findall(r';">(\d{6,7})</div>', message)[0]

View file

@ -19,7 +19,7 @@ class Completion:
stream_completed = False
@staticmethod
def request(prompt: str, proxy: Optional[str]=None):
def request(prompt: str, proxy: Optional[str] = None):
headers = {
'authority': 'chatbot.theb.ai',
'content-type': 'application/json',
@ -40,7 +40,7 @@ class Completion:
Completion.stream_completed = True
@staticmethod
def create(prompt: str, proxy: Optional[str]=None) -> Generator[str, None, None]:
def create(prompt: str, proxy: Optional[str] = None) -> Generator[str, None, None]:
Thread(target=Completion.request, args=[prompt, proxy]).start()
while not Completion.stream_completed or not Completion.message_queue.empty():

View file

@ -1,6 +1,7 @@
import requests
import json
import requests
class Completion:
headers = {

View file

@ -30,12 +30,12 @@ class Completion:
include_links: bool = False,
detailed: bool = False,
debug: bool = False,
proxy: Optional[str] = None
proxy: Optional[str] = None,
) -> PoeResponse:
if chat is None:
chat = []
proxies = { 'http': 'http://' + proxy, 'https': 'http://' + proxy } if proxy else {}
proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else {}
client = Session(client_identifier='chrome_108')
client.headers = Completion.__get_headers()

View file

@ -11,7 +11,6 @@ while True:
print(f"Answer: {req['text']}")
message_id = req["id"]
import gpt4free
message_id = ""
@ -20,8 +19,7 @@ while True:
if prompt == "!stop":
break
req = gpt4free.Completion.create(provider = gpt4free.Provider.UseLess,
prompt=prompt, parentMessageId=message_id)
req = gpt4free.Completion.create(provider=gpt4free.Provider.UseLess, prompt=prompt, parentMessageId=message_id)
print(f"Answer: {req['text']}")
message_id = req["id"]