diff options
Diffstat (limited to 'gpt4free/__init__.py')
-rw-r--r-- | gpt4free/__init__.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/gpt4free/__init__.py b/gpt4free/__init__.py index 1e652897..739cf763 100644 --- a/gpt4free/__init__.py +++ b/gpt4free/__init__.py @@ -15,6 +15,7 @@ class Provider(Enum): ForeFront = 'fore_front' Theb = 'theb' UseLess = 'useless' + DeepAI = 'deepai' class Completion: @@ -40,6 +41,8 @@ class Completion: return Completion.__theb_service(prompt, **kwargs) elif provider == Provider.UseLess: return Completion.__useless_service(prompt, **kwargs) + elif provider == Provider.DeepAI: + return Completion.__deepai_service(prompt, **kwargs) else: raise Exception('Provider not exist, Please try again') @@ -62,4 +65,7 @@ class Completion: @staticmethod def __theb_service(prompt: str, **kwargs): return ''.join(theb.Completion.create(prompt=prompt)) - + + @staticmethod + def __deepai_service(prompt: str, **kwargs): + return ''.join(deepai.Completion.create(prompt=prompt)) |