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',
|