summaryrefslogtreecommitdiffstats
path: root/g4f/Provider
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2023-11-16 18:51:24 +0100
committerGitHub <noreply@github.com>2023-11-16 18:51:24 +0100
commit5c5235671e734441b80734a66f653844d8758364 (patch)
tree78d32cbf478fdcffe5117022738020c973cb46cb /g4f/Provider
parentUpdate PerplexityAi.py (diff)
downloadgpt4free-5c5235671e734441b80734a66f653844d8758364.tar
gpt4free-5c5235671e734441b80734a66f653844d8758364.tar.gz
gpt4free-5c5235671e734441b80734a66f653844d8758364.tar.bz2
gpt4free-5c5235671e734441b80734a66f653844d8758364.tar.lz
gpt4free-5c5235671e734441b80734a66f653844d8758364.tar.xz
gpt4free-5c5235671e734441b80734a66f653844d8758364.tar.zst
gpt4free-5c5235671e734441b80734a66f653844d8758364.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/MyShell.py16
1 files changed, 6 insertions, 10 deletions
diff --git a/g4f/Provider/MyShell.py b/g4f/Provider/MyShell.py
index 09f6feb9..dc3934fb 100644
--- a/g4f/Provider/MyShell.py
+++ b/g4f/Provider/MyShell.py
@@ -2,15 +2,9 @@ from __future__ import annotations
import time, json
-try:
- from selenium.webdriver.remote.webdriver import WebDriver
-except ImportError:
- class WebDriver():
- pass
-
from ..typing import CreateResult, Messages
from .base_provider import BaseProvider
-from .helper import format_prompt, get_browser
+from .helper import WebDriver, format_prompt, get_browser
class MyShell(BaseProvider):
url = "https://app.myshell.ai/chat"
@@ -27,11 +21,11 @@ class MyShell(BaseProvider):
proxy: str = None,
timeout: int = 120,
browser: WebDriver = None,
- display: bool = True,
+ hidden_display: bool = True,
**kwargs
) -> CreateResult:
if not browser:
- if display:
+ if hidden_display:
driver, display = get_browser("", True, proxy)
else:
display = get_browser("", False, proxy)
@@ -44,9 +38,11 @@ class MyShell(BaseProvider):
driver.get(cls.url)
try:
+ # Wait for page load
WebDriverWait(driver, timeout).until(
EC.presence_of_element_located((By.CSS_SELECTOR, "body:not(.no-js)"))
)
+ # Send message
script = """
response = await fetch("https://api.myshell.ai/v1/bot/chat/send_message", {
"headers": {
@@ -90,5 +86,5 @@ return content;
if not browser:
time.sleep(0.1)
driver.quit()
- if display:
+ if hidden_display:
display.stop() \ No newline at end of file