gpt4free/g4f/Provider/AItianhuSpace.py
Heiner Lohaus c617b18d12 Add support for all models
Add AbstractProvider class
Add ProviderType type
Add get_last_provider function
Add version module and  VersionUtils
Display used provider in gui
Fix error response in api
2024-01-01 17:48:57 +01:00

117 lines
3.8 KiB
Python

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)