Added testing file and init in gpt4free
このコミットが含まれているのは:
コミット
223f09a4f1
|
@ -5,18 +5,20 @@ from gpt4free import quora
|
||||||
from gpt4free import theb
|
from gpt4free import theb
|
||||||
from gpt4free import usesless
|
from gpt4free import usesless
|
||||||
from gpt4free import you
|
from gpt4free import you
|
||||||
|
from gpt4free import aicolors
|
||||||
from gpt4free import deepai
|
from gpt4free import deepai
|
||||||
|
|
||||||
|
|
||||||
class Provider(Enum):
|
class Provider(Enum):
|
||||||
"""An enum representing different providers."""
|
"""An enum representing different providers."""
|
||||||
|
|
||||||
You = 'you'
|
You = "you"
|
||||||
Poe = 'poe'
|
Poe = "poe"
|
||||||
ForeFront = 'fore_front'
|
ForeFront = "fore_front"
|
||||||
Theb = 'theb'
|
Theb = "theb"
|
||||||
UseLess = 'useless'
|
UseLess = "useless"
|
||||||
DeepAI = 'deepai'
|
AiColors = "ai_colors"
|
||||||
|
DeepAI = "deepai"
|
||||||
|
|
||||||
|
|
||||||
class Completion:
|
class Completion:
|
||||||
|
@ -42,10 +44,16 @@ class Completion:
|
||||||
return Completion.__theb_service(prompt, **kwargs)
|
return Completion.__theb_service(prompt, **kwargs)
|
||||||
elif provider == Provider.UseLess:
|
elif provider == Provider.UseLess:
|
||||||
return Completion.__useless_service(prompt, **kwargs)
|
return Completion.__useless_service(prompt, **kwargs)
|
||||||
|
elif provider == Provider.AiColors:
|
||||||
|
return Completion.__ai_colors_service(prompt, **kwargs)
|
||||||
elif provider == Provider.DeepAI:
|
elif provider == Provider.DeepAI:
|
||||||
return Completion.__deepai_service(prompt, **kwargs)
|
return Completion.__deepai_service(prompt, **kwargs)
|
||||||
else:
|
else:
|
||||||
raise Exception('Provider not exist, Please try again')
|
raise Exception("Provider not exist, Please try again")
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def __ai_colors_service(prompt: str):
|
||||||
|
return aicolors.Completion.create(prompt=prompt)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def __useless_service(prompt: str, **kwargs) -> str:
|
def __useless_service(prompt: str, **kwargs) -> str:
|
||||||
|
@ -65,11 +73,12 @@ class Completion:
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def __theb_service(prompt: str, **kwargs):
|
def __theb_service(prompt: str, **kwargs):
|
||||||
return ''.join(theb.Completion.create(prompt=prompt))
|
return "".join(theb.Completion.create(prompt=prompt))
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def __deepai_service(prompt: str, **kwargs):
|
def __deepai_service(prompt: str, **kwargs):
|
||||||
return ''.join(deepai.Completion.create(prompt=prompt))
|
return "".join(deepai.Completion.create(prompt=prompt))
|
||||||
|
|
||||||
|
|
||||||
class ChatCompletion:
|
class ChatCompletion:
|
||||||
"""This class is used to execute a chat completion for a specified provider"""
|
"""This class is used to execute a chat completion for a specified provider"""
|
||||||
|
@ -87,8 +96,8 @@ class ChatCompletion:
|
||||||
if provider == Provider.DeepAI:
|
if provider == Provider.DeepAI:
|
||||||
return ChatCompletion.__deepai_service(messages, **kwargs)
|
return ChatCompletion.__deepai_service(messages, **kwargs)
|
||||||
else:
|
else:
|
||||||
raise Exception('Provider not exist, Please try again')
|
raise Exception("Provider not exist, Please try again")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def __deepai_service(messages: list, **kwargs):
|
def __deepai_service(messages: list, **kwargs):
|
||||||
return ''.join(deepai.ChatCompletion.create(messages=messages))
|
return "".join(deepai.ChatCompletion.create(messages=messages))
|
||||||
|
|
|
@ -0,0 +1,6 @@
|
||||||
|
from gpt4free import aicolors
|
||||||
|
|
||||||
|
prompt = "Light green color"
|
||||||
|
req = aicolors.Completion.create(prompt=prompt)
|
||||||
|
|
||||||
|
print(req)
|
読み込み中…
新しいイシューから参照