Spaces:
Running
Running
File size: 584 Bytes
09321b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from ..agent_types import AgentType
from .messages_prompt import MessagesGenerator
from .mrkl_prompt import MrklPromptGenerator
from .ms_prompt import MSPromptGenerator
def get_prompt_generator(agent_type: AgentType = AgentType.DEFAULT, **kwargs):
if AgentType.DEFAULT == agent_type or agent_type == AgentType.MS_AGENT:
return MSPromptGenerator(**kwargs)
elif AgentType.MRKL == agent_type:
return MrklPromptGenerator(**kwargs)
elif AgentType.Messages == agent_type:
return MessagesGenerator(**kwargs)
else:
raise NotImplementedError
|