64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
from enum import Enum
|
|
|
|
from gpt4free import cocalc
|
|
from gpt4free import forefront
|
|
from gpt4free import quora
|
|
from gpt4free import theb
|
|
from gpt4free import you
|
|
|
|
|
|
class Provider(Enum):
|
|
"""An enum representing different providers."""
|
|
|
|
You = 'you'
|
|
Poe = 'poe'
|
|
ForeFront = 'fore_front'
|
|
Theb = 'theb'
|
|
CoCalc = 'cocalc'
|
|
|
|
|
|
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.CoCalc:
|
|
return Completion.__cocalc_service(prompt, **kwargs)
|
|
else:
|
|
raise Exception('Provider not exist, Please try again')
|
|
|
|
@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 __cocalc_service(prompt: str, **kwargs):
|
|
return cocalc.Completion.create(prompt, cookie_input=kwargs.get('cookie_input', '')).text
|