diff options
author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-01-22 04:33:19 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-01-22 04:33:19 +0100 |
commit | efbcc04d8ac45f5acf1e394d9445ad8569a0a127 (patch) | |
tree | fe9d386a04a280c8aaeec2faaa05819058801d12 /g4f/Provider/selenium/AItianhuSpace.py | |
parent | ~ | to do list | survey: https://forms.gle/FhtGtUorygyJn7i48 (diff) | |
parent | Improve CreateImagesBing (diff) | |
download | gpt4free-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/AItianhuSpace.py')
-rw-r--r-- | g4f/Provider/selenium/AItianhuSpace.py | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/g4f/Provider/selenium/AItianhuSpace.py b/g4f/Provider/selenium/AItianhuSpace.py new file mode 100644 index 00000000..9878bd5a --- /dev/null +++ b/g4f/Provider/selenium/AItianhuSpace.py @@ -0,0 +1,117 @@ +from __future__ import annotations + +import time +import random + +from ...typing import CreateResult, Messages +from ..base_provider import AbstractProvider +from ..helper import format_prompt, get_random_string +from ...webdriver import WebDriver, WebDriverSession +from ... import debug + +class AItianhuSpace(AbstractProvider): + url = "https://chat3.aiyunos.top/" + working = True + supports_stream = True + supports_gpt_35_turbo = True + _domains = ["aitianhu.com", "aitianhu1.top"] + + @classmethod + def create_completion( + cls, + model: str, + messages: Messages, + stream: bool, + domain: str = None, + proxy: str = None, + timeout: int = 120, + webdriver: WebDriver = None, + headless: bool = True, + **kwargs + ) -> CreateResult: + if not model: + model = "gpt-3.5-turbo" + if not domain: + rand = get_random_string(6) + domain = random.choice(cls._domains) + domain = f"{rand}.{domain}" + if debug.logging: + print(f"AItianhuSpace | using domain: {domain}") + url = f"https://{domain}" + prompt = format_prompt(messages) + + with WebDriverSession(webdriver, "", headless=headless, proxy=proxy) as driver: + from selenium.webdriver.common.by import By + from selenium.webdriver.support.ui import WebDriverWait + from selenium.webdriver.support import expected_conditions as EC + + wait = WebDriverWait(driver, timeout) + + # Bypass devtools detection + driver.get("https://blank.page/") + wait.until(EC.visibility_of_element_located((By.ID, "sheet"))) + driver.execute_script(f""" + document.getElementById('sheet').addEventListener('click', () => {{ + window.open('{url}', '_blank'); + }}); + """) + driver.find_element(By.ID, "sheet").click() + time.sleep(10) + + 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 + + # Wait for page load + wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "textarea.n-input__textarea-el"))) + + # Register hook in XMLHttpRequest + script = """ +const _http_request_open = XMLHttpRequest.prototype.open; +window._last_message = window._message = ""; +window._loadend = false; +XMLHttpRequest.prototype.open = function(method, url) { + if (url == "/api/chat-process") { + this.addEventListener("progress", (event) => { + const lines = this.responseText.split("\\n"); + try { + window._message = JSON.parse(lines[lines.length-1])["text"]; + } catch(e) { } + }); + this.addEventListener("loadend", (event) => { + window._loadend = true; + }); + } + return _http_request_open.call(this, method, url); +} +""" + driver.execute_script(script) + + # Submit prompt + driver.find_element(By.CSS_SELECTOR, "textarea.n-input__textarea-el").send_keys(prompt) + driver.find_element(By.CSS_SELECTOR, "button.n-button.n-button--primary-type.n-button--medium-type").click() + + # Read response + while True: + chunk = driver.execute_script(""" +if (window._message && window._message != window._last_message) { + try { + return window._message.substring(window._last_message.length); + } finally { + window._last_message = window._message; + } +} +if (window._loadend) { + return null; +} +return ""; +""") + if chunk: + yield chunk + elif chunk != "": + break + else: + time.sleep(0.1)
\ No newline at end of file |