Re-enable complex multi-team problem solving

This commit is contained in:
Mahesh Kommareddi 2024-07-17 03:54:24 -04:00
parent 93d244cb36
commit fcd29de36a
4 changed files with 19 additions and 19 deletions

View File

@ -134,24 +134,24 @@ public class IoASystem {
// Create all tasks // Create all tasks
List<Task> tasks = Arrays.asList( List<Task> tasks = Arrays.asList(
new Task("task1", "Plan a weekend trip to Paris", // new Task("task1", "Plan a weekend trip to Paris",
Arrays.asList("travel", "booking"), // Arrays.asList("travel", "booking"),
Arrays.asList("bookTravel", "findRestaurants", "getWeather")), // Arrays.asList("bookTravel", "findRestaurants", "getWeather")),
new Task("task2", "Organize a corporate team-building event in New York", // new Task("task2", "Organize a corporate team-building event in New York",
Arrays.asList("event planning", "team management"), // Arrays.asList("event planning", "team management"),
Arrays.asList("findRestaurants", "bookTravel", "scheduleAppointment")), // Arrays.asList("findRestaurants", "bookTravel", "scheduleAppointment")),
new Task("task3", "Develop a personalized fitness and nutrition plan", // new Task("task3", "Develop a personalized fitness and nutrition plan",
Arrays.asList("health", "nutrition"), // Arrays.asList("health", "nutrition"),
Arrays.asList("getWeather", "findFitnessClasses", "getRecipe")), // Arrays.asList("getWeather", "findFitnessClasses", "getRecipe")),
new Task("task4", "Research and summarize recent advancements in renewable energy", // new Task("task4", "Research and summarize recent advancements in renewable energy",
Arrays.asList("research", "writing"), // Arrays.asList("research", "writing"),
Arrays.asList("webSearch", "getNewsUpdates", "translate")), // Arrays.asList("webSearch", "getNewsUpdates", "translate")),
new Task("task5", "Plan and execute a social media marketing campaign for a new product launch", // new Task("task5", "Plan and execute a social media marketing campaign for a new product launch",
Arrays.asList("marketing", "social media"), // Arrays.asList("marketing", "social media"),
Arrays.asList("webSearch", "getNewsUpdates", "scheduleAppointment")), // Arrays.asList("webSearch", "getNewsUpdates", "scheduleAppointment")),
new Task("task6", "Assist in planning a multi-city European vacation for a family of four", // new Task("task6", "Assist in planning a multi-city European vacation for a family of four",
Arrays.asList("travel", "family planning"), // Arrays.asList("travel", "family planning"),
Arrays.asList("bookTravel", "calculateDistance", "getWeather", "findRestaurants")), // Arrays.asList("bookTravel", "calculateDistance", "getWeather", "findRestaurants")),
new Task("task7", "Organize an international tech conference with virtual and in-person components", new Task("task7", "Organize an international tech conference with virtual and in-person components",
Arrays.asList("event planning", "tech expertise", "marketing", "travel coordination", "content creation"), Arrays.asList("event planning", "tech expertise", "marketing", "travel coordination", "content creation"),

View File

@ -28,7 +28,7 @@ public class TeamFormation {
public List<AgentInfo> formTeam(Task task) { public List<AgentInfo> formTeam(Task task) {
List<String> requiredCapabilities = task.getRequiredCapabilities(); List<String> requiredCapabilities = task.getRequiredCapabilities();
List<String> requiredTools = task.getRequiredTools(); List<String> requiredTools = task.getRequiredTools();
List<AgentInfo> potentialAgents = agentRegistry.searchAgents(requiredCapabilities); List<AgentInfo> potentialAgents = agentRegistry.searchAgentsPartial(requiredCapabilities);
String teamFormationTask = "Form the best team for this task: " + task.getDescription() + String teamFormationTask = "Form the best team for this task: " + task.getDescription() +
"\nRequired capabilities: " + requiredCapabilities + "\nRequired capabilities: " + requiredCapabilities +