summaryrefslogtreecommitdiffstats
path: root/quora
diff options
context:
space:
mode:
authorRaju Komati <komatiraju032@gmail.com>2023-04-24 10:22:43 +0200
committerRaju Komati <komatiraju032@gmail.com>2023-04-24 10:22:43 +0200
commit2200e777c16687c22dad2cc8759b590ac9a83d6a (patch)
treeac58d7ad36e4de8f828a337cac5b065e80075f90 /quora
parentadded create bot method in Poe class (diff)
downloadgpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.tar
gpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.tar.gz
gpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.tar.bz2
gpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.tar.lz
gpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.tar.xz
gpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.tar.zst
gpt4free-2200e777c16687c22dad2cc8759b590ac9a83d6a.zip
Diffstat (limited to 'quora')
-rw-r--r--quora/README.md7
-rw-r--r--quora/__init__.py27
2 files changed, 23 insertions, 11 deletions
diff --git a/quora/README.md b/quora/README.md
index ca50f644..ea87d1c0 100644
--- a/quora/README.md
+++ b/quora/README.md
@@ -52,7 +52,12 @@ print(response.completion.choices[0].text)
```python
from quora import Poe
-poe = Poe(model='sage')
+# available models: ['Sage', 'GPT-4', 'Claude+', 'Claude-instant', 'ChatGPT', 'Dragonfly', 'NeevaAI']
+
+poe = Poe(model='gpt-3.5-turbo')
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')
+
``` \ No newline at end of file
diff --git a/quora/__init__.py b/quora/__init__.py
index e739ee3c..88c9d0c4 100644
--- a/quora/__init__.py
+++ b/quora/__init__.py
@@ -6,6 +6,7 @@ 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
from urllib.parse import unquote
import selenium.webdriver.support.expected_conditions as EC
@@ -18,17 +19,18 @@ from tls_client import Session as TLS
from quora.api import Client as PoeClient
from quora.mail import Emailnator
-from typing import Optional
# from twocaptcha import TwoCaptcha
# solver = TwoCaptcha('72747bf24a9d89b4dcc1b24875efd358')
MODELS = {
- "sage": "capybara",
- "gpt-4": "beaver",
- "claude-v1.2": "a2_2",
- "claude-instant-v1.0": "a2",
- "gpt-3.5-turbo": "chinchilla",
+ "Sage": "capybara",
+ "GPT-4": "beaver",
+ "Claude+": "a2_2",
+ "Claude-instant": "a2",
+ "ChatGPT": "chinchilla",
+ "Dragonfly": "nutria",
+ "NeevaAI": "hutia",
}
@@ -377,7 +379,7 @@ class Completion:
class Poe:
- def __init__(self, model: str = "gpt-3.5-turbo"):
+ def __init__(self, model: str = "ChatGPT"):
self.cookie = self.__load_cookie()
self.model = MODELS[model]
self.client = PoeClient(self.cookie)
@@ -441,7 +443,7 @@ class Poe:
return cookie
def chat(self, message: str, model: Optional[str] = None) -> str:
- model = model or self.model
+ model = MODELS[model] or self.model
response = None
for chunk in self.client.send_message(model, message):
response = chunk["text"]
@@ -452,11 +454,13 @@ class Poe:
name: str,
/,
prompt: str = "",
- base_model: str = "gpt-3.5-turbo",
+ 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.')
+ raise RuntimeError(
+ "Sorry, the base_model you provided does not exist. Please check and try again."
+ )
response = self.client.create_bot(
handle=name,
@@ -465,3 +469,6 @@ class Poe:
description=description,
)
print(f'Successfully created bot with name: {response["bot"]["displayName"]}')
+
+ def list_bots(self) -> list:
+ return list(self.client.bot_names.values())