From a231c29c858f68eebb3429887c1ba6c8537b38fb Mon Sep 17 00:00:00 2001
From: Richard Anthony Hein <richard.hein@gmail.com>
Date: Mon, 12 Aug 2024 14:14:52 +0000
Subject: [PATCH] Added Role Enum

---
 swarms/prompts/chat_prompt.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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):
     """