Spaces:
Sleeping
Sleeping
Ajjack4 commited on
Commit ·
91a4057
1
Parent(s): 22ebfb6
fix log format to key=value, default ENV_BASE_URL to live HF Space
Browse files- inference.py +5 -13
inference.py
CHANGED
|
@@ -33,7 +33,7 @@ HF_TOKEN: str = (
|
|
| 33 |
or os.getenv("OPENAI_API_KEY")
|
| 34 |
or ""
|
| 35 |
)
|
| 36 |
-
ENV_BASE_URL: str = os.getenv("ENV_BASE_URL", "
|
| 37 |
|
| 38 |
if not HF_TOKEN:
|
| 39 |
print("[DEBUG] Warning: no API key found (HF_TOKEN / OPENAI_API_KEY). LLM calls will fail.", flush=True)
|
|
@@ -48,26 +48,18 @@ SUCCESS_SCORE_THRESHOLD = 0.5
|
|
| 48 |
# ---------------------------------------------------------------------------
|
| 49 |
|
| 50 |
def log_start(*, task: str, env: str, model: str) -> None:
|
| 51 |
-
print(
|
| 52 |
-
f"[START] {json.dumps({'task': task, 'env': env, 'model': model})}",
|
| 53 |
-
flush=True,
|
| 54 |
-
)
|
| 55 |
|
| 56 |
|
| 57 |
def log_step(
|
| 58 |
*, step: int, action: str, reward: float, done: bool, error: Optional[str]
|
| 59 |
) -> None:
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
flush=True,
|
| 63 |
-
)
|
| 64 |
|
| 65 |
|
| 66 |
def log_end(*, success: bool, steps: int, score: float, rewards: List[float]) -> None:
|
| 67 |
-
print(
|
| 68 |
-
f"[END] {json.dumps({'success': success, 'steps': steps, 'score': score, 'rewards': rewards})}",
|
| 69 |
-
flush=True,
|
| 70 |
-
)
|
| 71 |
|
| 72 |
|
| 73 |
# ---------------------------------------------------------------------------
|
|
|
|
| 33 |
or os.getenv("OPENAI_API_KEY")
|
| 34 |
or ""
|
| 35 |
)
|
| 36 |
+
ENV_BASE_URL: str = os.getenv("ENV_BASE_URL", "https://ajjack404-dev-env-debugger.hf.space")
|
| 37 |
|
| 38 |
if not HF_TOKEN:
|
| 39 |
print("[DEBUG] Warning: no API key found (HF_TOKEN / OPENAI_API_KEY). LLM calls will fail.", flush=True)
|
|
|
|
| 48 |
# ---------------------------------------------------------------------------
|
| 49 |
|
| 50 |
def log_start(*, task: str, env: str, model: str) -> None:
|
| 51 |
+
print(f"[START] task={task} env={env} model={model}", flush=True)
|
|
|
|
|
|
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
def log_step(
|
| 55 |
*, step: int, action: str, reward: float, done: bool, error: Optional[str]
|
| 56 |
) -> None:
|
| 57 |
+
error_str = error if error else "none"
|
| 58 |
+
print(f"[STEP] step={step} reward={reward} done={done} error={error_str}", flush=True)
|
|
|
|
|
|
|
| 59 |
|
| 60 |
|
| 61 |
def log_end(*, success: bool, steps: int, score: float, rewards: List[float]) -> None:
|
| 62 |
+
print(f"[END] success={success} steps={steps} score={score} rewards={rewards}", flush=True)
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
|
| 65 |
# ---------------------------------------------------------------------------
|