| from autogen.agentchat import ( | |
| Agent, | |
| GroupChat, | |
| GroupChatManager, | |
| AssistantAgent, | |
| UserProxyAgent, | |
| ) | |
| import openai | |
| from typing import Dict, List, Optional, Union | |
| import os | |
| from dotenv import load_dotenv | |
| from HumanResources import HumanResources | |
| from AgentSelector import AgentSelector | |
| from AgentSpawner import AgentSpawner | |
| from GroupChatSpawner import GroupChatSpawner | |
| openai.api_key = os.getenv("OPENAI_API_KEY") | |
| CHAT_INITIATOR = "finance" | |
| hr = HumanResources() | |
| all_available_agents = hr.select_agents() | |
| print(all_available_agents) | |
| task = "revenue in the east coast is falling and the competitor is doing great with their product." | |
| selector = AgentSelector(task=task, available_agents=all_available_agents, n_agents=3) | |
| selected_agents = selector.run_selection() | |
| print("*" * 10, selected_agents) | |
| selected_agents = ["sales", "marketing", "engineer"] | |
| agent_spawner = AgentSpawner(selected_agents, CHAT_INITIATOR) | |
| agents = agent_spawner.spawn() | |
| print(agents) | |
| messages = [] | |
| llm_config = {"config_list": [{"model": "gpt-4"}], "seed": 42, "request_timeout": 600} | |
| groupchat = GroupChatSpawner( | |
| agents=agents, llm_config=llm_config, messages=messages, max_round=10 | |
| ) | |
| chat_initiator.initiate_chat( | |
| manager, | |
| message="I want to design an app to make me one million dollars in one month. Yes, your heard that right.", | |
| ) | |