Spaces:
Running
Running
Nitish commited on
Commit Β·
9e52d37
1
Parent(s): 309c392
fix(inference): handle empty env vars gracefully to prevent httpx crashes
Browse files- inference.py +4 -4
inference.py
CHANGED
|
@@ -22,10 +22,10 @@ from openai import OpenAI
|
|
| 22 |
load_dotenv()
|
| 23 |
|
| 24 |
# ββ Config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 25 |
-
API_BASE_URL = os.getenv("API_BASE_URL"
|
| 26 |
-
MODEL_NAME = os.getenv("MODEL_NAME"
|
| 27 |
-
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 28 |
-
ENV_URL = os.getenv("ENV_URL"
|
| 29 |
BENCHMARK = "code-security-review"
|
| 30 |
|
| 31 |
if not HF_TOKEN:
|
|
|
|
| 22 |
load_dotenv()
|
| 23 |
|
| 24 |
# ββ Config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 25 |
+
API_BASE_URL = os.getenv("API_BASE_URL") or "https://api.openai.com/v1"
|
| 26 |
+
MODEL_NAME = os.getenv("MODEL_NAME") or "gpt-4o-mini"
|
| 27 |
+
HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
|
| 28 |
+
ENV_URL = os.getenv("ENV_URL") or "http://localhost:7860"
|
| 29 |
BENCHMARK = "code-security-review"
|
| 30 |
|
| 31 |
if not HF_TOKEN:
|