junaid0600 commited on
Commit
3db1e7f
Β·
1 Parent(s): 0ac8fe8
Files changed (1) hide show
  1. inference.py +4 -0
inference.py CHANGED
@@ -14,6 +14,7 @@ from dotenv import load_dotenv
14
  load_dotenv()
15
 
16
  from env.environment import SQLDebuggerEnvironment
 
17
  # ─────────────────────────────────────────────
18
  # ENVIRONMENT VARIABLES
19
  # ─────────────────────────────────────────────
@@ -25,6 +26,9 @@ if HF_TOKEN is None:
25
  raise ValueError("HF_TOKEN environment variable is required")
26
 
27
  API_KEY = HF_TOKEN
 
 
 
28
  # ─────────────────────────────────────────────
29
 
30
  def log_start(task: str, env: str, model: str) -> None:
 
14
  load_dotenv()
15
 
16
  from env.environment import SQLDebuggerEnvironment
17
+ from env.models import Action, ActionType, DifficultyLevel
18
  # ─────────────────────────────────────────────
19
  # ENVIRONMENT VARIABLES
20
  # ─────────────────────────────────────────────
 
26
  raise ValueError("HF_TOKEN environment variable is required")
27
 
28
  API_KEY = HF_TOKEN
29
+ BENCHMARK = "sql-query-debugger"
30
+ MAX_STEPS = 10
31
+ SUCCESS_SCORE_THRESHOLD = 0.5
32
  # ─────────────────────────────────────────────
33
 
34
  def log_start(task: str, env: str, model: str) -> None: