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
Files changed (1) hide show
  1. 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", "http://localhost:7860")
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
- print(
61
- f"[STEP] {json.dumps({'step': step, 'action': action, 'reward': reward, 'done': done, 'error': error})}",
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
  # ---------------------------------------------------------------------------