Spaces:
Running
Running
refactor: move PROVIDER_REGEX to GlobalConfig class
Browse filesCentralize the PROVIDER_REGEX pattern in the GlobalConfig class to improve code maintainability and reduce redundancy. This change ensures consistent usage across the application.
- app.py +1 -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',
|