| """Post-process model output for GAIA exact-match submission.""" |
|
|
| import re |
| from typing import Any, Union |
|
|
|
|
| _FINAL_ANSWER_RE = re.compile( |
| r"^\s*(?:FINAL\s*ANSWER\s*[:：]?\s*)", |
| re.IGNORECASE, |
| ) |
|
|
|
|
| def normalize_answer(raw: Union[str, int, float, None]) -> Union[str, int, float]: |
| """ |
| Strip wrappers and forbidden prefixes. Prefer returning a string for API compatibility. |
| """ |
| if raw is None: |
| return "" |
| if isinstance(raw, (int, float)) and not isinstance(raw, bool): |
| return raw |
| text = str(raw).strip() |
| if not text: |
| return "" |
| text = _FINAL_ANSWER_RE.sub("", text, count=1).strip() |
| |
| for prefix in ("The answer is", "Answer:", "ANSWER:", "```", "`"): |
| if text.lower().startswith(prefix.lower()): |
| text = text[len(prefix) :].strip() |
| if text.startswith('"') and text.endswith('"') and len(text) >= 2: |
| text = text[1:-1].strip() |
| if text.startswith("```"): |
| text = re.sub(r"^```\w*\s*", "", text) |
| text = re.sub(r"\s*```$", "", text).strip() |
| return text.strip() |
|
|
|
|
| def maybe_numeric(text: str) -> Union[str, int, float]: |
| """If the prompt expects a plain number, allow int/float submission.""" |
| t = text.strip() |
| if re.fullmatch(r"-?\d+", t): |
| return int(t) |
| if re.fullmatch(r"-?\d+\.\d+", t): |
| return float(t) |
| return text |
|
|
