Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5460fbd108 |
@@ -15,12 +15,13 @@ submission poller in a child process.
|
||||
|
||||
## Runtime Environment
|
||||
|
||||
The image includes a single-account submission token fallback for the agent
|
||||
platform. Environment variables can override it without rebuilding the image.
|
||||
The image includes single-account ModelHub and Hugging Face token fallbacks for
|
||||
the agent platform. Environment variables can override them without rebuilding
|
||||
the image.
|
||||
|
||||
- `MODELHUB_XC_TOKEN`, `XC_TOKEN`, or `MODELHUB_TOKEN` for ModelHub API authentication
|
||||
- `MODELHUB_JWT_TOKEN` or `JWT_TOKEN` can be used instead when the platform provides a JWT
|
||||
- `HF_TOKEN` optional, but recommended for Hugging Face API limits
|
||||
- `HF_TOKEN` optional override for the embedded Hugging Face fallback token
|
||||
- `STRATEGY_ID` is expected to be injected by the ModelHub agent platform and is attached to submissions for strategy attribution; it is not an API authentication token
|
||||
|
||||
Optional tuning:
|
||||
|
||||
4
main.py
4
main.py
@@ -9,7 +9,7 @@ import time
|
||||
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||||
from pathlib import Path
|
||||
|
||||
from modelhub_submmit_api.defaults import EMBEDDED_MODELHUB_XC_TOKEN
|
||||
from modelhub_submmit_api.defaults import EMBEDDED_HF_TOKEN, EMBEDDED_MODELHUB_XC_TOKEN
|
||||
|
||||
|
||||
HOST = "0.0.0.0"
|
||||
@@ -75,7 +75,7 @@ def _worker_command() -> list[str]:
|
||||
def _config() -> dict[str, object]:
|
||||
return {
|
||||
"strategy_id_present": bool(os.getenv("STRATEGY_ID")),
|
||||
"hf_token_present": bool(os.getenv("HF_TOKEN")),
|
||||
"hf_token_present": bool(os.getenv("HF_TOKEN") or EMBEDDED_HF_TOKEN),
|
||||
"modelhub_auth_present": _has_modelhub_auth(),
|
||||
"config_error": config_error,
|
||||
"worker_running": worker is not None and worker.poll() is None,
|
||||
|
||||
@@ -1 +1,2 @@
|
||||
EMBEDDED_MODELHUB_XC_TOKEN = "a14776f6e7ad4c04a1710260613c294c"
|
||||
EMBEDDED_HF_TOKEN = "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"
|
||||
|
||||
@@ -4,7 +4,7 @@ import argparse
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
from defaults import EMBEDDED_MODELHUB_XC_TOKEN
|
||||
from defaults import EMBEDDED_HF_TOKEN, EMBEDDED_MODELHUB_XC_TOKEN
|
||||
|
||||
|
||||
DEFAULT_KEY_PATH = Path("KEY.md")
|
||||
@@ -43,6 +43,8 @@ def ensure_tokens(args: argparse.Namespace) -> None:
|
||||
|
||||
if not getattr(args, "hf_token", None):
|
||||
args.hf_token = os.getenv("HF_TOKEN") or values.get("HF_TOKEN")
|
||||
if not getattr(args, "hf_token", None):
|
||||
args.hf_token = EMBEDDED_HF_TOKEN
|
||||
|
||||
modelhub_token = getattr(args, "modelhub_token", None)
|
||||
if not modelhub_token:
|
||||
|
||||
Reference in New Issue
Block a user