rsrini7 commited on
Commit
6c5a6a1
·
1 Parent(s): b4ec8f7

refactor: move PROVIDER_REGEX to GlobalConfig class

Browse files

Centralize the PROVIDER_REGEX pattern in the GlobalConfig class to improve code maintainability and reduce redundancy. This change ensures consistent usage across the application.

Files changed (2) hide show
  1. app.py +1 -2
  2. global_config.py +1 -0
app.py CHANGED
@@ -30,7 +30,6 @@ load_dotenv()
30
 
31
  RUN_IN_OFFLINE_MODE = os.getenv('RUN_IN_OFFLINE_MODE', 'False').lower() == 'true'
32
 
33
- PROVIDER_REGEX = re.compile(r'\[(.*?)\]')
34
 
35
  @st.cache_data
36
  def _load_strings() -> dict:
@@ -183,7 +182,7 @@ with st.sidebar:
183
  ).split(' ')[0]
184
 
185
  # --- Automatically fetch API key from .env if available ---
186
- provider_match = PROVIDER_REGEX.match(llm_provider_to_use)
187
  selected_provider = provider_match.group(1) if provider_match else llm_provider_to_use
188
  env_key_name = GlobalConfig.PROVIDER_ENV_KEYS.get(selected_provider)
189
  default_api_key = os.getenv(env_key_name, "") if env_key_name else ""
 
30
 
31
  RUN_IN_OFFLINE_MODE = os.getenv('RUN_IN_OFFLINE_MODE', 'False').lower() == 'true'
32
 
 
33
 
34
  @st.cache_data
35
  def _load_strings() -> dict:
 
182
  ).split(' ')[0]
183
 
184
  # --- Automatically fetch API key from .env if available ---
185
+ provider_match = GlobalConfig.PROVIDER_REGEX.match(llm_provider_to_use)
186
  selected_provider = provider_match.group(1) if provider_match else llm_provider_to_use
187
  env_key_name = GlobalConfig.PROVIDER_ENV_KEYS.get(selected_provider)
188
  default_api_key = os.getenv(env_key_name, "") if env_key_name else ""
global_config.py CHANGED
@@ -41,6 +41,7 @@ class GlobalConfig:
41
  PROVIDER_OPENROUTER: "OPENROUTER_API_KEY",
42
  PROVIDER_TOGETHER_AI: "TOGETHER_API_KEY",
43
  }
 
44
  VALID_MODELS = {
45
  '[az]azure/open-ai': {
46
  'description': 'faster, detailed',
 
41
  PROVIDER_OPENROUTER: "OPENROUTER_API_KEY",
42
  PROVIDER_TOGETHER_AI: "TOGETHER_API_KEY",
43
  }
44
+ PROVIDER_REGEX = re.compile(r'\[(.*?)\]')
45
  VALID_MODELS = {
46
  '[az]azure/open-ai': {
47
  'description': 'faster, detailed',