| """trace_publish.py - upload /data/traces to the Hub as a dataset, using the |
| HF_TOKEN secret. No command line; the custom gradio.Server app calls it directly. |
| """ |
| from __future__ import annotations |
|
|
| import os |
| import tempfile |
|
|
| import paths |
|
|
|
|
| def publish_traces(repo_id: str) -> str: |
| repo_id = (repo_id or "").strip() |
| if "/" not in repo_id: |
| return "⚠️ Enter a repo id like `username/dataset-name`." |
| token = (os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN") |
| or "").strip() |
| if not token: |
| return ("⚠️ No `HF_TOKEN` secret found. Add it in Space → Settings → " |
| "Variables and secrets (a **write** token), then try again.") |
| try: |
| traces = [f for f in os.listdir(paths.TRACES_DIR) if f.endswith(".json")] |
| except OSError: |
| traces = [] |
| if not traces: |
| return "⚠️ No traces yet — run a few Auto Research reports first." |
| try: |
| from huggingface_hub import HfApi |
| api = HfApi(token=token) |
| api.create_repo(repo_id, repo_type="dataset", exist_ok=True) |
| card = ( |
| "---\nlicense: mit\ntags: [agent-trace, finance, chan-theory, llama-cpp]\n---\n\n" |
| "# Chan Compass — agent traces\n\n" |
| "JSON traces from the Chan Compass multi-agent research desk. Each file " |
| "is one ticker's run: the plan, every evidence-tool call and its result, " |
| "and each local sub-agent's request and response. Shared for the " |
| "Build Small hackathon (*Sharing is Caring*).\n\n" |
| "*Educational data — not investment advice.*\n") |
| rp = os.path.join(tempfile.gettempdir(), "README.md") |
| with open(rp, "w", encoding="utf-8") as f: |
| f.write(card) |
| api.upload_file(path_or_fileobj=rp, path_in_repo="README.md", |
| repo_id=repo_id, repo_type="dataset") |
| api.upload_folder(folder_path=paths.TRACES_DIR, path_in_repo="traces", |
| repo_id=repo_id, repo_type="dataset") |
| url = f"https://huggingface.co/datasets/{repo_id}" |
| return (f"✅ Published {len(traces)} trace(s) to {repo_id} ({url}). " |
| f"Put that link in your submission for the Sharing is Caring badge.") |
| except Exception as e: |
| return f"❌ Upload failed: {e}" |
|
|
