From dbe96b8083d68a3d078eb2fa47822a5ed8bfd8a8 Mon Sep 17 00:00:00 2001
From: Ayush Maniar <ayushmaniar70@gmail.com>
Date: Tue, 25 Mar 2025 11:05:56 -0700
Subject: [PATCH] Human names for 2+ agents, modifications to goal prompts for
 2+ agent tasks, and more milk_buckets in chest

---
 evaluation_script.py                  |  6 +++++-
 src/agent/task_types/cooking_tasks.js | 12 ++++++------
 src/agent/tasks.js                    | 14 +++++++++++++-
 3 files changed, 24 insertions(+), 8 deletions(-)

diff --git a/evaluation_script.py b/evaluation_script.py
index 7428fea..4da8b59 100644
--- a/evaluation_script.py
+++ b/evaluation_script.py
@@ -331,11 +331,15 @@ def launch_server_experiment(task_path,
         models = [model] * 2
         apis = [api] * 2
     else:
+        # Lets use an ordered list of 10 human names.
+        human_names = ["Andy", "Jill", "Bob", "Sally", "Mike", "Laura", "John", "Emma", "Tom", "Kate"]
         agent_names = []
         for i in range(num_agents):
-            agent_names.append(f"Agent_{i}_{session_name}")
+            name = human_names[i % len(human_names)]
+            agent_names.append(f"{name}_{session_name}")
         models = [model] * num_agents
         apis = [api] * num_agents
+        
     make_profiles(agent_names, models, apis, template_profile=template_profile, url=url)
 
     agent_profiles = [f"./{agent}.json" for agent in agent_names]
diff --git a/src/agent/task_types/cooking_tasks.js b/src/agent/task_types/cooking_tasks.js
index dcb98b4..5785c0b 100644
--- a/src/agent/task_types/cooking_tasks.js
+++ b/src/agent/task_types/cooking_tasks.js
@@ -318,8 +318,8 @@ export class CookingTaskInitiator {
                     ['minecraft:sugar', 64],
                     ['minecraft:cocoa_beans', 64],
                     ['minecraft:apple', 64],
-                    ['minecraft:cookie', 64],
-                    ['minecraft:mutton', 64],
+                    ['minecraft:milk_bucket', 1],
+                    ['minecraft:milk_bucket', 1],
                     ['minecraft:salmon', 64],
                     ['minecraft:cod', 64],
                     ['minecraft:kelp', 64],
@@ -328,10 +328,10 @@ export class CookingTaskInitiator {
                     ['minecraft:honey_bottle', 1],     // Non-stackable
                     ['minecraft:glow_berries', 64],
                     ['minecraft:bowl', 64],
-                    ['minecraft:golden_carrot', 64],
-                    ['minecraft:golden_apple', 64],
-                    ['minecraft:enchanted_golden_apple', 64],
-                    ['minecraft:cooked_mutton', 64],
+                    ['minecraft:milk_bucket', 1],
+                    ['minecraft:milk_bucket', 1],
+                    ['minecraft:milk_bucket', 1],
+                    ['minecraft:milk_bucket', 1],
                     ['minecraft:cooked_salmon', 64],
                     ['minecraft:cooked_cod', 64],
                     ['minecraft:gold_ingot', 64],
diff --git a/src/agent/tasks.js b/src/agent/tasks.js
index 076b81a..8b674a3 100644
--- a/src/agent/tasks.js
+++ b/src/agent/tasks.js
@@ -203,7 +203,19 @@ export class Task {
         let add_string = '';
 
         if (this.task_type === 'cooking') {
-            add_string = '\nIn the end, all the food items should be given to one single bot.';
+
+            if (this.data.agent_count > 2) {
+
+                if (this.name.toLowerCase().startsWith('andy')) {
+                    add_string = '\nIn the end, all the food items should be given to you by other bots. Make sure to talk to all the agents using startConversation command to coordinate the task instead of talking to just one agent. You can even end current conversation with any agent using endConversation command and then talk to a new agent using startConversation command.';
+                } 
+                else {
+                    add_string = '\nIn the end, all the food items should be given to one single bot whose name starts with andy or Andy. Make sure to talk to all the agents using startConversation command to coordinate the task instead of talking to just one agent. You can even end current conversation with any agent using endConversation command and then talk to a new agent using startConversation command.';
+                }   
+            } 
+            else {
+                add_string = '\nIn the end, all the food items should be given to one single bot.';
+            }
         }
 
         // If goal is a string, all agents share the same goal