PROTHAM commited on
Commit
d3623ef
·
1 Parent(s): d09805b

Update inference.py environment variable parsing to match submission checklist

Browse files
Files changed (1) hide show
  1. 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") or "https://router.huggingface.co/v1"
10
- MODEL_NAME = os.getenv("MODEL_NAME") or "mistralai/Mistral-7B-Instruct-v0.3"
11
- API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
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=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: