summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Aivvm.py
diff options
context:
space:
mode:
authorHeiner Lohaus <heiner@lohaus.eu>2023-10-07 19:10:26 +0200
committerHeiner Lohaus <heiner@lohaus.eu>2023-10-07 19:10:26 +0200
commit3430b04f870d982d7fba34e3b9d6e5cf3bd3b847 (patch)
tree0756a28c14a8796c37dd8eee30539ef6aa388ddd /g4f/Provider/Aivvm.py
parentImprove code with ai (diff)
downloadgpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.tar
gpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.tar.gz
gpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.tar.bz2
gpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.tar.lz
gpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.tar.xz
gpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.tar.zst
gpt4free-3430b04f870d982d7fba34e3b9d6e5cf3bd3b847.zip
Diffstat (limited to 'g4f/Provider/Aivvm.py')
-rw-r--r--g4f/Provider/Aivvm.py68
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