summaryrefslogtreecommitdiffstats
path: root/gpt4free/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'gpt4free/__init__.py')
-rw-r--r--gpt4free/__init__.py8
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))