summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/selenium/MyShell.py
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-01-22 04:33:19 +0100
committerGitHub <noreply@github.com>2024-01-22 04:33:19 +0100
commitefbcc04d8ac45f5acf1e394d9445ad8569a0a127 (patch)
treefe9d386a04a280c8aaeec2faaa05819058801d12 /g4f/Provider/selenium/MyShell.py
parent~ | to do list | survey: https://forms.gle/FhtGtUorygyJn7i48 (diff)
parentImprove CreateImagesBing (diff)
downloadgpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.tar
gpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.tar.gz
gpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.tar.bz2
gpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.tar.lz
gpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.tar.xz
gpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.tar.zst
gpt4free-efbcc04d8ac45f5acf1e394d9445ad8569a0a127.zip
Diffstat (limited to 'g4f/Provider/selenium/MyShell.py')
-rw-r--r--g4f/Provider/selenium/MyShell.py76
1 files changed, 76 insertions, 0 deletions
diff --git a/g4f/Provider/selenium/MyShell.py b/g4f/Provider/selenium/MyShell.py
new file mode 100644
index 00000000..a3f246ff
--- /dev/null
+++ b/g4f/Provider/selenium/MyShell.py
@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+import time, json
+
+from ...typing import CreateResult, Messages
+from ..base_provider import AbstractProvider
+from ..helper import format_prompt
+from ...webdriver import WebDriver, WebDriverSession, bypass_cloudflare
+
+class MyShell(AbstractProvider):
+ url = "https://app.myshell.ai/chat"
+ working = True
+ supports_gpt_35_turbo = True
+ supports_stream = True
+
+ @classmethod
+ def create_completion(
+ cls,
+ model: str,
+ messages: Messages,
+ stream: bool,
+ proxy: str = None,
+ timeout: int = 120,
+ webdriver: WebDriver = None,
+ **kwargs
+ ) -> CreateResult:
+ with WebDriverSession(webdriver, "", proxy=proxy) as driver:
+ bypass_cloudflare(driver, cls.url, timeout)
+
+ # Send request with message
+ data = {
+ "botId": "4738",
+ "conversation_scenario": 3,
+ "message": format_prompt(messages),
+ "messageType": 1
+ }
+ script = """
+response = await fetch("https://api.myshell.ai/v1/bot/chat/send_message", {
+ "headers": {
+ "accept": "application/json",
+ "content-type": "application/json",
+ "myshell-service-name": "organics-api",
+ "visitor-id": localStorage.getItem("mix_visitorId")
+ },
+ "body": '{body}',
+ "method": "POST"
+})
+window._reader = response.body.pipeThrough(new TextDecoderStream()).getReader();
+"""
+ driver.execute_script(script.replace("{body}", json.dumps(data)))
+ script = """
+chunk = await window._reader.read();
+if (chunk.done) {
+ return null;
+}
+content = '';
+chunk.value.split('\\n').forEach((line, index) => {
+ if (line.startsWith('data: ')) {
+ try {
+ const data = JSON.parse(line.substring('data: '.length));
+ if ('content' in data) {
+ content += data['content'];
+ }
+ } catch(e) {}
+ }
+});
+return content;
+"""
+ while True:
+ chunk = driver.execute_script(script)
+ if chunk:
+ yield chunk
+ elif chunk != "":
+ break
+ else:
+ time.sleep(0.1) \ No newline at end of file