Update query_solver.py

query_solver.py

@@ -14,23 +14,30 @@ def run_query(math_problem, api_key):
 
     autogen.ChatCompletion.start_logging()
 
-    assistant = autogen.AssistantAgent(
-        name="assistant",
-        system_message="You are a helpful assistant.",
-        llm_config={
-            "request_timeout": 600,
-            "seed": 42,
-            "config_list": config_list,
-        }
+    # Create user proxy agent, coder, product manager
+    user_proxy = autogen.UserProxyAgent(
+        name="User_proxy",
+        system_message="A human admin who will give the idea and run the code provided by Coder.",
+        code_execution_config={"last_n_messages": 2, "work_dir": "groupchat"},
+        human_input_mode="ALWAYS",
     )
-
-    mathproxyagent = MathUserProxyAgent(
-        name="mathproxyagent",
-        human_input_mode="NEVER",
-        code_execution_config={"use_docker": False},
+    coder = autogen.AssistantAgent(
+        name="Coder",
+        llm_config=llm_config,
+    )
+    pm = autogen.AssistantAgent(
+        name="product_manager",
+        system_message="You will help break down the initial idea into a well scoped requirement for the coder; Do not involve in future conversations or error fixing",
+        llm_config=llm_config,
     )
 
-    return mathproxyagent.initiate_chat(assistant, problem=math_problem)
+    # Create groupchat
+    groupchat = autogen.GroupChat(
+        agents=[user_proxy, coder, pm], messages=[])
+    manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=llm_config)
+
+
+    return user_proxy.initiate_chat(manager, message=math_problem)
    
 
 if __name__ == "__main__":