summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/AItianhuSpace.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2023-11-17 04:50:37 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2023-11-17 04:50:37 +0100
commit2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43 (patch)
tree5ed9c277ecceb2b90aabb32dd886a7eb704de101 /g4f/Provider/AItianhuSpace.py
parentAdd Poe Provider, Update AItianhuSpace Porvider (diff)
downloadgpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.tar
gpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.tar.gz
gpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.tar.bz2
gpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.tar.lz
gpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.tar.xz
gpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.tar.zst
gpt4free-2ebbcdf0b81670e2b987e4d44f1efee8f68b3b43.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/AItianhuSpace.py17
1 files changed, 5 insertions, 12 deletions
diff --git a/g4f/Provider/AItianhuSpace.py b/g4f/Provider/AItianhuSpace.py
index a9a824cf..545944c0 100644
--- a/g4f/Provider/AItianhuSpace.py
+++ b/g4f/Provider/AItianhuSpace.py
@@ -24,7 +24,7 @@ class AItianhuSpace(BaseProvider):
proxy: str = None,
timeout: int = 120,
browser: WebDriver = None,
- hidden_display: bool = True,
+ headless: bool = True,
**kwargs
) -> CreateResult:
if not model:
@@ -38,13 +38,7 @@ class AItianhuSpace(BaseProvider):
print(f"AItianhuSpace | using domain: {domain}")
url = f"https://{domain}"
prompt = format_prompt(messages)
- if browser:
- driver = browser
- else:
- if hidden_display:
- driver, display = get_browser("", True, proxy)
- else:
- driver = get_browser("", False, proxy)
+ driver = browser if browser else get_browser("", headless, proxy)
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
@@ -66,6 +60,7 @@ document.getElementById('sheet').addEventListener('click', () => {{
original_window = driver.current_window_handle
for window_handle in driver.window_handles:
if window_handle != original_window:
+ driver.close()
driver.switch_to.window(window_handle)
break
@@ -120,9 +115,7 @@ return "";
else:
time.sleep(0.1)
finally:
- driver.close()
if not browser:
+ driver.close()
time.sleep(0.1)
- driver.quit()
- if hidden_display:
- display.stop() \ No newline at end of file
+ driver.quit() \ No newline at end of file