summaryrefslogtreecommitdiffstats
path: root/openai_rev
diff options
context:
space:
mode:
authornoptuno <repollo.marrero@gmail.com>2023-04-28 12:33:51 +0200
committernoptuno <repollo.marrero@gmail.com>2023-04-28 12:33:51 +0200
commit396d7e11b2623cf05c3b0fd4ff9b6cce667835fd (patch)
tree4d7f4531e082deba4fb6bac7e5b221d887e066d9 /openai_rev
parentRemove ignored files from repository (diff)
parentMerge pull request #241 from ethanx40/main (diff)
downloadgpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.gz
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.bz2
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.lz
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.xz
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.zst
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.zip
Diffstat (limited to 'openai_rev')
-rw-r--r--openai_rev/__init__.py1
-rw-r--r--openai_rev/forefront/README.md15
-rw-r--r--openai_rev/forefront/__init__.py189
-rw-r--r--openai_rev/forefront/mail.py52
-rw-r--r--openai_rev/forefront/models.py26
-rw-r--r--openai_rev/openai_rev.py42
-rw-r--r--openai_rev/quora/README.md68
-rw-r--r--openai_rev/quora/__init__.py472
-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/theb/README.md10
-rw-r--r--openai_rev/theb/__init__.py52
-rw-r--r--openai_rev/theb/theb_test.py4
-rw-r--r--openai_rev/you/README.md37
-rw-r--r--openai_rev/you/__init__.py108
50 files changed, 0 insertions, 2710 deletions
diff --git a/openai_rev/__init__.py b/openai_rev/__init__.py
deleted file mode 100644
index 9076abe9..00000000
--- a/openai_rev/__init__.py
+++ /dev/null
@@ -1 +0,0 @@
-from .openai_rev import Provider
diff --git a/openai_rev/forefront/README.md b/openai_rev/forefront/README.md
deleted file mode 100644
index 94089faa..00000000
--- a/openai_rev/forefront/README.md
+++ /dev/null
@@ -1,15 +0,0 @@
-### 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
deleted file mode 100644
index 6fece65a..00000000
--- a/openai_rev/forefront/__init__.py
+++ /dev/null
@@ -1,189 +0,0 @@
-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
- 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
diff --git a/openai_rev/forefront/mail.py b/openai_rev/forefront/mail.py
deleted file mode 100644
index 2c00051c..00000000
--- a/openai_rev/forefront/mail.py
+++ /dev/null
@@ -1,52 +0,0 @@
-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
deleted file mode 100644
index 23e90903..00000000
--- a/openai_rev/forefront/models.py
+++ /dev/null
@@ -1,26 +0,0 @@
-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
deleted file mode 100644
index 6b88298b..00000000
--- a/openai_rev/openai_rev.py
+++ /dev/null
@@ -1,42 +0,0 @@
-from enum import Enum
-
-from openai_rev import forefront
-from openai_rev import quora
-from openai_rev import theb
-from openai_rev import you
-
-
-class Provider(Enum):
- You = 'you'
- Poe = 'poe'
- ForeFront = 'fore_front'
- Theb = 'theb'
-
-
-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)
- elif provider == Provider.ForeFront:
- return Completion.__fore_front_service(prompt, **kwargs)
- elif provider == Provider.Theb:
- return Completion.__theb_service(prompt, **kwargs)
-
- @classmethod
- def __you_service(cls, prompt: str, **kwargs) -> str:
- return you.Completion.create(prompt, **kwargs).text
-
- @classmethod
- def __poe_service(cls, prompt: str, **kwargs) -> str:
- return quora.Completion.create(prompt=prompt, **kwargs).text
-
- @classmethod
- def __fore_front_service(cls, prompt: str, **kwargs) -> str:
- return forefront.Completion.create(prompt=prompt, **kwargs).text
-
- @classmethod
- def __theb_service(cls, prompt: str, **kwargs):
- return ''.join(theb.Completion.create(prompt=prompt))
diff --git a/openai_rev/quora/README.md b/openai_rev/quora/README.md
deleted file mode 100644
index dc2bb32d..00000000
--- a/openai_rev/quora/README.md
+++ /dev/null
@@ -1,68 +0,0 @@
-#### 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
deleted file mode 100644
index 5303f206..00000000
--- a/openai_rev/quora/__init__.py
+++ /dev/null
@@ -1,472 +0,0 @@
-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:
-
- 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:
- _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'],
- '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
deleted file mode 100644
index 42814f2c..00000000
--- a/openai_rev/quora/api.py
+++ /dev/null
@@ -1,545 +0,0 @@
-# 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
deleted file mode 100644
index 9cccf6ba..00000000
--- a/openai_rev/quora/cookies.txt
+++ /dev/null
@@ -1,30 +0,0 @@
-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
deleted file mode 100644
index 01e6bc8c..00000000
--- a/openai_rev/quora/graphql/AddHumanMessageMutation.graphql
+++ /dev/null
@@ -1,52 +0,0 @@
-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
deleted file mode 100644
index b28d9903..00000000
--- a/openai_rev/quora/graphql/AddMessageBreakMutation.graphql
+++ /dev/null
@@ -1,17 +0,0 @@
-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
deleted file mode 100644
index 6cf7bf74..00000000
--- a/openai_rev/quora/graphql/AutoSubscriptionMutation.graphql
+++ /dev/null
@@ -1,7 +0,0 @@
-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
deleted file mode 100644
index c4218030..00000000
--- a/openai_rev/quora/graphql/BioFragment.graphql
+++ /dev/null
@@ -1,8 +0,0 @@
-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
deleted file mode 100644
index 664b107f..00000000
--- a/openai_rev/quora/graphql/ChatAddedSubscription.graphql
+++ /dev/null
@@ -1,5 +0,0 @@
-subscription ChatAddedSubscription {
- chatAdded {
- ...ChatFragment
- }
-}
diff --git a/openai_rev/quora/graphql/ChatFragment.graphql b/openai_rev/quora/graphql/ChatFragment.graphql
deleted file mode 100644
index 605645ff..00000000
--- a/openai_rev/quora/graphql/ChatFragment.graphql
+++ /dev/null
@@ -1,6 +0,0 @@
-fragment ChatFragment on Chat {
- id
- chatId
- defaultBotNickname
- shouldShowDisclaimer
-}
diff --git a/openai_rev/quora/graphql/ChatListPaginationQuery.graphql b/openai_rev/quora/graphql/ChatListPaginationQuery.graphql
deleted file mode 100644
index 6d9ae884..00000000
--- a/openai_rev/quora/graphql/ChatListPaginationQuery.graphql
+++ /dev/null
@@ -1,378 +0,0 @@
-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
deleted file mode 100644
index f2452cd6..00000000
--- a/openai_rev/quora/graphql/ChatPaginationQuery.graphql
+++ /dev/null
@@ -1,26 +0,0 @@
-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
deleted file mode 100644
index c330107d..00000000
--- a/openai_rev/quora/graphql/ChatViewQuery.graphql
+++ /dev/null
@@ -1,8 +0,0 @@
-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
deleted file mode 100644
index 42692c6e..00000000
--- a/openai_rev/quora/graphql/DeleteHumanMessagesMutation.graphql
+++ /dev/null
@@ -1,7 +0,0 @@
-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
deleted file mode 100644
index 7b9e36d4..00000000
--- a/openai_rev/quora/graphql/DeleteMessageMutation.graphql
+++ /dev/null
@@ -1,7 +0,0 @@
-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
deleted file mode 100644
index f53c484b..00000000
--- a/openai_rev/quora/graphql/HandleFragment.graphql
+++ /dev/null
@@ -1,8 +0,0 @@
-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
deleted file mode 100644
index 723b1f44..00000000
--- a/openai_rev/quora/graphql/LoginWithVerificationCodeMutation.graphql
+++ /dev/null
@@ -1,13 +0,0 @@
-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
deleted file mode 100644
index 8dc9499c..00000000
--- a/openai_rev/quora/graphql/MessageAddedSubscription.graphql
+++ /dev/null
@@ -1,100 +0,0 @@
-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
deleted file mode 100644
index 54c1c164..00000000
--- a/openai_rev/quora/graphql/MessageDeletedSubscription.graphql
+++ /dev/null
@@ -1,6 +0,0 @@
-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
deleted file mode 100644
index cc860811..00000000
--- a/openai_rev/quora/graphql/MessageFragment.graphql
+++ /dev/null
@@ -1,13 +0,0 @@
-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
deleted file mode 100644
index d5e6e610..00000000
--- a/openai_rev/quora/graphql/MessageRemoveVoteMutation.graphql
+++ /dev/null
@@ -1,7 +0,0 @@
-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
deleted file mode 100644
index 76000df0..00000000
--- a/openai_rev/quora/graphql/MessageSetVoteMutation.graphql
+++ /dev/null
@@ -1,7 +0,0 @@
-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
deleted file mode 100644
index 971b4248..00000000
--- a/openai_rev/quora/graphql/PoeBotCreateMutation.graphql
+++ /dev/null
@@ -1,73 +0,0 @@
-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
deleted file mode 100644
index fdd309ef..00000000
--- a/openai_rev/quora/graphql/PoeBotEditMutation.graphql
+++ /dev/null
@@ -1,24 +0,0 @@
-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
deleted file mode 100644
index 4b0a4383..00000000
--- a/openai_rev/quora/graphql/SendMessageMutation.graphql
+++ /dev/null
@@ -1,40 +0,0 @@
-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
deleted file mode 100644
index 45af4799..00000000
--- a/openai_rev/quora/graphql/SendVerificationCodeForLoginMutation.graphql
+++ /dev/null
@@ -1,12 +0,0 @@
-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
deleted file mode 100644
index 92e80db5..00000000
--- a/openai_rev/quora/graphql/ShareMessagesMutation.graphql
+++ /dev/null
@@ -1,9 +0,0 @@
-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
deleted file mode 100644
index 06b2826f..00000000
--- a/openai_rev/quora/graphql/SignupWithVerificationCodeMutation.graphql
+++ /dev/null
@@ -1,13 +0,0 @@
-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
deleted file mode 100644
index de203d47..00000000
--- a/openai_rev/quora/graphql/StaleChatUpdateMutation.graphql
+++ /dev/null
@@ -1,7 +0,0 @@
-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
deleted file mode 100644
index b864bd60..00000000
--- a/openai_rev/quora/graphql/SubscriptionsMutation.graphql
+++ /dev/null
@@ -1,9 +0,0 @@
-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
deleted file mode 100644
index afa2a84c..00000000
--- a/openai_rev/quora/graphql/SummarizePlainPostQuery.graphql
+++ /dev/null
@@ -1,3 +0,0 @@
-query SummarizePlainPostQuery($comment: String!) {
- summarizePlainPost(comment: $comment)
-}
diff --git a/openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql b/openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql
deleted file mode 100644
index 5147c3c5..00000000
--- a/openai_rev/quora/graphql/SummarizeQuotePostQuery.graphql
+++ /dev/null
@@ -1,3 +0,0 @@
-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
deleted file mode 100644
index cb4a623c..00000000
--- a/openai_rev/quora/graphql/SummarizeSharePostQuery.graphql
+++ /dev/null
@@ -1,3 +0,0 @@
-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
deleted file mode 100644
index 17fc8426..00000000
--- a/openai_rev/quora/graphql/UserSnippetFragment.graphql
+++ /dev/null
@@ -1,14 +0,0 @@
-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
deleted file mode 100644
index 1ecaf9e8..00000000
--- a/openai_rev/quora/graphql/ViewerInfoQuery.graphql
+++ /dev/null
@@ -1,21 +0,0 @@
-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
deleted file mode 100644
index 3cd83e9c..00000000
--- a/openai_rev/quora/graphql/ViewerStateFragment.graphql
+++ /dev/null
@@ -1,30 +0,0 @@
-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
deleted file mode 100644
index 03fc73d1..00000000
--- a/openai_rev/quora/graphql/ViewerStateUpdatedSubscription.graphql
+++ /dev/null
@@ -1,43 +0,0 @@
-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
deleted file mode 100644
index e69de29b..00000000
--- a/openai_rev/quora/graphql/__init__.py
+++ /dev/null
diff --git a/openai_rev/quora/mail.py b/openai_rev/quora/mail.py
deleted file mode 100644
index 864d9568..00000000
--- a/openai_rev/quora/mail.py
+++ /dev/null
@@ -1,80 +0,0 @@
-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/theb/README.md b/openai_rev/theb/README.md
deleted file mode 100644
index bd37ba3f..00000000
--- a/openai_rev/theb/README.md
+++ /dev/null
@@ -1,10 +0,0 @@
-### Example: `theb` (use like openai pypi package) <a name="example-theb"></a>
-
-```python
-# import library
-from openai_rev import theb
-
-# simple streaming completion
-for token in theb.Completion.create('hello world'):
- print(token, end='', flush=True)
-``` \ No newline at end of file
diff --git a/openai_rev/theb/__init__.py b/openai_rev/theb/__init__.py
deleted file mode 100644
index fa79fdd9..00000000
--- a/openai_rev/theb/__init__.py
+++ /dev/null
@@ -1,52 +0,0 @@
-from json import loads
-from queue import Queue, Empty
-from re import findall
-from threading import Thread
-
-from curl_cffi import requests
-
-
-class Completion:
- # experimental
- part1 = '{"role":"assistant","id":"chatcmpl'
- part2 = '"},"index":0,"finish_reason":null}]}}'
- regex = rf'{part1}(.*){part2}'
-
- timer = None
- message_queue = Queue()
- stream_completed = False
-
- @classmethod
- def request(cls, prompt: str):
- headers = {
- 'authority': 'chatbot.theb.ai',
- 'content-type': 'application/json',
- 'origin': 'https://chatbot.theb.ai',
- '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',
- }
-
- requests.post(
- 'https://chatbot.theb.ai/api/chat-process',
- headers=headers,
- content_callback=Completion.handle_stream_response,
- json={'prompt': prompt, 'options': {}},
- )
-
- Completion.stream_completed = True
-
- @staticmethod
- def create(prompt: str):
- Thread(target=Completion.request, args=[prompt]).start()
-
- while not Completion.stream_completed or not Completion.message_queue.empty():
- try:
- message = Completion.message_queue.get(timeout=0.01)
- for message in findall(Completion.regex, message):
- yield loads(Completion.part1 + message + Completion.part2)['delta']
-
- except Empty:
- pass
-
- @staticmethod
- def handle_stream_response(response):
- Completion.message_queue.put(response.decode())
diff --git a/openai_rev/theb/theb_test.py b/openai_rev/theb/theb_test.py
deleted file mode 100644
index 805ef094..00000000
--- a/openai_rev/theb/theb_test.py
+++ /dev/null
@@ -1,4 +0,0 @@
-from openai_rev import theb
-
-for token in theb.Completion.create('hello world'):
- print(token, end='', flush=True)
diff --git a/openai_rev/you/README.md b/openai_rev/you/README.md
deleted file mode 100644
index f759c27c..00000000
--- a/openai_rev/you/README.md
+++ /dev/null
@@ -1,37 +0,0 @@
-### 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
deleted file mode 100644
index 50d74152..00000000
--- a/openai_rev/you/__init__.py
+++ /dev/null
@@ -1,108 +0,0 @@
-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.')