ochsncon commited on
Commit
b7dc0ee
·
verified ·
1 Parent(s): 14cc73a

Update src/config.py

Browse files
Files changed (1) hide show
  1. src/config.py +2 -2
src/config.py CHANGED
@@ -19,8 +19,8 @@ PRICE_MODEL_PATH = MODEL_DIR / "price_model.pkl"
19
  PREPROCESSOR_PATH = MODEL_DIR / "preprocessor.pkl"
20
  MODEL_METADATA_PATH = MODEL_DIR / "model_metadata.json"
21
 
22
- LLM_API_KEY = os.getenv("LLM_API_KEY", "")
23
- LLM_MODEL = os.getenv("LLM_MODEL", "gpt-4o-mini")
24
  LLM_API_BASE = os.getenv("LLM_API_BASE", "https://api.openai.com/v1/chat/completions")
25
  USE_LLM = os.getenv("USE_LLM", "false").strip().lower() in {"1", "true", "yes"}
26
  LLM_PROMPT_VERSION = os.getenv("LLM_PROMPT_VERSION", "structured").strip().lower()
 
19
  PREPROCESSOR_PATH = MODEL_DIR / "preprocessor.pkl"
20
  MODEL_METADATA_PATH = MODEL_DIR / "model_metadata.json"
21
 
22
+ LLM_API_KEY = os.getenv("LLM_API_KEY") or os.getenv("OPENAI_API_KEY") or ""
23
+ LLM_MODEL = os.getenv("LLM_MODEL") or os.getenv("OPENAI_MODEL") or "gpt-4o-mini"
24
  LLM_API_BASE = os.getenv("LLM_API_BASE", "https://api.openai.com/v1/chat/completions")
25
  USE_LLM = os.getenv("USE_LLM", "false").strip().lower() in {"1", "true", "yes"}
26
  LLM_PROMPT_VERSION = os.getenv("LLM_PROMPT_VERSION", "structured").strip().lower()