From 5f28d67c320070e2c418a1854baedbd0addaab31 Mon Sep 17 00:00:00 2001
From: Sashin <sashinchetty8@gmail.com>
Date: Sat, 4 Nov 2023 02:05:57 +0200
Subject: [PATCH] Fixed

---
 playground/models/groupchat.py | 47 ++++++++++++++++++++++++++++++++++
 swarms/swarms/groupchat.py     | 44 -------------------------------
 2 files changed, 47 insertions(+), 44 deletions(-)
 create mode 100644 playground/models/groupchat.py

diff --git a/playground/models/groupchat.py b/playground/models/groupchat.py
new file mode 100644
index 00000000..ad663355
--- /dev/null
+++ b/playground/models/groupchat.py
@@ -0,0 +1,47 @@
+
+from swarms import OpenAI, Flow
+from swarms.swarms.groupchat import GroupChatManager, GroupChat
+
+llm = OpenAI(
+    openai_api_key="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",
+    temperature=0.5,
+    max_tokens=3000,
+)
+
+# Initialize the flow
+flow1 = Flow(
+    llm=llm,
+    max_loops=1,
+    system_message="YOU ARE SILLY, YOU OFFER NOTHING OF VALUE",
+    name='silly',
+    dashboard=True,
+)
+flow2 = Flow(
+    llm=llm,
+    max_loops=1,
+    system_message="YOU ARE VERY SMART AND ANSWER RIDDLES",
+    name='detective',
+    dashboard=True,
+)
+flow3 = Flow(
+    llm=llm,
+    max_loops=1,
+    system_message="YOU MAKE RIDDLES",
+    name='riddler',
+    dashboard=True,
+)
+manager = Flow(
+    llm=llm,
+    max_loops=1,
+    system_message="YOU ARE A GROUP CHAT MANAGER",
+    name='manager',
+    dashboard=True,
+)
+
+
+# Example usage:
+agents = [flow1, flow2, flow3]
+
+group_chat = GroupChat(agents=agents, messages=[], max_round=10)
+chat_manager = GroupChatManager(groupchat=group_chat, selector = manager)
+chat_history = chat_manager("Write me a riddle")
\ No newline at end of file
diff --git a/swarms/swarms/groupchat.py b/swarms/swarms/groupchat.py
index cb775351..0d9ffff3 100644
--- a/swarms/swarms/groupchat.py
+++ b/swarms/swarms/groupchat.py
@@ -99,47 +99,3 @@ class GroupChatManager:
 
         return reply
 
-
-llm = OpenAI(
-    openai_api_key="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",
-    temperature=0.5,
-    max_tokens=3000,
-)
-
-# Initialize the flow
-flow1 = Flow(
-    llm=llm,
-    max_loops=1,
-    system_message="YOU ARE SILLY, YOU OFFER NOTHING OF VALUE",
-    name='silly',
-    dashboard=True,
-)
-flow2 = Flow(
-    llm=llm,
-    max_loops=1,
-    system_message="YOU ARE VERY SMART AND ANSWER RIDDLES",
-    name='detective',
-    dashboard=True,
-)
-flow3 = Flow(
-    llm=llm,
-    max_loops=1,
-    system_message="YOU MAKE RIDDLES",
-    name='riddler',
-    dashboard=True,
-)
-manager = Flow(
-    llm=llm,
-    max_loops=1,
-    system_message="YOU ARE A GROUP CHAT MANAGER",
-    name='manager',
-    dashboard=True,
-)
-
-
-# Example usage:
-agents = [flow1, flow2, flow3]
-
-group_chat = GroupChat(agents=agents, messages=[], max_round=10)
-chat_manager = GroupChatManager(groupchat=group_chat, selector = manager)
-chat_history = chat_manager("Write me a riddle")
\ No newline at end of file