File size: 3,887 Bytes
f4b4235 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
from enum import Enum
from gpt4free import forefront
from gpt4free import quora
from gpt4free import theb
from gpt4free import usesless
from gpt4free import you
from gpt4free import aicolors
from gpt4free import deepai
class Provider(Enum):
"""An enum representing different providers."""
You = "you"
Poe = "poe"
ForeFront = "fore_front"
Theb = "theb"
UseLess = "useless"
AiColors = "ai_colors"
DeepAI = "deepai"
class Completion:
"""This class will be used for invoking the given provider"""
@staticmethod
def create(provider: Provider, prompt: str, **kwargs) -> str:
"""
Invokes the given provider with given prompt and addition arguments and returns the string response
:param provider: an enum representing the provider to use while invoking
:param prompt: input provided by the user
:param kwargs: Additional keyword arguments to pass to the provider while invoking
:return: A string representing the response from the provider
"""
if provider == Provider.Poe:
return Completion.__poe_service(prompt, **kwargs)
elif provider == Provider.You:
return Completion.__you_service(prompt, **kwargs)
elif provider == Provider.ForeFront:
return Completion.__fore_front_service(prompt, **kwargs)
elif provider == Provider.Theb:
return Completion.__theb_service(prompt, **kwargs)
elif provider == Provider.UseLess:
return Completion.__useless_service(prompt, **kwargs)
elif provider == Provider.AiColors:
return Completion.__ai_colors_service(prompt, **kwargs)
elif provider == Provider.DeepAI:
return Completion.__deepai_service(prompt, **kwargs)
else:
raise Exception("Provider not exist, Please try again")
@staticmethod
def __ai_colors_service(prompt: str):
return aicolors.Completion.create(prompt=prompt)
@staticmethod
def __useless_service(prompt: str, **kwargs) -> str:
return usesless.Completion.create(prompt=prompt, **kwargs)
@staticmethod
def __you_service(prompt: str, **kwargs) -> str:
return you.Completion.create(prompt, **kwargs).text
@staticmethod
def __poe_service(prompt: str, **kwargs) -> str:
return quora.Completion.create(prompt=prompt, **kwargs).text
@staticmethod
def __fore_front_service(prompt: str, **kwargs) -> str:
return forefront.Completion.create(prompt=prompt, **kwargs).text
@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))
class ChatCompletion:
"""This class is used to execute a chat completion for a specified provider"""
@staticmethod
def create(provider: Provider, messages: list, **kwargs) -> str:
"""
Invokes the given provider with given chat messages and addition arguments and returns the string response
:param provider: an enum representing the provider to use while invoking
:param messages: a list of chat messages, see the OpenAI docs for how to format this (https://platform.openai.com/docs/guides/chat/introduction)
:param kwargs: Additional keyword arguments to pass to the provider while invoking
:return: A string representing the response from the provider
"""
if provider == Provider.DeepAI:
return ChatCompletion.__deepai_service(messages, **kwargs)
else:
raise Exception("Provider not exist, Please try again")
@staticmethod
def __deepai_service(messages: list, **kwargs):
return "".join(deepai.ChatCompletion.create(messages=messages))
|