Joash commited on
Commit
2130931
·
1 Parent(s): 9f46175

Remove hardcoded token and require it from environment variables

Browse files
Files changed (1) hide show
  1. src/config.py +3 -2
src/config.py CHANGED
@@ -26,7 +26,8 @@ class Config:
26
  MAX_OUTPUT_LENGTH = int(os.getenv("MAX_OUTPUT_LENGTH", 1024))
27
  TEMPERATURE = float(os.getenv("TEMPERATURE", 0.7))
28
  TOP_P = float(os.getenv("TOP_P", 0.95))
29
- HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN", "")
 
30
 
31
  # Database Settings
32
  DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./code_review.db")
@@ -59,7 +60,7 @@ class Config:
59
  logger.info(f"DEBUG: {Config.DEBUG}")
60
 
61
  if not Config.HUGGING_FACE_TOKEN:
62
- raise ValueError("HUGGING_FACE_TOKEN must be set to access Hugging Face gated models.")
63
  if Config.TEMPERATURE < 0 or Config.TEMPERATURE > 1:
64
  raise ValueError("TEMPERATURE must be between 0 and 1.")
65
  if Config.TOP_P < 0 or Config.TOP_P > 1:
 
26
  MAX_OUTPUT_LENGTH = int(os.getenv("MAX_OUTPUT_LENGTH", 1024))
27
  TEMPERATURE = float(os.getenv("TEMPERATURE", 0.7))
28
  TOP_P = float(os.getenv("TOP_P", 0.95))
29
+ # Token must be set in environment variables
30
+ HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
31
 
32
  # Database Settings
33
  DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./code_review.db")
 
60
  logger.info(f"DEBUG: {Config.DEBUG}")
61
 
62
  if not Config.HUGGING_FACE_TOKEN:
63
+ raise ValueError("HUGGING_FACE_TOKEN must be set in environment variables to access Hugging Face models.")
64
  if Config.TEMPERATURE < 0 or Config.TEMPERATURE > 1:
65
  raise ValueError("TEMPERATURE must be between 0 and 1.")
66
  if Config.TOP_P < 0 or Config.TOP_P > 1: