BiGuan/Agent

1"""2answer_key.py —— "标准答案对照表" + "题目自动分类器"（仅供我们自己调试看结果用）3 41. REFERENCE_ANSWERS：一张"标准答案表"。GAIA 这 20 道题的官方正确答案是公开的（来自数据集里5    的 metadata.jsonl 文件），我们把它抄在这里。6    注意：这张表【不参与答题】！agent 答题时根本不会看它。它唯一的用途是：在结果表格里，把7    "正确答案"和"我们 agent 提交的答案"并排显示出来，方便我们一眼看出哪几道答错了。8    （因为评分服务器只告诉我们总分，不会告诉我们具体哪题对哪题错。）9 102. classify_question：根据题目内容/附件类型，自动给每道题贴一个"类型标签"（比如"看图题""音频题"11    "网页检索题"），同样是为了让结果表格更直观。12"""13 14import re15 16# 一个"字典"（dict）：左边是每道题的唯一编号 task_id，右边是这道题的官方标准答案。17REFERENCE_ANSWERS = {18    "8e867cd7-cff9-4e6c-867a-ff5ddc2550be": "3",19    "a1e91b78-d3d8-4675-bb8d-62741b4b68a6": "3",20    "2d83110e-a098-4ebb-9987-066c06fa42d0": "Right",21    "cca530fc-4052-43b2-b130-b30968d8aa44": "Rd5",22    "4fc2f1ae-8625-45b5-ab34-ad4433bc21f8": "FunkMonk",23    "6f37996b-2ac7-44b0-8e68-6d28256631b4": "b, e",24    "9d191bce-651d-4746-be2d-7ef8ecadb9c2": "Extremely",25    "cabe07ed-9eca-40ea-8ead-410ef5e83f91": "Louvrier",26    "3cef3a44-215e-4aed-8e3b-b1e3f08063b7": "broccoli, celery, fresh basil, lettuce, sweet potatoes",27    "99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3": "cornstarch, freshly squeezed lemon juice, granulated sugar, pure vanilla extract, ripe strawberries",28    "305ac316-eef6-4446-960a-92d80d542f82": "Wojciech",29    "f918266a-b3e0-4914-865d-4faa564f1aef": "0",30    "3f57289b-8c60-48be-bd80-01f8099ca449": "519",31    "1f975693-876d-457b-a649-393859e79bf3": "132, 133, 134, 197, 245",32    "840bfca7-4f7b-481a-8794-c560c340185d": "80GSFC21M0002",33    "bda648d7-d618-4883-88f4-3466eabd860e": "Saint Petersburg",34    "cf106601-ab4f-4af9-b045-5295fe67b37d": "CUB",35    "a0c07678-e491-4bbc-8f0b-07405144218f": "Yoshida, Uehara",36    "7bd855d8-463d-4ed5-93ca-5fe35145f733": "89706.00",37    "5a0c1adf-205e-4841-a666-7c3ef95def9d": "Claus",38}39 40 41def classify_question(question: str, file_name: str = "") -> str:42    """根据题目和附件名，猜测这道题属于哪种类型，返回一个标签字符串。"""43    # 先看有没有附件、附件是什么后缀名（.xlsx/.mp3/.png...），据此判断题型。44    # .lower() 把文件名转成小写，避免大小写不一致导致判断失败（.MP3 和 .mp3 应视为同一种）。45    fn = (file_name or "").lower()46    if fn.endswith((".xlsx", ".xls", ".csv", ".tsv")):   # 表格类附件47        return "file: spreadsheet"48    if fn.endswith((".mp3", ".wav", ".m4a", ".flac", ".ogg")):  # 音频类附件49        return "file: audio"50    if fn.endswith((".png", ".jpg", ".jpeg", ".gif", ".bmp", ".webp")):  # 图片类附件51        return "file: image"52    if fn.endswith((".py", ".txt", ".json", ".c", ".cpp", ".java", ".js")):  # 代码/纯文本类附件53        return "file: code/text"54    if fn.endswith((".pdf", ".docx")):   # 文档类附件55        return "file: document"56    if fn:   # 有附件但后缀不在上面列举的范围内57        return "file: other"58 59    # 没有附件时，转而分析题目文字本身。60    q = question.lower()61    if "youtube.com" in q or "youtu.be" in q:   # 题目里出现 YouTube 链接 → 视频题62        return "video (YouTube)"63    # 有的题目把整句话倒着写（比如把 "the" 写成 "eht"），用正则查这种倒写的标志词 → 文字处理题。64    if re.search(r"\beht\b", question) or re.search(r"\brewsna\b", q):65        return "text manipulation"66    # 出现表格符号 "|*|" 或"交换律 commutative"等字样 → 逻辑/表格运算题。67    if "|*|" in question or ("commutative" in q) or ("set s" in q):68        return "logic/table"69    if "wikipedia" in q:   # 明确提到维基百科70        return "web: wikipedia"71    return "web research"   # 以上都不是 → 默认归为"上网检索题"72