Fixed

kyegomez · Nov 4, 2023 · 5f28d67 · 5f28d67
1 parent 73deed8
commit 5f28d67
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 44 deletions.
diff --git a/playground/models/groupchat.py 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")
diff --git a/swarms/swarms/groupchat.py b/swarms/swarms/groupchat.py
@@ -99,47 +99,3 @@ def __call__(self, task: str):
 
         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")