返回提交历史
Modified
gpt4free/__init__.py
+22
-13
Added
testing/aicolors_test.py
+6
-0
XFEstudio/gpt4free
Added testing file and init in gpt4free
223f09a4
代码差异
2 个文件
+28
-13
@@ -5,18 +5,20 @@ from gpt4free import quora
5
5
from gpt4free import theb
6
6
from gpt4free import usesless
7
7
from gpt4free import you
8
from gpt4free import aicolors
8
9
from gpt4free import deepai
9
10
10
11
11
12
class Provider(Enum):
12
13
"""An enum representing different providers."""
13
14
14
You = 'you'
15
Poe = 'poe'
16
ForeFront = 'fore_front'
17
Theb = 'theb'
18
UseLess = 'useless'
19
DeepAI = 'deepai'
15
You = "you"
16
Poe = "poe"
17
ForeFront = "fore_front"
18
Theb = "theb"
19
UseLess = "useless"
20
AiColors = "ai_colors"
21
DeepAI = "deepai"
20
22
21
23
22
24
class Completion:
@@ -42,10 +44,16 @@ class Completion:
42
44
return Completion.__theb_service(prompt, **kwargs)
43
45
elif provider == Provider.UseLess:
44
46
return Completion.__useless_service(prompt, **kwargs)
47
elif provider == Provider.AiColors:
48
return Completion.__ai_colors_service(prompt, **kwargs)
45
49
elif provider == Provider.DeepAI:
46
50
return Completion.__deepai_service(prompt, **kwargs)
47
51
else:
48
raise Exception('Provider not exist, Please try again')
52
raise Exception("Provider not exist, Please try again")
53
54
@staticmethod
55
def __ai_colors_service(prompt: str):
56
return aicolors.Completion.create(prompt=prompt)
49
57
50
58
@staticmethod
51
59
def __useless_service(prompt: str, **kwargs) -> str:
@@ -65,11 +73,12 @@ class Completion:
65
73
66
74
@staticmethod
67
75
def __theb_service(prompt: str, **kwargs):
68
return ''.join(theb.Completion.create(prompt=prompt))
69
76
return "".join(theb.Completion.create(prompt=prompt))
77
70
78
@staticmethod
71
79
def __deepai_service(prompt: str, **kwargs):
72
return ''.join(deepai.Completion.create(prompt=prompt))
80
return "".join(deepai.Completion.create(prompt=prompt))
81
73
82
74
83
class ChatCompletion:
75
84
"""This class is used to execute a chat completion for a specified provider"""
@@ -87,8 +96,8 @@ class ChatCompletion:
87
96
if provider == Provider.DeepAI:
88
97
return ChatCompletion.__deepai_service(messages, **kwargs)
89
98
else:
90
raise Exception('Provider not exist, Please try again')
91
99
raise Exception("Provider not exist, Please try again")
100
92
101
@staticmethod
93
102
def __deepai_service(messages: list, **kwargs):
94
return ''.join(deepai.ChatCompletion.create(messages=messages))
103
return "".join(deepai.ChatCompletion.create(messages=messages))
@@ -0,0 +1,6 @@
1
from gpt4free import aicolors
2
3
prompt = "Light green color"
4
req = aicolors.Completion.create(prompt=prompt)
5
6
print(req)