diff options
author | Tekky <98614666+xtekky@users.noreply.github.com> | 2023-10-07 22:03:50 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-10-07 22:03:50 +0200 |
commit | 6334dac211898fb607d0281e9f23176172de4feb (patch) | |
tree | 45fad72c866cb03db744ae53e2993a4f31b29f0d /g4f/Provider/Aivvm.py | |
parent | ~ | g4f `v-0.1.5.4` [new gui & interference] (diff) | |
parent | Remove Aivvm from working providers (diff) | |
download | gpt4free-6334dac211898fb607d0281e9f23176172de4feb.tar gpt4free-6334dac211898fb607d0281e9f23176172de4feb.tar.gz gpt4free-6334dac211898fb607d0281e9f23176172de4feb.tar.bz2 gpt4free-6334dac211898fb607d0281e9f23176172de4feb.tar.lz gpt4free-6334dac211898fb607d0281e9f23176172de4feb.tar.xz gpt4free-6334dac211898fb607d0281e9f23176172de4feb.tar.zst gpt4free-6334dac211898fb607d0281e9f23176172de4feb.zip |
Diffstat (limited to 'g4f/Provider/Aivvm.py')
-rw-r--r-- | g4f/Provider/Aivvm.py | 68 |
1 files changed, 0 insertions, 68 deletions
diff --git a/g4f/Provider/Aivvm.py b/g4f/Provider/Aivvm.py deleted file mode 100644 index 1e780953..00000000 --- a/g4f/Provider/Aivvm.py +++ /dev/null @@ -1,68 +0,0 @@ -from __future__ import annotations - -from ..requests import StreamSession -from .base_provider import AsyncGeneratorProvider -from ..typing import AsyncGenerator - -# to recreate this easily, send a post request to https://chat.aivvm.com/api/models -models = { - 'gpt-3.5-turbo': {'id': 'gpt-3.5-turbo', 'name': 'GPT-3.5'}, - 'gpt-3.5-turbo-0613': {'id': 'gpt-3.5-turbo-0613', 'name': 'GPT-3.5-0613'}, - 'gpt-3.5-turbo-16k': {'id': 'gpt-3.5-turbo-16k', 'name': 'GPT-3.5-16K'}, - 'gpt-3.5-turbo-16k-0613': {'id': 'gpt-3.5-turbo-16k-0613', 'name': 'GPT-3.5-16K-0613'}, - 'gpt-4': {'id': 'gpt-4', 'name': 'GPT-4'}, - 'gpt-4-0613': {'id': 'gpt-4-0613', 'name': 'GPT-4-0613'}, - 'gpt-4-32k': {'id': 'gpt-4-32k', 'name': 'GPT-4-32K'}, - 'gpt-4-32k-0613': {'id': 'gpt-4-32k-0613', 'name': 'GPT-4-32K-0613'}, -} - -class Aivvm(AsyncGeneratorProvider): - url = 'https://chat.aivvm.com' - supports_stream = True - working = True - supports_gpt_35_turbo = True - supports_gpt_4 = True - - @classmethod - async def create_async_generator( - cls, - model: str, - messages: list[dict[str, str]], - stream: bool, - timeout: int = 30, - **kwargs - ) -> AsyncGenerator: - if not model: - model = "gpt-3.5-turbo" - elif model not in models: - raise ValueError(f"Model is not supported: {model}") - - json_data = { - "model" : models[model], - "messages" : messages, - "key" : "", - "prompt" : kwargs.get("system_message", "You are ChatGPT, a large language model trained by OpenAI. Follow the user's instructions carefully. Respond using markdown."), - "temperature" : kwargs.get("temperature", 0.7) - } - headers = { - "Accept": "*/*", - "Origin": cls.url, - "Referer": f"{cls.url}/", - } - async with StreamSession(impersonate="chrome107", headers=headers, timeout=timeout) as session: - async with session.post(f"{cls.url}/api/chat", json=json_data) as response: - response.raise_for_status() - async for chunk in response.iter_content(): - yield chunk.decode() - - @classmethod - @property - def params(cls): - params = [ - ('model', 'str'), - ('messages', 'list[dict[str, str]]'), - ('stream', 'bool'), - ('temperature', 'float'), - ] - param = ', '.join([': '.join(p) for p in params]) - return f'g4f.provider.{cls.__name__} supports: ({param})'
\ No newline at end of file |