mirrors/mindcraft
1
0
Fork 0
mirror of https://github.com/kolbytn/mindcraft.git synced 2025-06-09 10:45:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix for crafting task analysis script for 3+ agent tasks

Browse source
This commit is contained in:
Ayush Maniar 2025-04-15 22:36:18 -07:00
parent a52905092d
commit c583e2d5e1
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
analyze_crafting_tasks.py
View file

@ -69,15 +69,16 @@ def analyze_json_file(file_path):
file_path (str): Path to the JSON file. file_path (str): Path to the JSON file.
Returns: Returns:
str or None: The task outcome string if found, otherwise None. bool: True if task was successful, False otherwise.
""" """
try: try:
with open(file_path, 'r') as f: with open(file_path, 'r') as f:
data = json.load(f) data = json.load(f)
if 'turns' in data and isinstance(data['turns'], list): if 'turns' in data and isinstance(data['turns'], list):
for turn in reversed(data['turns']): # Check turns from the end for turn in data['turns']: # Check all turns, not just from the end
if turn.get('role') == 'system' and isinstance(turn.get('content'), str): if turn.get('role') == 'system' and isinstance(turn.get('content'), str):
if "Task successful ended with code : 2" in turn['content'] or "Task ended with score : 1" in turn["content"] or "Task ended in score: 1" in turn["content"]: if "Task successful ended with code : 2" in turn['content'] or "Task ended with score : 1" in turn["content"] or "Task ended in score: 1" in turn["content"]:
# print(f"Success found in {file_path}")
return True return True
return False return False
except FileNotFoundError: except FileNotFoundError:
@ -93,18 +94,17 @@ def analyze_json_file(file_path):
def extract_result(folder_path): def extract_result(folder_path):
folder_name = os.path.basename(folder_path) folder_name = os.path.basename(folder_path)
json_files = glob.glob(os.path.join(folder_path, "*.json")) json_files = glob.glob(os.path.join(folder_path, "*.json"))
assert len(json_files) == 2, f"Expected 2 json files in {folder_name}, found {len(json_files)}"
if not json_files: if not json_files:
print(f"No JSON files found in {folder_name}") print(f"No JSON files found in {folder_name}")
return None return None
else: else:
outcome = False # Check each JSON file in the folder for success indication
for json_file in json_files: for json_file in json_files:
outcome = analyze_json_file(json_file) outcome = analyze_json_file(json_file)
if outcome: if outcome: # If any file indicates success, return True
return True return True
return False return False # Return False only if no files indicate success
def is_base(folder_path): def is_base(folder_path):
return "full_plan" in folder_path and "depth_0" in folder_path and "missing" not in folder_path return "full_plan" in folder_path and "depth_0" in folder_path and "missing" not in folder_path
@ -307,7 +307,7 @@ if __name__ == "__main__":
folders = download_s3_folders(args.aws_bucket_name, args.s3_folder_prefix, args.local_download_dir) folders = download_s3_folders(args.aws_bucket_name, args.s3_folder_prefix, args.local_download_dir)
else: else:
folders = get_immediate_subdirectories(args.local_download_dir) folders = get_immediate_subdirectories(args.local_download_dir)
print(folders) # print(folders)
results = aggregate_results(folders) results = aggregate_results(folders)
print(results) print(results)