Spaces:
Sleeping
Sleeping
Commit Β·
3db1e7f
1
Parent(s): 0ac8fe8
done
Browse files- 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:
|