summaryrefslogtreecommitdiffstats
path: root/openai_rev
diff options
context:
space:
mode:
Diffstat (limited to 'openai_rev')
-rw-r--r--openai_rev/__init__.py0
-rw-r--r--openai_rev/forefront/README.md15
-rw-r--r--openai_rev/forefront/__init__.py180
-rw-r--r--openai_rev/forefront/mail.py52
-rw-r--r--openai_rev/forefront/models.py26
-rw-r--r--openai_rev/openai_rev.py35
-rw-r--r--openai_rev/phind/README.md37
-rw-r--r--openai_rev/quora/README.md68
-rw-r--r--openai_rev/quora/__init__.py485
-rw-r--r--openai_rev/quora/api.py545
-rw-r--r--openai_rev/quora/cookies.txt30
-rw-r--r--openai_rev/quora/graphql/AddHumanMessageMutation.graphql52
-rw-r--r--openai_rev/quora/graphql/AddMessageBreakMutation.graphql17
-rw-r--r--openai_rev/quora/graphql/AutoSubscriptionMutation.graphql7
-rw-r--r--openai_rev/quora/graphql/BioFragment.graphql8
-rw-r--r--openai_rev/quora/graphql/ChatAddedSubscription.graphql5
-rw-r--r--openai_rev/quora/graphql/ChatFragment.graphql6
-rw-r--r--openai_rev/quora/graphql/ChatListPaginationQuery.graphql378
-rw-r--r--openai_rev/quora/graphql/ChatPaginationQuery.graphql26
-rw-r--r--openai_rev/quora/graphql/ChatViewQuery.graphql8
-rw-r--r--openai_rev/quora/graphql/DeleteHumanMessagesMutation.graphql7
-rw-r--r--openai_rev/quora/graphql/DeleteMessageMutation.graphql7
-rw-r--r--openai_rev/quora/graphql/HandleFragment.graphql8
-rw-r--r--openai_rev/quora/graphql/LoginWithVerificationCodeMutation.graphql13
-rw-r--r--openai_rev/quora/graphql/MessageAddedSubscription.graphql100
-rw-r--r--openai_rev/quora/graphql/MessageDeletedSubscription.graphql6
-rw-r--r--openai_rev/quora/graphql/MessageFragment.graphql13
-rw-r--r--openai_rev/quora/graphql/MessageRemoveVoteMutation.graphql7
-rw-r--r--openai_rev/quora/graphql/MessageSetVoteMutation.graphql7
-rw-r--r--openai_rev/quora/graphql/PoeBotCreateMutation.graphql73
-rw-r--r--openai_rev/quora/graphql/PoeBotEditMutation.graphql24
-rw-r--r--openai_rev/quora/graphql/SendMessageMutation.graphql40
-rw-r--r--openai_rev/quora/graphql/SendVerificationCodeForLoginMutation.graphql12
-rw-r--r--openai_rev/quora/graphql/ShareMessagesMutation.graphql9
-rw-r--r--openai_rev/quora/graphql/SignupWithVerificationCodeMutation.graphql13
-rw-r--r--openai_rev/quora/graphql/StaleChatUpdateMutation.graphql7
-rw-r--r--openai_rev/quora/graphql/SubscriptionsMutation.graphql9
-rw-r--r--openai_rev/quora/graphql/SummarizePlainPostQuery.graphql3
-rw-r--r--openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql3
-rw-r--r--openai_rev/quora/graphql/SummarizeSharePostQuery.graphql3
-rw-r--r--openai_rev/quora/graphql/UserSnippetFragment.graphql14
-rw-r--r--openai_rev/quora/graphql/ViewerInfoQuery.graphql21
-rw-r--r--openai_rev/quora/graphql/ViewerStateFragment.graphql30
-rw-r--r--openai_rev/quora/graphql/ViewerStateUpdatedSubscription.graphql43
-rw-r--r--openai_rev/quora/graphql/__init__.py0
-rw-r--r--openai_rev/quora/mail.py80
-rw-r--r--openai_rev/you/README.md37
-rw-r--r--openai_rev/you/__init__.py108
48 files changed, 2677 insertions, 0 deletions
diff --git a/openai_rev/__init__.py b/openai_rev/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/openai_rev/__init__.py
diff --git a/openai_rev/forefront/README.md b/openai_rev/forefront/README.md
new file mode 100644
index 00000000..94089faa
--- /dev/null
+++ b/openai_rev/forefront/README.md
@@ -0,0 +1,15 @@
+### Example: `forefront` (use like openai pypi package) <a name="example-forefront"></a>
+
+```python
+
+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='')
+``` \ No newline at end of file
diff --git a/openai_rev/forefront/__init__.py b/openai_rev/forefront/__init__.py
new file mode 100644
index 00000000..bef10e9e
--- /dev/null
+++ b/openai_rev/forefront/__init__.py
@@ -0,0 +1,180 @@
+from json import loads
+from re import match
+from time import time, sleep
+from uuid import uuid4
+
+from altair.vegalite.v3 import Generator
+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=None, logging=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():
+ 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:
+ final_response = None
+ for response in StreamingCompletion.create(
+ token=token,
+ chat_id=chat_id,
+ prompt=prompt,
+ action_type=action_type,
+ default_persona=default_persona,
+ model=model,
+ ):
+ final_response = response
+
+ return final_response
diff --git a/openai_rev/forefront/mail.py b/openai_rev/forefront/mail.py
new file mode 100644
index 00000000..2c00051c
--- /dev/null
+++ b/openai_rev/forefront/mail.py
@@ -0,0 +1,52 @@
+from random import choices
+from string import ascii_letters
+
+from requests import Session
+
+
+class Mail:
+ def __init__(self, proxies: dict = None) -> None:
+ self.client = Session()
+ self.client.proxies = proxies
+ self.client.headers = {
+ "host": "api.mail.tm",
+ "connection": "keep-alive",
+ "sec-ch-ua": "\"Google Chrome\";v=\"111\", \"Not(A:Brand\";v=\"8\", \"Chromium\";v=\"111\"",
+ "accept": "application/json, text/plain, */*",
+ "content-type": "application/json",
+ "sec-ch-ua-mobile": "?0",
+ "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36",
+ "sec-ch-ua-platform": "\"macOS\"",
+ "origin": "https://mail.tm",
+ "sec-fetch-site": "same-site",
+ "sec-fetch-mode": "cors",
+ "sec-fetch-dest": "empty",
+ "referer": "https://mail.tm/",
+ "accept-encoding": "gzip, deflate, br",
+ "accept-language": "en-GB,en-US;q=0.9,en;q=0.8",
+ }
+
+ def get_mail(self) -> str:
+ token = ''.join(choices(ascii_letters, k=14)).lower()
+ init = self.client.post(
+ "https://api.mail.tm/accounts", json={"address": f"{token}@bugfoo.com", "password": token}
+ )
+
+ if init.status_code == 201:
+ resp = self.client.post("https://api.mail.tm/token", json={**init.json(), "password": token})
+
+ self.client.headers['authorization'] = 'Bearer ' + resp.json()['token']
+
+ return f"{token}@bugfoo.com"
+
+ else:
+ raise Exception("Failed to create email")
+
+ def fetch_inbox(self):
+ return self.client.get(f"https://api.mail.tm/messages").json()["hydra:member"]
+
+ def get_message(self, message_id: str):
+ return self.client.get(f"https://api.mail.tm/messages/{message_id}").json()
+
+ def get_message_content(self, message_id: str):
+ return self.get_message(message_id)["text"]
diff --git a/openai_rev/forefront/models.py b/openai_rev/forefront/models.py
new file mode 100644
index 00000000..23e90903
--- /dev/null
+++ b/openai_rev/forefront/models.py
@@ -0,0 +1,26 @@
+from typing import Any, List
+
+from pydantic import BaseModel
+
+
+class Choice(BaseModel):
+ text: str
+ index: int
+ logprobs: Any
+ finish_reason: str
+
+
+class Usage(BaseModel):
+ prompt_tokens: int
+ completion_tokens: int
+ total_tokens: int
+
+
+class ForeFrontResponse(BaseModel):
+ id: str
+ object: str
+ created: int
+ model: str
+ choices: List[Choice]
+ usage: Usage
+ text: str
diff --git a/openai_rev/openai_rev.py b/openai_rev/openai_rev.py
new file mode 100644
index 00000000..6e1341c7
--- /dev/null
+++ b/openai_rev/openai_rev.py
@@ -0,0 +1,35 @@
+from enum import Enum
+
+import quora
+import you
+
+
+class Provider(Enum):
+ You = 'you'
+ Poe = 'poe'
+
+
+class Completion:
+ @staticmethod
+ def create(provider: Provider, prompt: str, **kwargs):
+ if provider == Provider.Poe:
+ return Completion.__poe_service(prompt, **kwargs)
+ elif provider == Provider.You:
+ return Completion.__you_service(prompt, **kwargs)
+
+ @classmethod
+ def __you_service(cls, prompt: str, **kwargs) -> str:
+ return you.Completion.create(prompt).text
+
+ @classmethod
+ def __poe_service(cls, prompt: str, **kwargs) -> str:
+ return quora.Completion.create(prompt=prompt).text
+
+
+# usage You
+response = Completion.create(Provider.You, prompt='Write a poem on Lionel Messi')
+print(response)
+
+# usage Poe
+response = Completion.create(Provider.Poe, prompt='Write a poem on Lionel Messi', token='GKzCahZYGKhp76LfE197xw==')
+print(response)
diff --git a/openai_rev/phind/README.md b/openai_rev/phind/README.md
new file mode 100644
index 00000000..85288c06
--- /dev/null
+++ b/openai_rev/phind/README.md
@@ -0,0 +1,37 @@
+### Example: `phind` (use like openai pypi package) <a name="example-phind"></a>
+
+```python
+
+from openai_rev import phind
+
+# set cf_clearance cookie (needed again)
+phind.cf_clearance = 'xx.xx-1682166681-0-160'
+phind.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' # same as the one from browser you got cf_clearance from
+
+prompt = 'who won the quatar world cup'
+
+# help needed: not getting newlines from the stream, please submit a PR if you know how to fix this
+# stream completion
+for result in phind.StreamingCompletion.create(
+ model='gpt-4',
+ prompt=prompt,
+ results=phind.Search.create(prompt, actualSearch=True),
+ # create search (set actualSearch to False to disable internet)
+ creative=False,
+ detailed=False,
+ codeContext=''): # up to 3000 chars of code
+
+ print(result.completion.choices[0].text, end='', flush=True)
+
+# normal completion
+result = phind.Completion.create(
+ model='gpt-4',
+ prompt=prompt,
+ results=phind.Search.create(prompt, actualSearch=True),
+ # create search (set actualSearch to False to disable internet)
+ creative=False,
+ detailed=False,
+ codeContext='') # up to 3000 chars of code
+
+print(result.completion.choices[0].text)
+```
diff --git a/openai_rev/quora/README.md b/openai_rev/quora/README.md
new file mode 100644
index 00000000..dc2bb32d
--- /dev/null
+++ b/openai_rev/quora/README.md
@@ -0,0 +1,68 @@
+#### warning !!!
+poe.com added security and can detect if you are making automated requests. You may get your account banned if you are using this api.
+The normal non-driver api is also currently not very stable
+
+
+### Example: `quora (poe)` (use like openai pypi package) - GPT-4 <a name="example-poe"></a>
+
+```python
+# quora model names: (use left key as argument)
+models = {
+ 'sage' : 'capybara',
+ 'gpt-4' : 'beaver',
+ 'claude-v1.2' : 'a2_2',
+ 'claude-instant-v1.0' : 'a2',
+ 'gpt-3.5-turbo' : 'chinchilla'
+}
+```
+
+#### !! new: bot creation
+
+```python
+# import quora (poe) package
+from openai_rev import quora
+
+# create account
+# make sure to set enable_bot_creation to True
+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 ...'
+)
+
+print(model.name) # gptx....
+
+# streaming response
+for response in quora.StreamingCompletion.create(
+ custom_model=model.name,
+ prompt='hello world',
+ token=token):
+ print(response.text)
+```
+
+#### Normal Response:
+```python
+import quora
+
+response = quora.Completion.create(model = 'gpt-4',
+ prompt = 'hello world',
+ token = 'token')
+
+print(response.text)
+```
+
+#### Update Use This For Poe
+```python
+from quora import Poe
+
+# available models: ['Sage', 'GPT-4', 'Claude+', 'Claude-instant', 'ChatGPT', 'Dragonfly', 'NeevaAI']
+
+poe = Poe(model='ChatGPT', driver='firefox', cookie_path='cookie.json', driver_path='path_of_driver')
+poe.chat('who won the football world cup most?')
+
+# new bot creation
+poe.create_bot('new_bot_name', prompt='You are new test bot', base_model='gpt-3.5-turbo')
+
+```
diff --git a/openai_rev/quora/__init__.py b/openai_rev/quora/__init__.py
new file mode 100644
index 00000000..f5d9e96e
--- /dev/null
+++ b/openai_rev/quora/__init__.py
@@ -0,0 +1,485 @@
+import json
+from datetime import datetime
+from hashlib import md5
+from json import dumps
+from pathlib import Path
+from random import choice, choices, randint
+from re import search, findall
+from string import ascii_letters, digits
+from typing import Optional, Union, List, Any, Generator
+from urllib.parse import unquote
+
+import selenium.webdriver.support.expected_conditions as EC
+from fake_useragent import UserAgent
+from pydantic import BaseModel
+from pypasser import reCaptchaV3
+from requests import Session
+from selenium.webdriver import Firefox, Chrome, FirefoxOptions, ChromeOptions
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.wait import WebDriverWait
+from tls_client import Session as TLS
+
+from .api import Client as PoeClient
+from .mail import Emailnator
+
+SELENIUM_WEB_DRIVER_ERROR_MSG = b'''The error message you are receiving is due to the `geckodriver` executable not
+being found in your system\'s PATH. To resolve this issue, you need to download the geckodriver and add its location
+to your system\'s PATH.\n\nHere are the steps to resolve the issue:\n\n1. Download the geckodriver for your platform
+(Windows, macOS, or Linux) from the following link: https://github.com/mozilla/geckodriver/releases\n\n2. Extract the
+downloaded archive and locate the geckodriver executable.\n\n3. Add the geckodriver executable to your system\'s
+PATH.\n\nFor macOS and Linux:\n\n- Open a terminal window.\n- Move the geckodriver executable to a directory that is
+already in your PATH, or create a new directory and add it to your PATH:\n\n```bash\n# Example: Move geckodriver to
+/usr/local/bin\nmv /path/to/your/geckodriver /usr/local/bin\n```\n\n- If you created a new directory, add it to your
+PATH:\n\n```bash\n# Example: Add a new directory to PATH\nexport PATH=$PATH:/path/to/your/directory\n```\n\nFor
+Windows:\n\n- Right-click on "My Computer" or "This PC" and select "Properties".\n- Click on "Advanced system
+settings".\n- Click on the "Environment Variables" button.\n- In the "System variables" section, find the "Path"
+variable, select it, and click "Edit".\n- Click "New" and add the path to the directory containing the geckodriver
+executable.\n\nAfter adding the geckodriver to your PATH, restart your terminal or command prompt and try running
+your script again. The error should be resolved.'''
+
+# from twocaptcha import TwoCaptcha
+# solver = TwoCaptcha('72747bf24a9d89b4dcc1b24875efd358')
+
+MODELS = {
+ 'Sage': 'capybara',
+ 'GPT-4': 'beaver',
+ 'Claude+': 'a2_2',
+ 'Claude-instant': 'a2',
+ 'ChatGPT': 'chinchilla',
+ 'Dragonfly': 'nutria',
+ 'NeevaAI': 'hutia',
+}
+
+
+def extract_formkey(html):
+ script_regex = r'<script>if\(.+\)throw new Error;(.+)</script>'
+ script_text = search(script_regex, html).group(1)
+ key_regex = r'var .="([0-9a-f]+)",'
+ key_text = search(key_regex, script_text).group(1)
+ cipher_regex = r'.\[(\d+)\]=.\[(\d+)\]'
+ cipher_pairs = findall(cipher_regex, script_text)
+
+ formkey_list = [''] * len(cipher_pairs)
+ for pair in cipher_pairs:
+ formkey_index, key_index = map(int, pair)
+ formkey_list[formkey_index] = key_text[key_index]
+ formkey = ''.join(formkey_list)
+
+ return formkey
+
+
+class Choice(BaseModel):
+ text: str
+ index: int
+ logprobs: Any
+ finish_reason: str
+
+
+class Usage(BaseModel):
+ prompt_tokens: int
+ completion_tokens: int
+ total_tokens: int
+
+
+class PoeResponse(BaseModel):
+ id: int
+ object: str
+ created: int
+ model: str
+ choices: List[Choice]
+ usage: Usage
+ text: str
+
+
+class ModelResponse:
+ def __init__(self, json_response: dict) -> None:
+ self.id = json_response['data']['poeBotCreate']['bot']['id']
+ self.name = json_response['data']['poeBotCreate']['bot']['displayName']
+ self.limit = json_response['data']['poeBotCreate']['bot']['messageLimit']['dailyLimit']
+ self.deleted = json_response['data']['poeBotCreate']['bot']['deletionState']
+
+
+class Model:
+ @staticmethod
+ def create(
+ token: str,
+ model: str = 'gpt-3.5-turbo', # claude-instant
+ system_prompt: str = 'You are ChatGPT a large language model developed by Openai. Answer as consisely as possible',
+ description: str = 'gpt-3.5 language model from openai, skidded by poe.com',
+ handle: str = None,
+ ) -> ModelResponse:
+ models = {
+ 'gpt-3.5-turbo': 'chinchilla',
+ 'claude-instant-v1.0': 'a2',
+ 'gpt-4': 'beaver',
+ }
+
+ if not handle:
+ handle = f'gptx{randint(1111111, 9999999)}'
+
+ client = Session()
+ client.cookies['p-b'] = token
+
+ formkey = extract_formkey(client.get('https://poe.com').text)
+ settings = client.get('https://poe.com/api/settings').json()
+
+ client.headers = {
+ 'host': 'poe.com',
+ 'origin': 'https://poe.com',
+ 'referer': 'https://poe.com/',
+ 'poe-formkey': formkey,
+ 'poe-tchannel': settings['tchannelData']['channel'],
+ 'user-agent': UserAgent().random,
+ 'connection': 'keep-alive',
+ '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"',
+ 'content-type': 'application/json',
+ 'sec-fetch-site': 'same-origin',
+ 'sec-fetch-mode': 'cors',
+ 'sec-fetch-dest': 'empty',
+ 'accept': '*/*',
+ 'accept-encoding': 'gzip, deflate, br',
+ 'accept-language': 'en-GB,en-US;q=0.9,en;q=0.8',
+ }
+
+ payload = dumps(
+ separators=(',', ':'),
+ obj={
+ 'queryName': 'CreateBotMain_poeBotCreate_Mutation',
+ 'variables': {
+ 'model': models[model],
+ 'handle': handle,
+ 'prompt': system_prompt,
+ 'isPromptPublic': True,
+ 'introduction': '',
+ 'description': description,
+ 'profilePictureUrl': 'https://qph.fs.quoracdn.net/main-qimg-24e0b480dcd946e1cc6728802c5128b6',
+ 'apiUrl': None,
+ 'apiKey': ''.join(choices(ascii_letters + digits, k=32)),
+ 'isApiBot': False,
+ 'hasLinkification': False,
+ 'hasMarkdownRendering': False,
+ 'hasSuggestedReplies': False,
+ 'isPrivateBot': False,
+ },
+ 'query': 'mutation CreateBotMain_poeBotCreate_Mutation(\n $model: String!\n $handle: String!\n $prompt: String!\n $isPromptPublic: Boolean!\n $introduction: String!\n $description: String!\n $profilePictureUrl: String\n $apiUrl: String\n $apiKey: String\n $isApiBot: Boolean\n $hasLinkification: Boolean\n $hasMarkdownRendering: Boolean\n $hasSuggestedReplies: Boolean\n $isPrivateBot: Boolean\n) {\n poeBotCreate(model: $model, handle: $handle, promptPlaintext: $prompt, isPromptPublic: $isPromptPublic, introduction: $introduction, description: $description, profilePicture: $profilePictureUrl, apiUrl: $apiUrl, apiKey: $apiKey, isApiBot: $isApiBot, hasLinkification: $hasLinkification, hasMarkdownRendering: $hasMarkdownRendering, hasSuggestedReplies: $hasSuggestedReplies, isPrivateBot: $isPrivateBot) {\n status\n bot {\n id\n ...BotHeader_bot\n }\n }\n}\n\nfragment BotHeader_bot on Bot {\n displayName\n messageLimit {\n dailyLimit\n }\n ...BotImage_bot\n ...BotLink_bot\n ...IdAnnotation_node\n ...botHelpers_useViewerCanAccessPrivateBot\n ...botHelpers_useDeletion_bot\n}\n\nfragment BotImage_bot on Bot {\n displayName\n ...botHelpers_useDeletion_bot\n ...BotImage_useProfileImage_bot\n}\n\nfragment BotImage_useProfileImage_bot on Bot {\n image {\n __typename\n ... on LocalBotImage {\n localName\n }\n ... on UrlBotImage {\n url\n }\n }\n ...botHelpers_useDeletion_bot\n}\n\nfragment BotLink_bot on Bot {\n displayName\n}\n\nfragment IdAnnotation_node on Node {\n __isNode: __typename\n id\n}\n\nfragment botHelpers_useDeletion_bot on Bot {\n deletionState\n}\n\nfragment botHelpers_useViewerCanAccessPrivateBot on Bot {\n isPrivateBot\n viewerIsCreator\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 'success' not in response.text:
+ raise Exception(
+ '''
+ Bot creation Failed
+ !! Important !!
+ Bot creation was not enabled on this account
+ please use: quora.Account.create with enable_bot_creation set to True
+ '''
+ )
+
+ return ModelResponse(response.json())
+
+
+class Account:
+ @staticmethod
+ def create(
+ proxy: Optional[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
+
+ mail_client = Emailnator()
+ mail_address = mail_client.get_mail()
+
+ if logging:
+ print('email', mail_address)
+
+ client.headers = {
+ 'authority': 'poe.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',
+ 'content-type': 'application/json',
+ 'origin': 'https://poe.com',
+ 'poe-tag-id': 'null',
+ 'referer': 'https://poe.com/login',
+ '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': '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',
+ 'poe-formkey': extract_formkey(client.get('https://poe.com/login').text),
+ '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,
+ },
+ '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()
+
+ print(dumps(client.headers, indent=4))
+
+ response = client.post('https://poe.com/api/gql_POST', data=payload)
+
+ if 'automated_request_detected' in response.text:
+ print('please try using a proxy / wait for fix')
+
+ if 'Bad Request' in response.text:
+ if logging:
+ print('bad request, retrying...', response.json())
+ quit()
+
+ 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)
+
+ 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',
+ },
+ )
+
+ 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())
+
+ def get(self):
+ cookies = open(Path(__file__).resolve().parent / 'cookies.txt', 'r').read().splitlines()
+ return choice(cookies)
+
+
+class StreamingCompletion:
+ @staticmethod
+ def create(
+ model: str = 'gpt-4',
+ custom_model: bool = None,
+ prompt: str = 'hello world',
+ token: str = '',
+ ) -> Generator[PoeResponse, None, None]:
+ _model = MODELS[model] if not custom_model else custom_model
+
+ client = PoeClient(token)
+
+ for chunk in client.send_message(_model, prompt):
+ yield PoeResponse(
+ **{
+ 'id': chunk['messageId'],
+ 'object': 'text_completion',
+ 'created': chunk['creationTime'],
+ 'model': _model,
+ 'text': chunk['text_new'],
+ 'choices': [
+ {
+ 'text': chunk['text_new'],
+ 'index': 0,
+ 'logprobs': None,
+ 'finish_reason': 'stop',
+ }
+ ],
+ 'usage': {
+ 'prompt_tokens': len(prompt),
+ 'completion_tokens': len(chunk['text_new']),
+ 'total_tokens': len(prompt) + len(chunk['text_new']),
+ },
+ }
+ )
+
+
+class Completion:
+ @staticmethod
+ def create(
+ model: str = 'gpt-4',
+ custom_model: str = None,
+ prompt: str = 'hello world',
+ token: str = '',
+ ) -> PoeResponse:
+ models = {
+ 'sage': 'capybara',
+ 'gpt-4': 'beaver',
+ 'claude-v1.2': 'a2_2',
+ 'claude-instant-v1.0': 'a2',
+ 'gpt-3.5-turbo': 'chinchilla',
+ }
+
+ _model = models[model] if not custom_model else custom_model
+
+ client = PoeClient(token)
+
+ chunk = None
+ for response in client.send_message(_model, prompt):
+ chunk = response
+
+ return PoeResponse(
+ **{
+ 'id': chunk['messageId'],
+ 'object': 'text_completion',
+ 'created': chunk['creationTime'],
+ 'model': _model,
+ 'text': chunk['text_new'],
+ 'choices': [
+ {
+ 'text': chunk['text'],
+ 'index': 0,
+ 'logprobs': None,
+ 'finish_reason': 'stop',
+ }
+ ],
+ 'usage': {
+ 'prompt_tokens': len(prompt),
+ 'completion_tokens': len(chunk['text']),
+ 'total_tokens': len(prompt) + len(chunk['text']),
+ },
+ }
+ )
+
+
+class Poe:
+ def __init__(
+ self,
+ model: str = 'ChatGPT',
+ driver: str = 'firefox',
+ download_driver: bool = False,
+ driver_path: Optional[str] = None,
+ cookie_path: str = './quora/cookie.json',
+ ):
+ # validating the model
+ if model and model not in MODELS:
+ raise RuntimeError('Sorry, the model you provided does not exist. Please check and try again.')
+ self.model = MODELS[model]
+ self.cookie_path = cookie_path
+ self.cookie = self.__load_cookie(driver, download_driver, driver_path=driver_path)
+ self.client = PoeClient(self.cookie)
+
+ def __load_cookie(self, driver: str, download_driver: bool, driver_path: Optional[str] = None) -> str:
+ if (cookie_file := Path(self.cookie_path)).exists():
+ with cookie_file.open() as fp:
+ cookie = json.load(fp)
+ if datetime.fromtimestamp(cookie['expiry']) < datetime.now():
+ cookie = self.__register_and_get_cookie(driver, driver_path=driver_path)
+ else:
+ print('Loading the cookie from file')
+ else:
+ cookie = self.__register_and_get_cookie(driver, driver_path=driver_path)
+
+ return unquote(cookie['value'])
+
+ def __register_and_get_cookie(self, driver: str, driver_path: Optional[str] = None) -> dict:
+ mail_client = Emailnator()
+ mail_address = mail_client.get_mail()
+
+ driver = self.__resolve_driver(driver, driver_path=driver_path)
+ driver.get("https://www.poe.com")
+
+ # clicking use email button
+ driver.find_element(By.XPATH, '//button[contains(text(), "Use email")]').click()
+
+ email = WebDriverWait(driver, 30).until(EC.presence_of_element_located((By.XPATH, '//input[@type="email"]')))
+ email.send_keys(mail_address)
+ driver.find_element(By.XPATH, '//button[text()="Go"]').click()
+
+ code = findall(r';">(\d{6,7})</div>', mail_client.get_message())[0]
+ print(code)
+
+ verification_code = WebDriverWait(driver, 30).until(
+ EC.presence_of_element_located((By.XPATH, '//input[@placeholder="Code"]'))
+ )
+ verification_code.send_keys(code)
+ verify_button = EC.presence_of_element_located((By.XPATH, '//button[text()="Verify"]'))
+ login_button = EC.presence_of_element_located((By.XPATH, '//button[text()="Log In"]'))
+
+ WebDriverWait(driver, 30).until(EC.any_of(verify_button, login_button)).click()
+
+ cookie = driver.get_cookie('p-b')
+
+ with open(self.cookie_path, 'w') as fw:
+ json.dump(cookie, fw)
+
+ driver.close()
+ return cookie
+
+ @classmethod
+ def __resolve_driver(cls, driver: str, driver_path: Optional[str] = None) -> Union[Firefox, Chrome]:
+ options = FirefoxOptions() if driver == 'firefox' else ChromeOptions()
+ options.add_argument('-headless')
+
+ if driver_path:
+ options.binary_location = driver_path
+ try:
+ return Firefox(options=options) if driver == 'firefox' else Chrome(options=options)
+ except Exception:
+ raise Exception(SELENIUM_WEB_DRIVER_ERROR_MSG)
+
+ def chat(self, message: str, model: Optional[str] = None) -> str:
+ if model and model not in MODELS:
+ raise RuntimeError('Sorry, the model you provided does not exist. Please check and try again.')
+ model = MODELS[model] if model else self.model
+ response = None
+ for chunk in self.client.send_message(model, message):
+ response = chunk['text']
+ return response
+
+ def create_bot(
+ self,
+ name: str,
+ /,
+ prompt: str = '',
+ base_model: str = 'ChatGPT',
+ description: str = '',
+ ) -> None:
+ if base_model not in MODELS:
+ raise RuntimeError('Sorry, the base_model you provided does not exist. Please check and try again.')
+
+ response = self.client.create_bot(
+ handle=name,
+ prompt=prompt,
+ base_model=MODELS[base_model],
+ description=description,
+ )
+ print(f'Successfully created bot with name: {response["bot"]["displayName"]}')
+
+ def list_bots(self) -> list:
+ return list(self.client.bot_names.values())
diff --git a/openai_rev/quora/api.py b/openai_rev/quora/api.py
new file mode 100644
index 00000000..42814f2c
--- /dev/null
+++ b/openai_rev/quora/api.py
@@ -0,0 +1,545 @@
+# This file was taken from the repository poe-api https://github.com/ading2210/poe-api and is unmodified
+# This file is licensed under the GNU GPL v3 and written by @ading2210
+
+# license:
+# ading2210/poe-api: a reverse engineered Python API wrapepr for Quora's Poe
+# Copyright (C) 2023 ading2210
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import hashlib
+import json
+import logging
+import queue
+import random
+import re
+import threading
+import time
+import traceback
+from pathlib import Path
+from urllib.parse import urlparse
+
+import requests
+import requests.adapters
+import websocket
+
+parent_path = Path(__file__).resolve().parent
+queries_path = parent_path / "graphql"
+queries = {}
+
+logging.basicConfig()
+logger = logging.getLogger()
+
+user_agent = "Mozilla/5.0 (X11; Linux x86_64; rv:102.0) Gecko/20100101 Firefox/102.0"
+
+
+def load_queries():
+ for path in queries_path.iterdir():
+ if path.suffix != ".graphql":
+ continue
+ with open(path) as f:
+ queries[path.stem] = f.read()
+
+
+def generate_payload(query_name, variables):
+ return {"query": queries[query_name], "variables": variables}
+
+
+def request_with_retries(method, *args, **kwargs):
+ attempts = kwargs.get("attempts") or 10
+ url = args[0]
+ for i in range(attempts):
+ r = method(*args, **kwargs)
+ if r.status_code == 200:
+ return r
+ logger.warn(
+ f"Server returned a status code of {r.status_code} while downloading {url}. Retrying ({i + 1}/{attempts})..."
+ )
+
+ raise RuntimeError(f"Failed to download {url} too many times.")
+
+
+class Client:
+ gql_url = "https://poe.com/api/gql_POST"
+ gql_recv_url = "https://poe.com/api/receive_POST"
+ home_url = "https://poe.com"
+ settings_url = "https://poe.com/api/settings"
+
+ def __init__(self, token, proxy=None):
+ self.proxy = proxy
+ self.session = requests.Session()
+ self.adapter = requests.adapters.HTTPAdapter(pool_connections=100, pool_maxsize=100)
+ self.session.mount("http://", self.adapter)
+ self.session.mount("https://", self.adapter)
+
+ if proxy:
+ self.session.proxies = {"http": self.proxy, "https": self.proxy}
+ logger.info(f"Proxy enabled: {self.proxy}")
+
+ self.active_messages = {}
+ self.message_queues = {}
+
+ self.session.cookies.set("p-b", token, domain="poe.com")
+ self.headers = {
+ "User-Agent": user_agent,
+ "Referrer": "https://poe.com/",
+ "Origin": "https://poe.com",
+ }
+ self.session.headers.update(self.headers)
+
+ self.setup_connection()
+ self.connect_ws()
+
+ def setup_connection(self):
+ self.ws_domain = f"tch{random.randint(1, 1e6)}"
+ self.next_data = self.get_next_data(overwrite_vars=True)
+ self.channel = self.get_channel_data()
+ self.bots = self.get_bots(download_next_data=False)
+ self.bot_names = self.get_bot_names()
+
+ self.gql_headers = {
+ "poe-formkey": self.formkey,
+ "poe-tchannel": self.channel["channel"],
+ }
+ self.gql_headers = {**self.gql_headers, **self.headers}
+ self.subscribe()
+
+ def extract_formkey(self, html):
+ script_regex = r"<script>if\(.+\)throw new Error;(.+)</script>"
+ script_text = re.search(script_regex, html).group(1)
+ key_regex = r'var .="([0-9a-f]+)",'
+ key_text = re.search(key_regex, script_text).group(1)
+ cipher_regex = r".\[(\d+)\]=.\[(\d+)\]"
+ cipher_pairs = re.findall(cipher_regex, script_text)
+
+ formkey_list = [""] * len(cipher_pairs)
+ for pair in cipher_pairs:
+ formkey_index, key_index = map(int, pair)
+ formkey_list[formkey_index] = key_text[key_index]
+ formkey = "".join(formkey_list)
+
+ return formkey
+
+ def get_next_data(self, overwrite_vars=False):
+ logger.info("Downloading next_data...")
+
+ r = request_with_retries(self.session.get, self.home_url)
+ json_regex = r'<script id="__NEXT_DATA__" type="application\/json">(.+?)</script>'
+ json_text = re.search(json_regex, r.text).group(1)
+ next_data = json.loads(json_text)
+
+ if overwrite_vars:
+ self.formkey = self.extract_formkey(r.text)
+ self.viewer = next_data["props"]["pageProps"]["payload"]["viewer"]
+ self.next_data = next_data
+
+ return next_data
+
+ def get_bot(self, display_name):
+ url = f'https://poe.com/_next/data/{self.next_data["buildId"]}/{display_name}.json'
+
+ r = request_with_retries(self.session.get, url)
+
+ chat_data = r.json()["pageProps"]["payload"]["chatOfBotDisplayName"]
+ return chat_data
+
+ def get_bots(self, download_next_data=True):
+ logger.info("Downloading all bots...")
+ if download_next_data:
+ next_data = self.get_next_data(overwrite_vars=True)
+ else:
+ next_data = self.next_data
+
+ if not "availableBots" in self.viewer:
+ raise RuntimeError("Invalid token or no bots are available.")
+ bot_list = self.viewer["availableBots"]
+
+ threads = []
+ bots = {}
+
+ def get_bot_thread(bot):
+ chat_data = self.get_bot(bot["displayName"])
+ bots[chat_data["defaultBotObject"]["nickname"]] = chat_data
+
+ for bot in bot_list:
+ thread = threading.Thread(target=get_bot_thread, args=(bot,), daemon=True)
+ threads.append(thread)
+
+ for thread in threads:
+ thread.start()
+ for thread in threads:
+ thread.join()
+
+ self.bots = bots
+ self.bot_names = self.get_bot_names()
+ return bots
+
+ def get_bot_names(self):
+ bot_names = {}
+ for bot_nickname in self.bots:
+ bot_obj = self.bots[bot_nickname]["defaultBotObject"]
+ bot_names[bot_nickname] = bot_obj["displayName"]
+ return bot_names
+
+ def get_remaining_messages(self, chatbot):
+ chat_data = self.get_bot(self.bot_names[chatbot])
+ return chat_data["defaultBotObject"]["messageLimit"]["numMessagesRemaining"]
+
+ def get_channel_data(self, channel=None):
+ logger.info("Downloading channel data...")
+ r = request_with_retries(self.session.get, self.settings_url)
+ data = r.json()
+
+ return data["tchannelData"]
+
+ def get_websocket_url(self, channel=None):
+ if channel is None:
+ channel = self.channel
+ query = f'?min_seq={channel["minSeq"]}&channel={channel["channel"]}&hash={channel["channelHash"]}'
+ return f'wss://{self.ws_domain}.tch.{channel["baseHost"]}/up/{channel["boxName"]}/updates' + query
+
+ def send_query(self, query_name, variables):
+ for i in range(20):
+ json_data = generate_payload(query_name, variables)
+ payload = json.dumps(json_data, separators=(",", ":"))
+
+ base_string = payload + self.gql_headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
+
+ headers = {
+ "content-type": "application/json",
+ "poe-tag-id": hashlib.md5(base_string.encode()).hexdigest(),
+ }
+ headers = {**self.gql_headers, **headers}
+
+ r = request_with_retries(self.session.post, self.gql_url, data=payload, headers=headers)
+
+ data = r.json()
+ if data["data"] == None:
+ logger.warn(f'{query_name} returned an error: {data["errors"][0]["message"]} | Retrying ({i + 1}/20)')
+ time.sleep(2)
+ continue
+
+ return r.json()
+
+ raise RuntimeError(f"{query_name} failed too many times.")
+
+ def subscribe(self):
+ logger.info("Subscribing to mutations")
+ result = self.send_query(
+ "SubscriptionsMutation",
+ {
+ "subscriptions": [
+ {
+ "subscriptionName": "messageAdded",
+ "query": queries["MessageAddedSubscription"],
+ },
+ {
+ "subscriptionName": "viewerStateUpdated",
+ "query": queries["ViewerStateUpdatedSubscription"],
+ },
+ ]
+ },
+ )
+
+ def ws_run_thread(self):
+ kwargs = {}
+ if self.proxy:
+ proxy_parsed = urlparse(self.proxy)
+ kwargs = {
+ "proxy_type": proxy_parsed.scheme,
+ "http_proxy_host": proxy_parsed.hostname,
+ "http_proxy_port": proxy_parsed.port,
+ }
+
+ self.ws.run_forever(**kwargs)
+
+ def connect_ws(self):
+ self.ws_connected = False
+ self.ws = websocket.WebSocketApp(
+ self.get_websocket_url(),
+ header={"User-Agent": user_agent},
+ on_message=self.on_message,
+ on_open=self.on_ws_connect,
+ on_error=self.on_ws_error,
+ on_close=self.on_ws_close,
+ )
+ t = threading.Thread(target=self.ws_run_thread, daemon=True)
+ t.start()
+ while not self.ws_connected:
+ time.sleep(0.01)
+
+ def disconnect_ws(self):
+ if self.ws:
+ self.ws.close()
+ self.ws_connected = False
+
+ def on_ws_connect(self, ws):
+ self.ws_connected = True
+
+ def on_ws_close(self, ws, close_status_code, close_message):
+ self.ws_connected = False
+ logger.warn(f"Websocket closed with status {close_status_code}: {close_message}")
+
+ def on_ws_error(self, ws, error):
+ self.disconnect_ws()
+ self.connect_ws()
+
+ def on_message(self, ws, msg):
+ try:
+ data = json.loads(msg)
+
+ if not "messages" in data:
+ return
+
+ for message_str in data["messages"]:
+ message_data = json.loads(message_str)
+ if message_data["message_type"] != "subscriptionUpdate":
+ continue
+ message = message_data["payload"]["data"]["messageAdded"]
+
+ copied_dict = self.active_messages.copy()
+ for key, value in copied_dict.items():
+ # add the message to the appropriate queue
+ if value == message["messageId"] and key in self.message_queues:
+ self.message_queues[key].put(message)
+ return
+
+ # indicate that the response id is tied to the human message id
+ elif key != "pending" and value == None and message["state"] != "complete":
+ self.active_messages[key] = message["messageId"]
+ self.message_queues[key].put(message)
+ return
+
+ except Exception:
+ logger.error(traceback.format_exc())
+ self.disconnect_ws()
+ self.connect_ws()
+
+ def send_message(self, chatbot, message, with_chat_break=False, timeout=20):
+ # if there is another active message, wait until it has finished sending
+ while None in self.active_messages.values():
+ time.sleep(0.01)
+
+ # None indicates that a message is still in progress
+ self.active_messages["pending"] = None
+
+ logger.info(f"Sending message to {chatbot}: {message}")
+
+ # reconnect websocket
+ if not self.ws_connected:
+ self.disconnect_ws()
+ self.setup_connection()
+ self.connect_ws()
+
+ message_data = self.send_query(
+ "SendMessageMutation",
+ {
+ "bot": chatbot,
+ "query": message,
+ "chatId": self.bots[chatbot]["chatId"],
+ "source": None,
+ "withChatBreak": with_chat_break,
+ },
+ )
+ del self.active_messages["pending"]
+
+ if not message_data["data"]["messageEdgeCreate"]["message"]:
+ raise RuntimeError(f"Daily limit reached for {chatbot}.")
+ try:
+ human_message = message_data["data"]["messageEdgeCreate"]["message"]
+ human_message_id = human_message["node"]["messageId"]
+ except TypeError:
+ raise RuntimeError(f"An unknown error occurred. Raw response data: {message_data}")
+
+ # indicate that the current message is waiting for a response
+ self.active_messages[human_message_id] = None
+ self.message_queues[human_message_id] = queue.Queue()
+
+ last_text = ""
+ message_id = None
+ while True:
+ try:
+ message = self.message_queues[human_message_id].get(timeout=timeout)
+ except queue.Empty:
+ del self.active_messages[human_message_id]
+ del self.message_queues[human_message_id]
+ raise RuntimeError("Response timed out.")
+
+ # only break when the message is marked as complete
+ if message["state"] == "complete":
+ if last_text and message["messageId"] == message_id:
+ break
+ else:
+ continue
+
+ # update info about response
+ message["text_new"] = message["text"][len(last_text) :]
+ last_text = message["text"]
+ message_id = message["messageId"]
+
+ yield message
+
+ del self.active_messages[human_message_id]
+ del self.message_queues[human_message_id]
+
+ def send_chat_break(self, chatbot):
+ logger.info(f"Sending chat break to {chatbot}")
+ result = self.send_query("AddMessageBreakMutation", {"chatId": self.bots[chatbot]["chatId"]})
+ return result["data"]["messageBreakCreate"]["message"]
+
+ def get_message_history(self, chatbot, count=25, cursor=None):
+ logger.info(f"Downloading {count} messages from {chatbot}")
+
+ messages = []
+ if cursor == None:
+ chat_data = self.get_bot(self.bot_names[chatbot])
+ if not chat_data["messagesConnection"]["edges"]:
+ return []
+ messages = chat_data["messagesConnection"]["edges"][:count]
+ cursor = chat_data["messagesConnection"]["pageInfo"]["startCursor"]
+ count -= len(messages)
+
+ cursor = str(cursor)
+ if count > 50:
+ messages = self.get_message_history(chatbot, count=50, cursor=cursor) + messages
+ while count > 0:
+ count -= 50
+ new_cursor = messages[0]["cursor"]
+ new_messages = self.get_message_history(chatbot, min(50, count), cursor=new_cursor)
+ messages = new_messages + messages
+ return messages
+ elif count <= 0:
+ return messages
+
+ result = self.send_query(
+ "ChatListPaginationQuery",
+ {"count": count, "cursor": cursor, "id": self.bots[chatbot]["id"]},
+ )
+ query_messages = result["data"]["node"]["messagesConnection"]["edges"]
+ messages = query_messages + messages
+ return messages
+
+ def delete_message(self, message_ids):
+ logger.info(f"Deleting messages: {message_ids}")
+ if not type(message_ids) is list:
+ message_ids = [int(message_ids)]
+
+ result = self.send_query("DeleteMessageMutation", {"messageIds": message_ids})
+
+ def purge_conversation(self, chatbot, count=-1):
+ logger.info(f"Purging messages from {chatbot}")
+ last_messages = self.get_message_history(chatbot, count=50)[::-1]
+ while last_messages:
+ message_ids = []
+ for message in last_messages:
+ if count == 0:
+ break
+ count -= 1
+ message_ids.append(message["node"]["messageId"])
+
+ self.delete_message(message_ids)
+
+ if count == 0:
+ return
+ last_messages = self.get_message_history(chatbot, count=50)[::-1]
+ logger.info(f"No more messages left to delete.")
+
+ def create_bot(
+ self,
+ handle,
+ prompt="",
+ base_model="chinchilla",
+ description="",
+ intro_message="",
+ api_key=None,
+ api_bot=False,
+ api_url=None,
+ prompt_public=True,
+ pfp_url=None,
+ linkification=False,
+ markdown_rendering=True,
+ suggested_replies=False,
+ private=False,
+ ):
+ result = self.send_query(
+ "PoeBotCreateMutation",
+ {
+ "model": base_model,
+ "handle": handle,
+ "prompt": prompt,
+ "isPromptPublic": prompt_public,
+ "introduction": intro_message,
+ "description": description,
+ "profilePictureUrl": pfp_url,
+ "apiUrl": api_url,
+ "apiKey": api_key,
+ "isApiBot": api_bot,
+ "hasLinkification": linkification,
+ "hasMarkdownRendering": markdown_rendering,
+ "hasSuggestedReplies": suggested_replies,
+ "isPrivateBot": private,
+ },
+ )
+
+ data = result["data"]["poeBotCreate"]
+ if data["status"] != "success":
+ raise RuntimeError(f"Poe returned an error while trying to create a bot: {data['status']}")
+ self.get_bots()
+ return data
+
+ def edit_bot(
+ self,
+ bot_id,
+ handle,
+ prompt="",
+ base_model="chinchilla",
+ description="",
+ intro_message="",
+ api_key=None,
+ api_url=None,
+ private=False,
+ prompt_public=True,
+ pfp_url=None,
+ linkification=False,
+ markdown_rendering=True,
+ suggested_replies=False,
+ ):
+ result = self.send_query(
+ "PoeBotEditMutation",
+ {
+ "baseBot": base_model,
+ "botId": bot_id,
+ "handle": handle,
+ "prompt": prompt,
+ "isPromptPublic": prompt_public,
+ "introduction": intro_message,
+ "description": description,
+ "profilePictureUrl": pfp_url,
+ "apiUrl": api_url,
+ "apiKey": api_key,
+ "hasLinkification": linkification,
+ "hasMarkdownRendering": markdown_rendering,
+ "hasSuggestedReplies": suggested_replies,
+ "isPrivateBot": private,
+ },
+ )
+
+ data = result["data"]["poeBotEdit"]
+ if data["status"] != "success":
+ raise RuntimeError(f"Poe returned an error while trying to edit a bot: {data['status']}")
+ self.get_bots()
+ return data
+
+
+load_queries()
diff --git a/openai_rev/quora/cookies.txt b/openai_rev/quora/cookies.txt
new file mode 100644
index 00000000..9cccf6ba
--- /dev/null
+++ b/openai_rev/quora/cookies.txt
@@ -0,0 +1,30 @@
+SmPiNXZI9hBTuf3viz74PA==
+zw7RoKQfeEehiaelYMRWeA==
+NEttgJ_rRQdO05Tppx6hFw==
+3OnmC0r9njYdNWhWszdQJg==
+8hZKR7MxwUTEHvO45TEViw==
+Eea6BqK0AmosTKzoI3AAow==
+pUEbtxobN_QUSpLIR8RGww==
+9_dUWxKkHHhpQRSvCvBk2Q==
+UV45rvGwUwi2qV9QdIbMcw==
+cVIN0pK1Wx-F7zCdUxlYqA==
+UP2wQVds17VFHh6IfCQFrA==
+18eKr0ME2Tzifdfqat38Aw==
+FNgKEpc2r-XqWe0rHBfYpg==
+juCAh6kB0sUpXHvKik2woA==
+nBvuNYRLaE4xE4HuzBPiIQ==
+oyae3iClomSrk6RJywZ4iw==
+1Z27Ul8BTdNOhncT5H6wdg==
+wfUfJIlwQwUss8l-3kDt3w==
+f6Jw_Nr0PietpNCtOCXJTw==
+6Jc3yCs7XhDRNHa4ZML09g==
+3vy44sIy-ZlTMofFiFDttw==
+p9FbMGGiK1rShKgL3YWkDg==
+pw6LI5Op84lf4HOY7fn91A==
+QemKm6aothMvqcEgeKFDlQ==
+cceZzucA-CEHR0Gt6VLYLQ==
+JRRObMp2RHVn5u4730DPvQ==
+XNt0wLTjX7Z-EsRR3TJMIQ==
+csjjirAUKtT5HT1KZUq1kg==
+8qZdCatCPQZyS7jsO4hkdQ==
+esnUxcBhvH1DmCJTeld0qw==
diff --git a/openai_rev/quora/graphql/AddHumanMessageMutation.graphql b/openai_rev/quora/graphql/AddHumanMessageMutation.graphql
new file mode 100644
index 00000000..01e6bc8c
--- /dev/null
+++ b/openai_rev/quora/graphql/AddHumanMessageMutation.graphql
@@ -0,0 +1,52 @@
+mutation AddHumanMessageMutation(
+ $chatId: BigInt!
+ $bot: String!
+ $query: String!
+ $source: MessageSource
+ $withChatBreak: Boolean! = false
+) {
+ messageCreateWithStatus(
+ chatId: $chatId
+ bot: $bot
+ query: $query
+ source: $source
+ withChatBreak: $withChatBreak
+ ) {
+ message {
+ id
+ __typename
+ messageId
+ text
+ linkifiedText
+ authorNickname
+ state
+ vote
+ voteReason
+ creationTime
+ suggestedReplies
+ chat {
+ id
+ shouldShowDisclaimer
+ }
+ }
+ messageLimit{
+ canSend
+ numMessagesRemaining
+ resetTime
+ shouldShowReminder
+ }
+ chatBreak {
+ id
+ __typename
+ messageId
+ text
+ linkifiedText
+ authorNickname
+ state
+ vote
+ voteReason
+ creationTime
+ suggestedReplies
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/AddMessageBreakMutation.graphql b/openai_rev/quora/graphql/AddMessageBreakMutation.graphql
new file mode 100644
index 00000000..b28d9903
--- /dev/null
+++ b/openai_rev/quora/graphql/AddMessageBreakMutation.graphql
@@ -0,0 +1,17 @@
+mutation AddMessageBreakMutation($chatId: BigInt!) {
+ messageBreakCreate(chatId: $chatId) {
+ message {
+ id
+ __typename
+ messageId
+ text
+ linkifiedText
+ authorNickname
+ state
+ vote
+ voteReason
+ creationTime
+ suggestedReplies
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/AutoSubscriptionMutation.graphql b/openai_rev/quora/graphql/AutoSubscriptionMutation.graphql
new file mode 100644
index 00000000..6cf7bf74
--- /dev/null
+++ b/openai_rev/quora/graphql/AutoSubscriptionMutation.graphql
@@ -0,0 +1,7 @@
+mutation AutoSubscriptionMutation($subscriptions: [AutoSubscriptionQuery!]!) {
+ autoSubscribe(subscriptions: $subscriptions) {
+ viewer {
+ id
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/BioFragment.graphql b/openai_rev/quora/graphql/BioFragment.graphql
new file mode 100644
index 00000000..c4218030
--- /dev/null
+++ b/openai_rev/quora/graphql/BioFragment.graphql
@@ -0,0 +1,8 @@
+fragment BioFragment on Viewer {
+ id
+ poeUser {
+ id
+ uid
+ bio
+ }
+}
diff --git a/openai_rev/quora/graphql/ChatAddedSubscription.graphql b/openai_rev/quora/graphql/ChatAddedSubscription.graphql
new file mode 100644
index 00000000..664b107f
--- /dev/null
+++ b/openai_rev/quora/graphql/ChatAddedSubscription.graphql
@@ -0,0 +1,5 @@
+subscription ChatAddedSubscription {
+ chatAdded {
+ ...ChatFragment
+ }
+}
diff --git a/openai_rev/quora/graphql/ChatFragment.graphql b/openai_rev/quora/graphql/ChatFragment.graphql
new file mode 100644
index 00000000..605645ff
--- /dev/null
+++ b/openai_rev/quora/graphql/ChatFragment.graphql
@@ -0,0 +1,6 @@
+fragment ChatFragment on Chat {
+ id
+ chatId
+ defaultBotNickname
+ shouldShowDisclaimer
+}
diff --git a/openai_rev/quora/graphql/ChatListPaginationQuery.graphql b/openai_rev/quora/graphql/ChatListPaginationQuery.graphql
new file mode 100644
index 00000000..6d9ae884
--- /dev/null
+++ b/openai_rev/quora/graphql/ChatListPaginationQuery.graphql
@@ -0,0 +1,378 @@
+query ChatListPaginationQuery(
+ $count: Int = 5
+ $cursor: String
+ $id: ID!
+) {
+ node(id: $id) {
+ __typename
+ ...ChatPageMain_chat_1G22uz
+ id
+ }
+}
+
+fragment BotImage_bot on Bot {
+ displayName
+ ...botHelpers_useDeletion_bot
+ ...BotImage_useProfileImage_bot
+}
+
+fragment BotImage_useProfileImage_bot on Bot {
+ image {
+ __typename
+ ... on LocalBotImage {
+ localName
+ }
+ ... on UrlBotImage {
+ url
+ }
+ }
+ ...botHelpers_useDeletion_bot
+}
+
+fragment ChatMessageDownvotedButton_message on Message {
+ ...MessageFeedbackReasonModal_message
+ ...MessageFeedbackOtherModal_message
+}
+
+fragment ChatMessageDropdownMenu_message on Message {
+ id
+ messageId
+ vote
+ text
+ author
+ ...chatHelpers_isBotMessage
+}
+
+fragment ChatMessageFeedbackButtons_message on Message {
+ id
+ messageId
+ vote
+ voteReason
+ ...ChatMessageDownvotedButton_message
+}
+
+fragment ChatMessageInputView_chat on Chat {
+ id
+ chatId
+ defaultBotObject {
+ nickname
+ messageLimit {
+ dailyBalance
+ shouldShowRemainingMessageCount
+ }
+ hasClearContext
+ isDown
+ ...botHelpers_useDeletion_bot
+ id
+ }
+ shouldShowDisclaimer
+ ...chatHelpers_useSendMessage_chat
+ ...chatHelpers_useSendChatBreak_chat
+}
+
+fragment ChatMessageInputView_edges on MessageEdge {
+ node {
+ ...chatHelpers_isChatBreak
+ ...chatHelpers_isHumanMessage
+ state
+ text
+ id
+ }
+}
+
+fragment ChatMessageOverflowButton_message on Message {
+ text
+ ...ChatMessageDropdownMenu_message
+ ...chatHelpers_isBotMessage
+}
+
+fragment ChatMessageSuggestedReplies_SuggestedReplyButton_chat on Chat {
+ ...chatHelpers_useSendMessage_chat
+}
+
+fragment ChatMessageSuggestedReplies_SuggestedReplyButton_message on Message {
+ messageId
+}
+
+fragment ChatMessageSuggestedReplies_chat on Chat {
+ ...ChatWelcomeView_chat
+ ...ChatMessageSuggestedReplies_SuggestedReplyButton_chat
+ defaultBotObject {
+ hasWelcomeTopics
+ id
+ }
+}
+
+fragment ChatMessageSuggestedReplies_message on Message {
+ suggestedReplies
+ ...ChatMessageSuggestedReplies_SuggestedReplyButton_message
+}
+
+fragment ChatMessage_chat on Chat {
+ defaultBotObject {
+ hasWelcomeTopics
+ hasSuggestedReplies
+ disclaimerText
+ messageLimit {
+ ...ChatPageRateLimitedBanner_messageLimit
+ }
+ ...ChatPageDisclaimer_bot
+ id
+ }
+ ...ChatMessageSuggestedReplies_chat
+ ...ChatWelcomeView_chat
+}
+
+fragment ChatMessage_message on Message {
+ id
+ messageId
+ text
+ author
+ linkifiedText
+ state
+ contentType
+ ...ChatMessageSuggestedReplies_message
+ ...ChatMessageFeedbackButtons_message
+ ...ChatMessageOverflowButton_message
+ ...chatHelpers_isHumanMessage
+ ...chatHelpers_isBotMessage
+ ...chatHelpers_isChatBreak
+ ...chatHelpers_useTimeoutLevel
+ ...MarkdownLinkInner_message
+ ...IdAnnotation_node
+}
+
+fragment ChatMessagesView_chat on Chat {
+ ...ChatMessage_chat
+ ...ChatWelcomeView_chat
+ ...IdAnnotation_node
+ defaultBotObject {
+ hasWelcomeTopics
+ messageLimit {
+ ...ChatPageRateLimitedBanner_messageLimit
+ }
+ id
+ }
+}
+
+fragment ChatMessagesView_edges on MessageEdge {
+ node {
+ id
+ messageId
+ creationTime
+ ...ChatMessage_message
+ ...chatHelpers_isBotMessage
+ ...chatHelpers_isHumanMessage
+ ...chatHelpers_isChatBreak
+ }
+}
+
+fragment ChatPageDeleteFooter_chat on Chat {
+ ...MessageDeleteConfirmationModal_chat
+}
+
+fragment ChatPageDisclaimer_bot on Bot {
+ disclaimerText
+}
+
+fragment ChatPageMainFooter_chat on Chat {
+ defaultBotObject {
+ ...ChatPageMainFooter_useAccessMessage_bot
+ id
+ }
+ ...ChatMessageInputView_chat
+ ...ChatPageShareFooter_chat
+ ...ChatPageDeleteFooter_chat
+}
+
+fragment ChatPageMainFooter_edges on MessageEdge {
+ ...ChatMessageInputView_edges
+}
+
+fragment ChatPageMainFooter_useAccessMessage_bot on Bot {
+ ...botHelpers_useDeletion_bot
+ ...botHelpers_useViewerCanAccessPrivateBot
+}
+
+fragment ChatPageMain_chat_1G22uz on Chat {
+ id
+ chatId
+ ...ChatPageShareFooter_chat
+ ...ChatPageDeleteFooter_chat
+ ...ChatMessagesView_chat
+ ...MarkdownLinkInner_chat
+ ...chatHelpers_useUpdateStaleChat_chat
+ ...ChatSubscriptionPaywallContextWrapper_chat
+ ...ChatPageMainFooter_chat
+ messagesConnection(last: $count, before: $cursor) {
+ edges {
+ ...ChatMessagesView_edges
+ ...ChatPageMainFooter_edges
+ ...MarkdownLinkInner_edges
+ node {
+ ...chatHelpers_useUpdateStaleChat_message
+ id
+ __typename
+ }
+ cursor
+ id
+ }
+ pageInfo {
+ hasPreviousPage
+ startCursor
+ }
+ id
+ }
+}
+
+fragment ChatPageRateLimitedBanner_messageLimit on MessageLimit {
+ numMessagesRemaining
+}
+
+fragment ChatPageShareFooter_chat on Chat {
+ chatId
+}
+
+fragment ChatSubscriptionPaywallContextWrapper_chat on Chat {
+ defaultBotObject {
+ messageLimit {
+ numMessagesRemaining
+ shouldShowRemainingMessageCount
+ }
+ ...SubscriptionPaywallModal_bot
+ id
+ }
+}
+
+fragment ChatWelcomeView_ChatWelcomeButton_chat on Chat {
+ ...chatHelpers_useSendMessage_chat
+}
+
+fragment ChatWelcomeView_chat on Chat {
+ ...ChatWelcomeView_ChatWelcomeButton_chat
+ defaultBotObject {
+ displayName
+ id
+ }
+}
+
+fragment IdAnnotation_node on Node {
+ __isNode: __typename
+ id
+}
+
+fragment MarkdownLinkInner_chat on Chat {
+ id
+ chatId
+ defaultBotObject {
+ nickname
+ id
+ }
+ ...chatHelpers_useSendMessage_chat
+}
+
+fragment MarkdownLinkInner_edges on MessageEdge {
+ node {
+ state
+ id
+ }
+}
+
+fragment MarkdownLinkInner_message on Message {
+ messageId
+}
+
+fragment MessageDeleteConfirmationModal_chat on Chat {
+ id
+}
+
+fragment MessageFeedbackOtherModal_message on Message {
+ id
+ messageId
+}
+
+fragment MessageFeedbackReasonModal_message on Message {
+ id
+ messageId
+}
+
+fragment SubscriptionPaywallModal_bot on Bot {
+ displayName
+ messageLimit {
+ dailyLimit
+ numMessagesRemaining
+ shouldShowRemainingMessageCount
+ resetTime
+ }
+ ...BotImage_bot
+}
+
+fragment botHelpers_useDeletion_bot on Bot {
+ deletionState
+}
+
+fragment botHelpers_useViewerCanAccessPrivateBot on Bot {
+ isPrivateBot
+ viewerIsCreator
+}
+
+fragment chatHelpers_isBotMessage on Message {
+ ...chatHelpers_isHumanMessage
+ ...chatHelpers_isChatBreak
+}
+
+fragment chatHelpers_isChatBreak on Message {
+ author
+}
+
+fragment chatHelpers_isHumanMessage on Message {
+ author
+}
+
+fragment chatHelpers_useSendChatBreak_chat on Chat {
+ id
+ chatId
+ defaultBotObject {
+ nickname
+ introduction
+ model
+ id
+ }
+ shouldShowDisclaimer
+}
+
+fragment chatHelpers_useSendMessage_chat on Chat {
+ id
+ chatId
+ defaultBotObject {
+ id
+ nickname
+ }
+ shouldShowDisclaimer
+}
+
+fragment chatHelpers_useTimeoutLevel on Message {
+ id
+ state
+ text
+ messageId
+ chat {
+ chatId
+ defaultBotNickname
+ id
+ }
+}
+
+fragment chatHelpers_useUpdateStaleChat_chat on Chat {
+ chatId
+ defaultBotObject {
+ contextClearWindowSecs
+ id
+ }
+ ...chatHelpers_useSendChatBreak_chat
+}
+
+fragment chatHelpers_useUpdateStaleChat_message on Message {
+ creationTime
+ ...chatHelpers_isChatBreak
+}
diff --git a/openai_rev/quora/graphql/ChatPaginationQuery.graphql b/openai_rev/quora/graphql/ChatPaginationQuery.graphql
new file mode 100644
index 00000000..f2452cd6
--- /dev/null
+++ b/openai_rev/quora/graphql/ChatPaginationQuery.graphql
@@ -0,0 +1,26 @@
+query ChatPaginationQuery($bot: String!, $before: String, $last: Int! = 10) {
+ chatOfBot(bot: $bot) {
+ id
+ __typename
+ messagesConnection(before: $before, last: $last) {
+ pageInfo {
+ hasPreviousPage
+ }
+ edges {
+ node {
+ id
+ __typename
+ messageId
+ text
+ linkifiedText
+ authorNickname
+ state
+ vote
+ voteReason
+ creationTime
+ suggestedReplies
+ }
+ }
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/ChatViewQuery.graphql b/openai_rev/quora/graphql/ChatViewQuery.graphql
new file mode 100644
index 00000000..c330107d
--- /dev/null
+++ b/openai_rev/quora/graphql/ChatViewQuery.graphql
@@ -0,0 +1,8 @@
+query ChatViewQuery($bot: String!) {
+ chatOfBot(bot: $bot) {
+ id
+ chatId
+ defaultBotNickname
+ shouldShowDisclaimer
+ }
+}
diff --git a/openai_rev/quora/graphql/DeleteHumanMessagesMutation.graphql b/openai_rev/quora/graphql/DeleteHumanMessagesMutation.graphql
new file mode 100644
index 00000000..42692c6e
--- /dev/null
+++ b/openai_rev/quora/graphql/DeleteHumanMessagesMutation.graphql
@@ -0,0 +1,7 @@
+mutation DeleteHumanMessagesMutation($messageIds: [BigInt!]!) {
+ messagesDelete(messageIds: $messageIds) {
+ viewer {
+ id
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/DeleteMessageMutation.graphql b/openai_rev/quora/graphql/DeleteMessageMutation.graphql
new file mode 100644
index 00000000..7b9e36d4
--- /dev/null
+++ b/openai_rev/quora/graphql/DeleteMessageMutation.graphql
@@ -0,0 +1,7 @@
+mutation deleteMessageMutation(
+ $messageIds: [BigInt!]!
+) {
+ messagesDelete(messageIds: $messageIds) {
+ edgeIds
+ }
+} \ No newline at end of file
diff --git a/openai_rev/quora/graphql/HandleFragment.graphql b/openai_rev/quora/graphql/HandleFragment.graphql
new file mode 100644
index 00000000..f53c484b
--- /dev/null
+++ b/openai_rev/quora/graphql/HandleFragment.graphql
@@ -0,0 +1,8 @@
+fragment HandleFragment on Viewer {
+ id
+ poeUser {
+ id
+ uid
+ handle
+ }
+}
diff --git a/openai_rev/quora/graphql/LoginWithVerificationCodeMutation.graphql b/openai_rev/quora/graphql/LoginWithVerificationCodeMutation.graphql
new file mode 100644
index 00000000..723b1f44
--- /dev/null
+++ b/openai_rev/quora/graphql/LoginWithVerificationCodeMutation.graphql
@@ -0,0 +1,13 @@
+mutation LoginWithVerificationCodeMutation(
+ $verificationCode: String!
+ $emailAddress: String
+ $phoneNumber: String
+) {
+ loginWithVerificationCode(
+ verificationCode: $verificationCode
+ emailAddress: $emailAddress
+ phoneNumber: $phoneNumber
+ ) {
+ status
+ }
+}
diff --git a/openai_rev/quora/graphql/MessageAddedSubscription.graphql b/openai_rev/quora/graphql/MessageAddedSubscription.graphql
new file mode 100644
index 00000000..8dc9499c
--- /dev/null
+++ b/openai_rev/quora/graphql/MessageAddedSubscription.graphql
@@ -0,0 +1,100 @@
+subscription messageAdded (
+ $chatId: BigInt!
+) {
+ messageAdded(chatId: $chatId) {
+ id
+ messageId
+ creationTime
+ state
+ ...ChatMessage_message
+ ...chatHelpers_isBotMessage
+ }
+}
+
+fragment ChatMessageDownvotedButton_message on Message {
+ ...MessageFeedbackReasonModal_message
+ ...MessageFeedbackOtherModal_message
+}
+
+fragment ChatMessageDropdownMenu_message on Message {
+ id
+ messageId
+ vote
+ text
+ linkifiedText
+ ...chatHelpers_isBotMessage
+}
+
+fragment ChatMessageFeedbackButtons_message on Message {
+ id
+ messageId
+ vote
+ voteReason
+ ...ChatMessageDownvotedButton_message
+}
+
+fragment ChatMessageOverflowButton_message on Message {
+ text
+ ...ChatMessageDropdownMenu_message
+ ...chatHelpers_isBotMessage
+}
+
+fragment ChatMessageSuggestedReplies_SuggestedReplyButton_message on Message {
+ messageId
+}
+
+fragment ChatMessageSuggestedReplies_message on Message {
+ suggestedReplies
+ ...ChatMessageSuggestedReplies_SuggestedReplyButton_message
+}
+
+fragment ChatMessage_message on Message {
+ id
+ messageId
+ text
+ author
+ linkifiedText
+ state
+ ...ChatMessageSuggestedReplies_message
+ ...ChatMessageFeedbackButtons_message
+ ...ChatMessageOverflowButton_message
+ ...chatHelpers_isHumanMessage
+ ...chatHelpers_isBotMessage
+ ...chatHelpers_isChatBreak
+ ...chatHelpers_useTimeoutLevel
+ ...MarkdownLinkInner_message
+}
+
+fragment MarkdownLinkInner_message on Message {
+ messageId
+}
+
+fragment MessageFeedbackOtherModal_message on Message {
+ id
+ messageId
+}
+
+fragment MessageFeedbackReasonModal_message on Message {
+ id
+ messageId
+}
+
+fragment chatHelpers_isBotMessage on Message {
+ ...chatHelpers_isHumanMessage
+ ...chatHelpers_isChatBreak
+}
+
+fragment chatHelpers_isChatBreak on Message {
+ author
+}
+
+fragment chatHelpers_isHumanMessage on Message {
+ author
+}
+
+fragment chatHelpers_useTimeoutLevel on Message {
+ id
+ state
+ text
+ messageId
+}
diff --git a/openai_rev/quora/graphql/MessageDeletedSubscription.graphql b/openai_rev/quora/graphql/MessageDeletedSubscription.graphql
new file mode 100644
index 00000000..54c1c164
--- /dev/null
+++ b/openai_rev/quora/graphql/MessageDeletedSubscription.graphql
@@ -0,0 +1,6 @@
+subscription MessageDeletedSubscription($chatId: BigInt!) {
+ messageDeleted(chatId: $chatId) {
+ id
+ messageId
+ }
+}
diff --git a/openai_rev/quora/graphql/MessageFragment.graphql b/openai_rev/quora/graphql/MessageFragment.graphql
new file mode 100644
index 00000000..cc860811
--- /dev/null
+++ b/openai_rev/quora/graphql/MessageFragment.graphql
@@ -0,0 +1,13 @@
+fragment MessageFragment on Message {
+ id
+ __typename
+ messageId
+ text
+ linkifiedText
+ authorNickname
+ state
+ vote
+ voteReason
+ creationTime
+ suggestedReplies
+}
diff --git a/openai_rev/quora/graphql/MessageRemoveVoteMutation.graphql b/openai_rev/quora/graphql/MessageRemoveVoteMutation.graphql
new file mode 100644
index 00000000..d5e6e610
--- /dev/null
+++ b/openai_rev/quora/graphql/MessageRemoveVoteMutation.graphql
@@ -0,0 +1,7 @@
+mutation MessageRemoveVoteMutation($messageId: BigInt!) {
+ messageRemoveVote(messageId: $messageId) {
+ message {
+ ...MessageFragment
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/MessageSetVoteMutation.graphql b/openai_rev/quora/graphql/MessageSetVoteMutation.graphql
new file mode 100644
index 00000000..76000df0
--- /dev/null
+++ b/openai_rev/quora/graphql/MessageSetVoteMutation.graphql
@@ -0,0 +1,7 @@
+mutation MessageSetVoteMutation($messageId: BigInt!, $voteType: VoteType!, $reason: String) {
+ messageSetVote(messageId: $messageId, voteType: $voteType, reason: $reason) {
+ message {
+ ...MessageFragment
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/PoeBotCreateMutation.graphql b/openai_rev/quora/graphql/PoeBotCreateMutation.graphql
new file mode 100644
index 00000000..971b4248
--- /dev/null
+++ b/openai_rev/quora/graphql/PoeBotCreateMutation.graphql
@@ -0,0 +1,73 @@
+mutation CreateBotMain_poeBotCreate_Mutation(
+ $model: String!
+ $handle: String!
+ $prompt: String!
+ $isPromptPublic: Boolean!
+ $introduction: String!
+ $description: String!
+ $profilePictureUrl: String
+ $apiUrl: String
+ $apiKey: String
+ $isApiBot: Boolean
+ $hasLinkification: Boolean
+ $hasMarkdownRendering: Boolean
+ $hasSuggestedReplies: Boolean
+ $isPrivateBot: Boolean
+) {
+ poeBotCreate(model: $model, handle: $handle, promptPlaintext: $prompt, isPromptPublic: $isPromptPublic, introduction: $introduction, description: $description, profilePicture: $profilePictureUrl, apiUrl: $apiUrl, apiKey: $apiKey, isApiBot: $isApiBot, hasLinkification: $hasLinkification, hasMarkdownRendering: $hasMarkdownRendering, hasSuggestedReplies: $hasSuggestedReplies, isPrivateBot: $isPrivateBot) {
+ status
+ bot {
+ id
+ ...BotHeader_bot
+ }
+ }
+}
+
+fragment BotHeader_bot on Bot {
+ displayName
+ messageLimit {
+ dailyLimit
+ }
+ ...BotImage_bot
+ ...BotLink_bot
+ ...IdAnnotation_node
+ ...botHelpers_useViewerCanAccessPrivateBot
+ ...botHelpers_useDeletion_bot
+}
+
+fragment BotImage_bot on Bot {
+ displayName
+ ...botHelpers_useDeletion_bot
+ ...BotImage_useProfileImage_bot
+}
+
+fragment BotImage_useProfileImage_bot on Bot {
+ image {
+ __typename
+ ... on LocalBotImage {
+ localName
+ }
+ ... on UrlBotImage {
+ url
+ }
+ }
+ ...botHelpers_useDeletion_bot
+}
+
+fragment BotLink_bot on Bot {
+ displayName
+}
+
+fragment IdAnnotation_node on Node {
+ __isNode: __typename
+ id
+}
+
+fragment botHelpers_useDeletion_bot on Bot {
+ deletionState
+}
+
+fragment botHelpers_useViewerCanAccessPrivateBot on Bot {
+ isPrivateBot
+ viewerIsCreator
+} \ No newline at end of file
diff --git a/openai_rev/quora/graphql/PoeBotEditMutation.graphql b/openai_rev/quora/graphql/PoeBotEditMutation.graphql
new file mode 100644
index 00000000..fdd309ef
--- /dev/null
+++ b/openai_rev/quora/graphql/PoeBotEditMutation.graphql
@@ -0,0 +1,24 @@
+mutation EditBotMain_poeBotEdit_Mutation(
+ $botId: BigInt!
+ $handle: String!
+ $description: String!
+ $introduction: String!
+ $isPromptPublic: Boolean!
+ $baseBot: String!
+ $profilePictureUrl: String
+ $prompt: String!
+ $apiUrl: String
+ $apiKey: String
+ $hasLinkification: Boolean
+ $hasMarkdownRendering: Boolean
+ $hasSuggestedReplies: Boolean
+ $isPrivateBot: Boolean
+) {
+ poeBotEdit(botId: $botId, handle: $handle, description: $description, introduction: $introduction, isPromptPublic: $isPromptPublic, model: $baseBot, promptPlaintext: $prompt, profilePicture: $profilePictureUrl, apiUrl: $apiUrl, apiKey: $apiKey, hasLinkification: $hasLinkification, hasMarkdownRendering: $hasMarkdownRendering, hasSuggestedReplies: $hasSuggestedReplies, isPrivateBot: $isPrivateBot) {
+ status
+ bot {
+ handle
+ id
+ }
+ }
+} \ No newline at end of file
diff --git a/openai_rev/quora/graphql/SendMessageMutation.graphql b/openai_rev/quora/graphql/SendMessageMutation.graphql
new file mode 100644
index 00000000..4b0a4383
--- /dev/null
+++ b/openai_rev/quora/graphql/SendMessageMutation.graphql
@@ -0,0 +1,40 @@
+mutation chatHelpers_sendMessageMutation_Mutation(
+ $chatId: BigInt!
+ $bot: String!
+ $query: String!
+ $source: MessageSource
+ $withChatBreak: Boolean!
+) {
+ messageEdgeCreate(chatId: $chatId, bot: $bot, query: $query, source: $source, withChatBreak: $withChatBreak) {
+ chatBreak {
+ cursor
+ node {
+ id
+ messageId
+ text
+ author
+ suggestedReplies
+ creationTime
+ state
+ }
+ id
+ }
+ message {
+ cursor
+ node {
+ id
+ messageId
+ text
+ author
+ suggestedReplies
+ creationTime
+ state
+ chat {
+ shouldShowDisclaimer
+ id
+ }
+ }
+ id
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/SendVerificationCodeForLoginMutation.graphql b/openai_rev/quora/graphql/SendVerificationCodeForLoginMutation.graphql
new file mode 100644
index 00000000..45af4799
--- /dev/null
+++ b/openai_rev/quora/graphql/SendVerificationCodeForLoginMutation.graphql
@@ -0,0 +1,12 @@
+mutation SendVerificationCodeForLoginMutation(
+ $emailAddress: String
+ $phoneNumber: String
+) {
+ sendVerificationCode(
+ verificationReason: login
+ emailAddress: $emailAddress
+ phoneNumber: $phoneNumber
+ ) {
+ status
+ }
+}
diff --git a/openai_rev/quora/graphql/ShareMessagesMutation.graphql b/openai_rev/quora/graphql/ShareMessagesMutation.graphql
new file mode 100644
index 00000000..92e80db5
--- /dev/null
+++ b/openai_rev/quora/graphql/ShareMessagesMutation.graphql
@@ -0,0 +1,9 @@
+mutation ShareMessagesMutation(
+ $chatId: BigInt!
+ $messageIds: [BigInt!]!
+ $comment: String
+) {
+ messagesShare(chatId: $chatId, messageIds: $messageIds, comment: $comment) {
+ shareCode
+ }
+}
diff --git a/openai_rev/quora/graphql/SignupWithVerificationCodeMutation.graphql b/openai_rev/quora/graphql/SignupWithVerificationCodeMutation.graphql
new file mode 100644
index 00000000..06b2826f
--- /dev/null
+++ b/openai_rev/quora/graphql/SignupWithVerificationCodeMutation.graphql
@@ -0,0 +1,13 @@
+mutation SignupWithVerificationCodeMutation(
+ $verificationCode: String!
+ $emailAddress: String
+ $phoneNumber: String
+) {
+ signupWithVerificationCode(
+ verificationCode: $verificationCode
+ emailAddress: $emailAddress
+ phoneNumber: $phoneNumber
+ ) {
+ status
+ }
+}
diff --git a/openai_rev/quora/graphql/StaleChatUpdateMutation.graphql b/openai_rev/quora/graphql/StaleChatUpdateMutation.graphql
new file mode 100644
index 00000000..de203d47
--- /dev/null
+++ b/openai_rev/quora/graphql/StaleChatUpdateMutation.graphql
@@ -0,0 +1,7 @@
+mutation StaleChatUpdateMutation($chatId: BigInt!) {
+ staleChatUpdate(chatId: $chatId) {
+ message {
+ ...MessageFragment
+ }
+ }
+}
diff --git a/openai_rev/quora/graphql/SubscriptionsMutation.graphql b/openai_rev/quora/graphql/SubscriptionsMutation.graphql
new file mode 100644
index 00000000..b864bd60
--- /dev/null
+++ b/openai_rev/quora/graphql/SubscriptionsMutation.graphql
@@ -0,0 +1,9 @@
+mutation subscriptionsMutation(
+ $subscriptions: [AutoSubscriptionQuery!]!
+) {
+ autoSubscribe(subscriptions: $subscriptions) {
+ viewer {
+ id
+ }
+ }
+} \ No newline at end of file
diff --git a/openai_rev/quora/graphql/SummarizePlainPostQuery.graphql b/openai_rev/quora/graphql/SummarizePlainPostQuery.graphql
new file mode 100644
index 00000000..afa2a84c
--- /dev/null
+++ b/openai_rev/quora/graphql/SummarizePlainPostQuery.graphql
@@ -0,0 +1,3 @@
+query SummarizePlainPostQuery($comment: String!) {
+ summarizePlainPost(comment: $comment)
+}
diff --git a/openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql b/openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql
new file mode 100644
index 00000000..5147c3c5
--- /dev/null
+++ b/openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql
@@ -0,0 +1,3 @@
+query SummarizeQuotePostQuery($comment: String, $quotedPostId: BigInt!) {
+ summarizeQuotePost(comment: $comment, quotedPostId: $quotedPostId)
+}
diff --git a/openai_rev/quora/graphql/SummarizeSharePostQuery.graphql b/openai_rev/quora/graphql/SummarizeSharePostQuery.graphql
new file mode 100644
index 00000000..cb4a623c
--- /dev/null
+++ b/openai_rev/quora/graphql/SummarizeSharePostQuery.graphql
@@ -0,0 +1,3 @@
+query SummarizeSharePostQuery($comment: String!, $chatId: BigInt!, $messageIds: [BigInt!]!) {
+ summarizeSharePost(comment: $comment, chatId: $chatId, messageIds: $messageIds)
+}
diff --git a/openai_rev/quora/graphql/UserSnippetFragment.graphql b/openai_rev/quora/graphql/UserSnippetFragment.graphql
new file mode 100644
index 00000000..17fc8426
--- /dev/null
+++ b/openai_rev/quora/graphql/UserSnippetFragment.graphql
@@ -0,0 +1,14 @@
+fragment UserSnippetFragment on PoeUser {
+ id
+ uid
+ bio
+ handle
+ fullName
+ viewerIsFollowing
+ isPoeOnlyUser
+ profilePhotoURLTiny: profilePhotoUrl(size: tiny)
+ profilePhotoURLSmall: profilePhotoUrl(size: small)
+ profilePhotoURLMedium: profilePhotoUrl(size: medium)
+ profilePhotoURLLarge: profilePhotoUrl(size: large)
+ isFollowable
+}
diff --git a/openai_rev/quora/graphql/ViewerInfoQuery.graphql b/openai_rev/quora/graphql/ViewerInfoQuery.graphql
new file mode 100644
index 00000000..1ecaf9e8
--- /dev/null
+++ b/openai_rev/quora/graphql/ViewerInfoQuery.graphql
@@ -0,0 +1,21 @@
+query ViewerInfoQuery {
+ viewer {
+ id
+ uid
+ ...ViewerStateFragment
+ ...BioFragment
+ ...HandleFragment
+ hasCompletedMultiplayerNux
+ poeUser {
+ id
+ ...UserSnippetFragment
+ }
+ messageLimit{
+ canSend
+ numMessagesRemaining
+ resetTime
+ shouldShowReminder
+ }
+ }
+}
+
diff --git a/openai_rev/quora/graphql/ViewerStateFragment.graphql b/openai_rev/quora/graphql/ViewerStateFragment.graphql
new file mode 100644
index 00000000..3cd83e9c
--- /dev/null
+++ b/openai_rev/quora/graphql/ViewerStateFragment.graphql
@@ -0,0 +1,30 @@
+fragment ViewerStateFragment on Viewer {
+ id
+ __typename
+ iosMinSupportedVersion: integerGate(gateName: "poe_ios_min_supported_version")
+ iosMinEncouragedVersion: integerGate(
+ gateName: "poe_ios_min_encouraged_version"
+ )
+ macosMinSupportedVersion: integerGate(
+ gateName: "poe_macos_min_supported_version"
+ )
+ macosMinEncouragedVersion: integerGate(
+ gateName: "poe_macos_min_encouraged_version"
+ )
+ showPoeDebugPanel: booleanGate(gateName: "poe_show_debug_panel")
+ enableCommunityFeed: booleanGate(gateName: "enable_poe_shares_feed")
+ linkifyText: booleanGate(gateName: "poe_linkify_response")
+ enableSuggestedReplies: booleanGate(gateName: "poe_suggested_replies")
+ removeInviteLimit: booleanGate(gateName: "poe_remove_invite_limit")
+ enableInAppPurchases: booleanGate(gateName: "poe_enable_in_app_purchases")
+ availableBots {
+ nickname
+ displayName
+ profilePicture
+ isDown
+ disclaimer
+ subtitle
+ poweredBy
+ }
+}
+
diff --git a/openai_rev/quora/graphql/ViewerStateUpdatedSubscription.graphql b/openai_rev/quora/graphql/ViewerStateUpdatedSubscription.graphql
new file mode 100644
index 00000000..03fc73d1
--- /dev/null
+++ b/openai_rev/quora/graphql/ViewerStateUpdatedSubscription.graphql
@@ -0,0 +1,43 @@
+subscription viewerStateUpdated {
+ viewerStateUpdated {
+ id
+ ...ChatPageBotSwitcher_viewer
+ }
+}
+
+fragment BotHeader_bot on Bot {
+ displayName
+ messageLimit {
+ dailyLimit
+ }
+ ...BotImage_bot
+}
+
+fragment BotImage_bot on Bot {
+ image {
+ __typename
+ ... on LocalBotImage {
+ localName
+ }
+ ... on UrlBotImage {
+ url
+ }
+ }
+ displayName
+}
+
+fragment BotLink_bot on Bot {
+ displayName
+}
+
+fragment ChatPageBotSwitcher_viewer on Viewer {
+ availableBots {
+ id
+ messageLimit {
+ dailyLimit
+ }
+ ...BotLink_bot
+ ...BotHeader_bot
+ }
+ allowUserCreatedBots: booleanGate(gateName: "enable_user_created_bots")
+}
diff --git a/openai_rev/quora/graphql/__init__.py b/openai_rev/quora/graphql/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/openai_rev/quora/graphql/__init__.py
diff --git a/openai_rev/quora/mail.py b/openai_rev/quora/mail.py
new file mode 100644
index 00000000..864d9568
--- /dev/null
+++ b/openai_rev/quora/mail.py
@@ -0,0 +1,80 @@
+from json import loads
+from re import findall
+from time import sleep
+
+from fake_useragent import UserAgent
+from requests import Session
+
+
+class Emailnator:
+ def __init__(self) -> None:
+ self.client = Session()
+ self.client.get("https://www.emailnator.com/", timeout=6)
+ self.cookies = self.client.cookies.get_dict()
+
+ self.client.headers = {
+ "authority": "www.emailnator.com",
+ "origin": "https://www.emailnator.com",
+ "referer": "https://www.emailnator.com/",
+ "user-agent": UserAgent().random,
+ "x-xsrf-token": self.client.cookies.get("XSRF-TOKEN")[:-3] + "=",
+ }
+
+ self.email = None
+
+ def get_mail(self):
+ response = self.client.post(
+ "https://www.emailnator.com/generate-email",
+ json={
+ "email": [
+ "domain",
+ "plusGmail",
+ "dotGmail",
+ ]
+ },
+ )
+
+ self.email = loads(response.text)["email"][0]
+ return self.email
+
+ def get_message(self):
+ print("Waiting for message...")
+
+ while True:
+ sleep(2)
+ mail_token = self.client.post("https://www.emailnator.com/message-list", json={"email": self.email})
+
+ mail_token = loads(mail_token.text)["messageData"]
+
+ if len(mail_token) == 2:
+ print("Message received!")
+ print(mail_token[1]["messageID"])
+ break
+
+ mail_context = self.client.post(
+ "https://www.emailnator.com/message-list",
+ json={
+ "email": self.email,
+ "messageID": mail_token[1]["messageID"],
+ },
+ )
+
+ return mail_context.text
+
+ def get_verification_code(self):
+ message = self.get_message()
+ code = findall(r';">(\d{6,7})</div>', message)[0]
+ print(f"Verification code: {code}")
+ return code
+
+ def clear_inbox(self):
+ print("Clearing inbox...")
+ self.client.post(
+ "https://www.emailnator.com/delete-all",
+ json={"email": self.email},
+ )
+ print("Inbox cleared!")
+
+ def __del__(self):
+ if self.email:
+ self.clear_inbox()
diff --git a/openai_rev/you/README.md b/openai_rev/you/README.md
new file mode 100644
index 00000000..f759c27c
--- /dev/null
+++ b/openai_rev/you/README.md
@@ -0,0 +1,37 @@
+### Example: `you` (use like openai pypi package) <a name="example-you"></a>
+
+```python
+
+from openai_rev import you
+
+# simple request with links and details
+response = you.Completion.create(
+ prompt="hello world",
+ detailed=True,
+ include_links=True, )
+
+print(response)
+
+# {
+# "response": "...",
+# "links": [...],
+# "extra": {...},
+# "slots": {...}
+# }
+# }
+
+# chatbot
+
+chat = []
+
+while True:
+ prompt = input("You: ")
+
+ response = you.Completion.create(
+ prompt=prompt,
+ chat=chat)
+
+ print("Bot:", response["response"])
+
+ chat.append({"question": prompt, "answer": response["response"]})
+``` \ No newline at end of file
diff --git a/openai_rev/you/__init__.py b/openai_rev/you/__init__.py
new file mode 100644
index 00000000..50d74152
--- /dev/null
+++ b/openai_rev/you/__init__.py
@@ -0,0 +1,108 @@
+import json
+import re
+from typing import Optional, List, Dict, Any
+from uuid import uuid4
+
+from fake_useragent import UserAgent
+from pydantic import BaseModel
+from tls_client import Session
+
+
+class PoeResponse(BaseModel):
+ text: Optional[str] = None
+ links: List[str] = []
+ extra: Dict[str, Any] = {}
+
+
+class Completion:
+ @staticmethod
+ def create(
+ prompt: str,
+ page: int = 1,
+ count: int = 10,
+ safe_search: str = 'Moderate',
+ on_shopping_page: bool = False,
+ mkt: str = '',
+ response_filter: str = 'WebPages,Translations,TimeZone,Computation,RelatedSearches',
+ domain: str = 'youchat',
+ query_trace_id: str = None,
+ chat: list = None,
+ include_links: bool = False,
+ detailed: bool = False,
+ debug: bool = False,
+ ) -> PoeResponse:
+ if chat is None:
+ chat = []
+
+ client = Session(client_identifier='chrome_108')
+ client.headers = Completion.__get_headers()
+
+ response = client.get(
+ f'https://you.com/api/streamingSearch',
+ params={
+ 'q': prompt,
+ 'page': page,
+ 'count': count,
+ 'safeSearch': safe_search,
+ 'onShoppingPage': on_shopping_page,
+ 'mkt': mkt,
+ 'responseFilter': response_filter,
+ 'domain': domain,
+ 'queryTraceId': str(uuid4()) if query_trace_id is None else query_trace_id,
+ 'chat': str(chat), # {'question':'','answer':' ''}
+ },
+ )
+
+ if debug:
+ print('\n\n------------------\n\n')
+ print(response.text)
+ print('\n\n------------------\n\n')
+
+ if 'youChatToken' not in response.text:
+ return Completion.__get_failure_response()
+
+ you_chat_serp_results = re.search(
+ r'(?<=event: youChatSerpResults\ndata:)(.*\n)*?(?=event: )', response.text
+ ).group()
+ third_party_search_results = re.search(
+ r'(?<=event: thirdPartySearchResults\ndata:)(.*\n)*?(?=event: )', response.text
+ ).group()
+ # slots = findall(r"slots\ndata: (.*)\n\nevent", response.text)[0]
+
+ text = ''.join(re.findall(r'{\"youChatToken\": \"(.*?)\"}', response.text))
+
+ extra = {
+ 'youChatSerpResults': json.loads(you_chat_serp_results),
+ # 'slots' : loads(slots)
+ }
+
+ response = PoeResponse(text=text.replace('\\n', '\n').replace('\\\\', '\\').replace('\\"', '"'))
+ if include_links:
+ response.links = json.loads(third_party_search_results)['search']['third_party_search_results']
+
+ if detailed:
+ response.extra = extra
+
+ return response
+
+ @classmethod
+ def __get_headers(cls) -> dict:
+ return {
+ 'authority': 'you.com',
+ 'accept': 'text/event-stream',
+ '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',
+ 'cache-control': 'no-cache',
+ 'referer': 'https://you.com/search?q=who+are+you&tbm=youchat',
+ 'sec-ch-ua': '"Not_A Brand";v="99", "Google Chrome";v="109", "Chromium";v="109"',
+ 'sec-ch-ua-mobile': '?0',
+ 'sec-ch-ua-platform': '"Windows"',
+ 'sec-fetch-dest': 'empty',
+ 'sec-fetch-mode': 'cors',
+ 'sec-fetch-site': 'same-origin',
+ 'cookie': f'safesearch_guest=Moderate; uuid_guest={str(uuid4())}',
+ 'user-agent': UserAgent().random,
+ }
+
+ @classmethod
+ def __get_failure_response(cls) -> PoeResponse:
+ return PoeResponse(text='Unable to fetch the response, Please try again.')