diff options
author | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-04 07:20:51 +0200 |
---|---|---|
committer | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-04 07:20:51 +0200 |
commit | 6a61cf811655fa87dbcb196025cc0b6040502293 (patch) | |
tree | 0e299a24d90095013854d04f9bf13617eebb8f6c /g4f/Provider/deprecated/Equing.py | |
parent | Use custom user dir (diff) | |
download | gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.tar gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.tar.gz gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.tar.bz2 gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.tar.lz gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.tar.xz gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.tar.zst gpt4free-6a61cf811655fa87dbcb196025cc0b6040502293.zip |
Diffstat (limited to 'g4f/Provider/deprecated/Equing.py')
-rw-r--r-- | g4f/Provider/deprecated/Equing.py | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/g4f/Provider/deprecated/Equing.py b/g4f/Provider/deprecated/Equing.py new file mode 100644 index 00000000..794274f2 --- /dev/null +++ b/g4f/Provider/deprecated/Equing.py @@ -0,0 +1,81 @@ +from __future__ import annotations + +import json +from abc import ABC, abstractmethod + +import requests + +from ...typing import Any, CreateResult +from ..base_provider import BaseProvider + + +class Equing(BaseProvider): + url: str = 'https://next.eqing.tech/' + working = False + supports_stream = True + supports_gpt_35_turbo = True + supports_gpt_4 = False + + @staticmethod + @abstractmethod + def create_completion( + model: str, + messages: list[dict[str, str]], + stream: bool, **kwargs: Any) -> CreateResult: + + headers = { + 'authority' : 'next.eqing.tech', + 'accept' : 'text/event-stream', + 'accept-language' : 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3', + 'cache-control' : 'no-cache', + 'content-type' : 'application/json', + 'origin' : 'https://next.eqing.tech', + 'plugins' : '0', + 'pragma' : 'no-cache', + 'referer' : 'https://next.eqing.tech/', + 'sec-ch-ua' : '"Not/A)Brand";v="99", "Google Chrome";v="115", "Chromium";v="115"', + 'sec-ch-ua-mobile' : '?0', + 'sec-ch-ua-platform': '"macOS"', + 'sec-fetch-dest' : 'empty', + 'sec-fetch-mode' : 'cors', + 'sec-fetch-site' : 'same-origin', + 'user-agent' : 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36', + 'usesearch' : 'false', + 'x-requested-with' : 'XMLHttpRequest' + } + + json_data = { + 'messages' : messages, + 'stream' : stream, + 'model' : model, + 'temperature' : kwargs.get('temperature', 0.5), + 'presence_penalty' : kwargs.get('presence_penalty', 0), + 'frequency_penalty' : kwargs.get('frequency_penalty', 0), + 'top_p' : kwargs.get('top_p', 1), + } + + response = requests.post('https://next.eqing.tech/api/openai/v1/chat/completions', + headers=headers, json=json_data, stream=stream) + + if not stream: + yield response.json()["choices"][0]["message"]["content"] + return + + for line in response.iter_content(chunk_size=1024): + if line: + if b'content' in line: + line_json = json.loads(line.decode('utf-8').split('data: ')[1]) + token = line_json['choices'][0]['delta'].get('content') + if token: + yield token + + @classmethod + @property + def params(cls): + params = [ + ("model", "str"), + ("messages", "list[dict[str, str]]"), + ("stream", "bool"), + ] + param = ", ".join([": ".join(p) for p in params]) + return f"g4f.provider.{cls.__name__} supports: ({param})"
\ No newline at end of file |