diff options
| author | Tekky <98614666+xtekky@users.noreply.github.com> | 2023-09-23 17:32:21 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-23 17:32:21 +0200 |
| commit | eaa8f712abd6d2ae181cf3f1524239658d7da5a3 (patch) | |
| tree | 107121cfa813bcdb1d5f0d86216ef74561306b9a /g4f/Provider | |
| parent | ~ | improve Vercel & g4f.Completion.create (diff) | |
| parent | TypeError: issubclass() arg 1 must be a class (diff) | |
| download | gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.tar gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.tar.gz gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.tar.bz2 gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.tar.lz gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.tar.xz gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.tar.zst gpt4free-eaa8f712abd6d2ae181cf3f1524239658d7da5a3.zip | |
Diffstat (limited to 'g4f/Provider')
| -rw-r--r-- | g4f/Provider/Aivvm.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/g4f/Provider/Aivvm.py b/g4f/Provider/Aivvm.py index 97c9dff0..c38c4a74 100644 --- a/g4f/Provider/Aivvm.py +++ b/g4f/Provider/Aivvm.py @@ -2,7 +2,7 @@ from __future__ import annotations import requests from .base_provider import BaseProvider -from ..typing import Any, CreateResult +from ..typing import CreateResult models = { 'gpt-3.5-turbo': {'id': 'gpt-3.5-turbo', 'name': 'GPT-3.5'}, @@ -26,8 +26,14 @@ class Aivvm(BaseProvider): def create_completion(cls, model: str, messages: list[dict[str, str]], - stream: bool, **kwargs: Any) -> CreateResult: - + stream: bool, + **kwargs + ) -> CreateResult: + if not model: + model = "gpt-3.5-turbo" + elif model not in models: + raise ValueError(f"Model are not supported: {model}") + headers = { "authority" : "chat.aivvm.com", "accept" : "*/*", |