FILE_CONFIGS = {
    f"2wiki_enhanced_toolcall_method_baseline_using_google": {
        "path": "./data/parquet_folder/2wiki_data_for_enhanced_toolcall_method_basline.parquet",
        "runs": 2,
        "tools": ["parallel_search"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },
    f"2wiki_mindpattern_using_google": {
        "path": "./data/parquet_folder/2wiki_data_for_toolcall_driven_futuremind_method_basline",
        "runs": 2,
        "tools": ["parallel_search", "futuremind"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },

    f"musique_enhanced_toolcall_method_baseline_using_google": {
        "path": "./data/parquet_folder/musique_data_for_enhanced_toolcall_method_basline.parquet",
        "runs": 2,
        "tools": ["parallel_search"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },
    f"musique_mindpattern_using_google": {
        "path": "./data/parquet_folder/musique_data_for_toolcall_driven_futuremind_method_basline",
        "runs": 2,
        "tools": ["parallel_search", "futuremind"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },


    f"bamboogle_enhanced_toolcall_method_baseline_using_google": {
        "path": "./data/parquet_folder/bamboogle_data_for_enhanced_toolcall_method_basline.parquet",
        "runs": 2,
        "tools": ["parallel_search"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },
    f"bamboogle_mindpattern_using_google": {
        "path": "./data/parquet_folder/bamboogle_data_for_toolcall_driven_futuremind_method_basline",
        "runs": 2,
        "tools": ["parallel_search", "futuremind"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },

    f"frames_enhanced_toolcall_method_baseline_using_google": {
        "path": "./data/parquet_folder/frames_data_for_enhanced_toolcall_method_basline.parquet",
        "runs": 2,
        "tools": ["parallel_search"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },
    f"frames_mindpattern_using_google": {
        "path": "./data/parquet_folder/frames_data_for_toolcall_driven_futuremind_method_basline",
        "runs": 2,
        "tools": ["parallel_search", "futuremind"],
        "prompt_key": "prompt",
        "ground_truth_key": "reward_model.ground_truth"
    },
}
