pt
/
gpt4free
mirror of https://github.com/xtekky/gpt4free.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added testing file and init in gpt4free

This commit is contained in:
ggindinson 2023-05-26 21:57:22 +03:00
parent 97346e6257
commit 223f09a4f1
2 changed files with 28 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
gpt4free/__init__.py
View File

@ -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))

6
testing/aicolors_test.py Normal file
View File

@ -0,0 +1,6 @@
from gpt4free import aicolors
prompt = "Light green color"
req = aicolors.Completion.create(prompt=prompt)
print(req)