Spaces:
Sleeping
Sleeping
PROTHAM commited on
Commit ·
d3623ef
1
Parent(s): d09805b
Update inference.py environment variable parsing to match submission checklist
Browse files- inference.py +4 -4
inference.py
CHANGED
|
@@ -6,15 +6,15 @@ from openai import OpenAI
|
|
| 6 |
from client import APIDebugEnv
|
| 7 |
from models import APIAction
|
| 8 |
|
| 9 |
-
API_BASE_URL = os.getenv("API_BASE_URL"
|
| 10 |
-
MODEL_NAME = os.getenv("MODEL_NAME"
|
| 11 |
-
|
| 12 |
ENV_URL = os.getenv("ENV_URL", "http://localhost:7860")
|
| 13 |
BENCHMARK = "api_debug_env"
|
| 14 |
MAX_STEPS = 5
|
| 15 |
SUCCESS_SCORE_THRESHOLD = 0.8
|
| 16 |
|
| 17 |
-
llm = OpenAI(base_url=API_BASE_URL, api_key=
|
| 18 |
|
| 19 |
SYSTEM_PROMPT = """You are an HTTP API debugger. You receive a broken request and must fix it to get HTTP 200.
|
| 20 |
Respond ONLY in valid JSON with exactly these fields:
|
|
|
|
| 6 |
from client import APIDebugEnv
|
| 7 |
from models import APIAction
|
| 8 |
|
| 9 |
+
API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
|
| 10 |
+
MODEL_NAME = os.getenv("MODEL_NAME", "mistralai/Mistral-7B-Instruct-v0.3")
|
| 11 |
+
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 12 |
ENV_URL = os.getenv("ENV_URL", "http://localhost:7860")
|
| 13 |
BENCHMARK = "api_debug_env"
|
| 14 |
MAX_STEPS = 5
|
| 15 |
SUCCESS_SCORE_THRESHOLD = 0.8
|
| 16 |
|
| 17 |
+
llm = OpenAI(base_url=API_BASE_URL, api_key=HF_TOKEN)
|
| 18 |
|
| 19 |
SYSTEM_PROMPT = """You are an HTTP API debugger. You receive a broken request and must fix it to get HTTP 200.
|
| 20 |
Respond ONLY in valid JSON with exactly these fields:
|