Somuai12 commited on
Commit
09a9c72
Β·
1 Parent(s): 5abef36

Compliance fix: strictly use API_KEY and API_BASE_URL to avoid proxy bypass

Browse files
Files changed (2) hide show
  1. inference.py +6 -7
  2. server/app.py +2 -2
inference.py CHANGED
@@ -24,9 +24,9 @@ from models import Action
24
 
25
  # ─── Environment Variables (Hackathon Mandatory) ───
26
  IMAGE_NAME = os.getenv("IMAGE_NAME")
27
- API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
28
- API_BASE_URL = os.getenv("API_BASE_URL") or "https://api.groq.com/openai/v1"
29
- MODEL_NAME = os.getenv("MODEL_NAME") or "llama-3.1-8b-instant"
30
  BENCHMARK = "policy_evolver_env"
31
  MAX_STEPS = 5
32
  TEMPERATURE = 0.0
@@ -282,10 +282,9 @@ async def main() -> None:
282
  try:
283
  # 1. Initialize OpenAI Client
284
  try:
285
- if not API_KEY or API_KEY == "dummy_key":
286
- # In validator, API_KEY might be missing if they didn't pass it yet, but we shouldn't crash
287
- pass
288
- client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY or "dummy")
289
  except Exception as e:
290
  setup_error = Exception(f"OpenAI client initialization failed: {e}")
291
 
 
24
 
25
  # ─── Environment Variables (Hackathon Mandatory) ───
26
  IMAGE_NAME = os.getenv("IMAGE_NAME")
27
+ API_KEY = os.environ.get("API_KEY")
28
+ API_BASE_URL = os.environ.get("API_BASE_URL")
29
+ MODEL_NAME = os.environ.get("MODEL_NAME")
30
  BENCHMARK = "policy_evolver_env"
31
  MAX_STEPS = 5
32
  TEMPERATURE = 0.0
 
282
  try:
283
  # 1. Initialize OpenAI Client
284
  try:
285
+ if not API_KEY or not API_BASE_URL:
286
+ raise Exception("Missing mandatory environment variables: API_KEY and/or API_BASE_URL")
287
+ client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
 
288
  except Exception as e:
289
  setup_error = Exception(f"OpenAI client initialization failed: {e}")
290
 
server/app.py CHANGED
@@ -103,8 +103,8 @@ def run_baseline_route():
103
  try:
104
  env_vars = {
105
  **os.environ,
106
- "HF_TOKEN": os.environ.get("HF_TOKEN", ""),
107
- "API_BASE_URL": os.environ.get("API_BASE_URL", "https://api.groq.com/openai/v1"),
108
  "MODEL_NAME": os.environ.get("MODEL_NAME", "llama-3.1-8b-instant"),
109
  "ENV_BASE_URL": "http://127.0.0.1:7860",
110
  }
 
103
  try:
104
  env_vars = {
105
  **os.environ,
106
+ "API_KEY": os.environ.get("API_KEY", ""),
107
+ "API_BASE_URL": os.environ.get("API_BASE_URL", ""),
108
  "MODEL_NAME": os.environ.get("MODEL_NAME", "llama-3.1-8b-instant"),
109
  "ENV_BASE_URL": "http://127.0.0.1:7860",
110
  }