| """Resolve the official submission's agent_code value for Space or local runs.""" |
|
|
| from __future__ import annotations |
|
|
| from pathlib import Path |
| from urllib.parse import urlparse |
|
|
| SOURCE_FILES = ( |
| "agent.py", |
| "answer_formatter.py", |
| "cache.py", |
| "config.py", |
| "evaluation.py", |
| "gaia_client.py", |
| "router.py", |
| "tools/__init__.py", |
| "tools/agent_tools.py", |
| "tools/audio.py", |
| "tools/image_chess.py", |
| "tools/markdown_logic.py", |
| "tools/python_exec.py", |
| "tools/retry.py", |
| "tools/spreadsheet.py", |
| "tools/text_transform.py", |
| "tools/video.py", |
| "tools/web.py", |
| "tools/youtube.py", |
| ) |
|
|
|
|
| def _public_url(value: str) -> str: |
| url = value.strip() |
| parsed = urlparse(url) |
| if parsed.scheme not in {"http", "https"} or not parsed.netloc: |
| raise ValueError("GAIA_AGENT_CODE_URL must be an absolute HTTP(S) URL") |
| return url |
|
|
|
|
| def build_inline_agent_code(root: Path) -> str: |
| """Bundle implementation sources without reading secrets, caches, or results.""" |
| sections: list[str] = [] |
| for relative in SOURCE_FILES: |
| path = (root / relative).resolve() |
| if not path.is_relative_to(root.resolve()) or not path.is_file(): |
| raise FileNotFoundError(f"Required source file is missing: {relative}") |
| sections.append( |
| f"# ===== {relative} =====\n{path.read_text(encoding='utf-8').rstrip()}" |
| ) |
| return "\n\n".join(sections) |
|
|
|
|
| def resolve_agent_code( |
| *, |
| space_id: str | None, |
| configured_url: str | None, |
| allow_inline: bool, |
| root: Path, |
| ) -> str: |
| """Prefer a public code URL; permit explicit inline source for local-only use.""" |
| if configured_url and configured_url.strip(): |
| return _public_url(configured_url) |
| if space_id and space_id.strip(): |
| return f"https://huggingface.co/spaces/{space_id.strip()}/tree/main" |
| if allow_inline: |
| return build_inline_agent_code(root) |
| raise RuntimeError( |
| "Local submission needs GAIA_AGENT_CODE_URL pointing to a public code repository, " |
| "or explicit GAIA_ALLOW_INLINE_AGENT_CODE=1 to send the local source bundle." |
| ) |
|
|
