2023-11-17 22:38:31 -05:00
|
|
|
from __future__ import annotations
|
|
|
|
|
|
|
|
import time, json, time
|
|
|
|
|
|
|
|
from ..typing import CreateResult, Messages
|
2024-01-01 11:48:57 -05:00
|
|
|
from .base_provider import AbstractProvider
|
2023-12-01 23:40:07 -05:00
|
|
|
from ..webdriver import WebDriver, WebDriverSession
|
2023-11-17 22:38:31 -05:00
|
|
|
|
2024-01-01 11:48:57 -05:00
|
|
|
class TalkAi(AbstractProvider):
|
2023-11-17 22:38:31 -05:00
|
|
|
url = "https://talkai.info"
|
|
|
|
working = True
|
|
|
|
supports_gpt_35_turbo = True
|
|
|
|
supports_stream = True
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def create_completion(
|
|
|
|
cls,
|
|
|
|
model: str,
|
|
|
|
messages: Messages,
|
|
|
|
stream: bool,
|
|
|
|
proxy: str = None,
|
2023-11-20 08:00:40 -05:00
|
|
|
webdriver: WebDriver = None,
|
2023-11-17 22:38:31 -05:00
|
|
|
**kwargs
|
|
|
|
) -> CreateResult:
|
2023-11-20 08:00:40 -05:00
|
|
|
with WebDriverSession(webdriver, "", virtual_display=True, proxy=proxy) as driver:
|
2023-11-18 23:36:04 -05:00
|
|
|
from selenium.webdriver.common.by import By
|
|
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
|
|
from selenium.webdriver.support import expected_conditions as EC
|
2023-11-17 22:38:31 -05:00
|
|
|
|
|
|
|
driver.get(f"{cls.url}/chat/")
|
|
|
|
|
|
|
|
# Wait for page load
|
|
|
|
WebDriverWait(driver, 240).until(
|
|
|
|
EC.presence_of_element_located((By.CSS_SELECTOR, "body.chat-page"))
|
|
|
|
)
|
|
|
|
|
|
|
|
data = {
|
|
|
|
"type": "chat",
|
|
|
|
"message": messages[-1]["content"],
|
|
|
|
"messagesHistory": [{
|
|
|
|
"from": "you" if message["role"] == "user" else "chatGPT",
|
|
|
|
"content": message["content"]
|
|
|
|
} for message in messages],
|
|
|
|
"model": model if model else "gpt-3.5-turbo",
|
|
|
|
"max_tokens": 256,
|
|
|
|
"temperature": 1,
|
|
|
|
"top_p": 1,
|
|
|
|
"presence_penalty": 0,
|
|
|
|
"frequency_penalty": 0,
|
|
|
|
**kwargs
|
|
|
|
}
|
|
|
|
script = """
|
|
|
|
const response = await fetch("/chat/send2/", {
|
|
|
|
"headers": {
|
|
|
|
"Accept": "application/json",
|
|
|
|
"Content-Type": "application/json",
|
|
|
|
},
|
|
|
|
"body": {body},
|
|
|
|
"method": "POST"
|
|
|
|
});
|
|
|
|
window._reader = response.body.pipeThrough(new TextDecoderStream()).getReader();
|
|
|
|
"""
|
|
|
|
driver.execute_script(
|
|
|
|
script.replace("{body}", json.dumps(json.dumps(data)))
|
|
|
|
)
|
|
|
|
# Read response
|
|
|
|
while True:
|
|
|
|
chunk = driver.execute_script("""
|
|
|
|
chunk = await window._reader.read();
|
|
|
|
if (chunk["done"]) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
content = "";
|
|
|
|
lines = chunk["value"].split("\\n")
|
|
|
|
lines.forEach((line, index) => {
|
|
|
|
if (line.startsWith('data: ')) {
|
|
|
|
content += line.substring('data: '.length);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
return content;
|
|
|
|
""")
|
|
|
|
if chunk:
|
|
|
|
yield chunk.replace("\\n", "\n")
|
|
|
|
elif chunk != "":
|
|
|
|
break
|
|
|
|
else:
|
2023-11-18 23:36:04 -05:00
|
|
|
time.sleep(0.1)
|