Compliance fix: strictly use API_KEY and API_BASE_URL to avoid proxy bypass
Browse files- inference.py +6 -7
- 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.
|
| 28 |
-
API_BASE_URL = os.
|
| 29 |
-
MODEL_NAME = os.
|
| 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
|
| 286 |
-
|
| 287 |
-
|
| 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 |
-
"
|
| 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 |
}
|
|
|
|
| 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 |
}
|