Update and format for review commits

Signed-off-by: Sairam Pillai <[email protected]>

app.py

@@ -195,31 +195,23 @@ with st.sidebar:
         api_version: str = ''
     else:
         # The online LLMs
-        selected_option = st.sidebar.selectbox(
+        llm_provider_to_use = st.sidebar.selectbox(
             label='2: Select a suitable LLM to use:\n\n(Gemini and Mistral-Nemo are recommended)',
             options=[f'{k} ({v["description"]})' for k, v in GlobalConfig.VALID_MODELS.items()],
             index=GlobalConfig.DEFAULT_MODEL_INDEX,
             help=GlobalConfig.LLM_PROVIDER_HELP,
             on_change=reset_api_key
-        )
+        ).split(' ')[0]
         
-        # Extract provider key more robustly using regex
-        provider_match = GlobalConfig.PROVIDER_REGEX.match(selected_option)
-        if provider_match:
-            llm_provider_to_use = selected_option  # Use full string for get_provider_model
-        else:
-            # Fallback: try to extract the key before the first space
-            llm_provider_to_use = selected_option.split(' ')[0]
-            logger.warning(f"Could not parse provider from selectbox option: {selected_option}")
-
         # --- Automatically fetch API key from .env if available ---
+        # Extract provider key using regex
         provider_match = GlobalConfig.PROVIDER_REGEX.match(llm_provider_to_use)
         if provider_match:
             selected_provider = provider_match.group(1)
         else:
             # If regex doesn't match, try to extract provider from the beginning
             selected_provider = llm_provider_to_use.split(' ')[0] if ' ' in llm_provider_to_use else llm_provider_to_use
-            logger.warning(f"Provider regex did not match for: {llm_provider_to_use}, using: {selected_provider}")
+            logger.warning("Provider regex did not match for: %s, using: %s", llm_provider_to_use, selected_provider)
         
         # Validate that the selected provider is valid
         if selected_provider not in GlobalConfig.VALID_PROVIDERS:

helpers/llm_helper.py

@@ -5,7 +5,7 @@ import logging
 import re
 import sys
 import urllib3
-from typing import Tuple, Union, Iterator
+from typing import Tuple, Union, Iterator, Optional
 
 import requests
 import os
@@ -121,7 +121,7 @@ def is_valid_llm_provider_model(
     return True
 
 
-def get_litellm_model_name(provider: str, model: str) -> str:
+def get_litellm_model_name(provider: str, model: str) -> Optional[str]:
     """
     Convert provider and model to LiteLLM model name format.
     
@@ -184,6 +184,8 @@ def stream_litellm_completion(
         litellm_model = f'azure/{azure_deployment_name}'
     else:
         litellm_model = get_litellm_model_name(provider, model)
+        if not litellm_model:
+            raise ValueError(f"Invalid model name: {model} for provider: {provider}")
     
     # Prepare the request parameters
     request_params = {