diff --git a/swarms/prompts/chat_prompt.py b/swarms/prompts/chat_prompt.py index 49a0aa23..4aca00d1 100644 --- a/swarms/prompts/chat_prompt.py +++ b/swarms/prompts/chat_prompt.py @@ -23,6 +23,14 @@ class Message: def get_type(self) -> str: pass +from enum import Enum + +class Role(Enum): + HUMAN = "Human" + AI = "AI" + SYSTEM = "System" + FUNCTIOn = "Function" + class HumanMessage(Message): """