From e0dc0ee1a52fcedc3be37f0bcc244385a1d2b734 Mon Sep 17 00:00:00 2001 From: Sashin Date: Sat, 4 Nov 2023 01:23:07 +0200 Subject: [PATCH] Fixed --- swarms/swarms/groupchat.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/swarms/swarms/groupchat.py b/swarms/swarms/groupchat.py index c0505fca..cb775351 100644 --- a/swarms/swarms/groupchat.py +++ b/swarms/swarms/groupchat.py @@ -87,7 +87,7 @@ class GroupChatManager: - def run_chat(self, task: str): + def __call__(self, task: str): self.groupchat.messages.append({'role':self.selector.name, 'content': task}) for i in range(self.groupchat.max_round): speaker = self.groupchat.select_speaker(last_speaker=self.selector, selector=self.selector) @@ -101,7 +101,7 @@ class GroupChatManager: llm = OpenAI( - openai_api_key="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", + openai_api_key="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", temperature=0.5, max_tokens=3000, ) @@ -140,6 +140,6 @@ manager = Flow( # Example usage: agents = [flow1, flow2, flow3] -group_chat = GroupChat(agents=agents, messages=[], max_round=5) +group_chat = GroupChat(agents=agents, messages=[], max_round=10) chat_manager = GroupChatManager(groupchat=group_chat, selector = manager) -chat_history = chat_manager.run_chat("Write me a riddle and answer it") \ No newline at end of file +chat_history = chat_manager("Write me a riddle") \ No newline at end of file