feat(auth): add interactive API key setup supporting 70+ providers

Refactors the credential setup utility to integrate support for standard API keys alongside existing OAuth providers.

Key changes:
- The main menu is updated to allow selection between OAuth credential setup and API key setup.
- Implements `setup_api_key` to interactively guide the user through selecting from a list of recognized LiteLLM providers.
- API keys are automatically saved to the project's `.env` file, utilizing key indexing (e.g., `OPENAI_API_KEY_1`) to support multiple keys per provider.
- Introduces `ensure_env_defaults` to automatically initialize the `.env` file and set essential default values like `PROXY_API_KEY`.

setup_env.bat

@@ -1,121 +0,0 @@
-@echo off
-setlocal enabledelayedexpansion
-
-REM --- Configuration ---
-set "ENV_FILE=.env"
-set "DEFAULT_PROXY_KEY=VerysecretKey"
-
-REM --- Provider Name to Variable Name Mapping ---
-set "provider_count=0"
-set "provider_list[1]=Gemini" & set "provider_vars[1]=GEMINI" & set /a provider_count+=1
-set "provider_list[2]=OpenRouter" & set "provider_vars[2]=OPENROUTER" & set /a provider_count+=1
-set "provider_list[3]=Chutes" & set "provider_vars[3]=CHUTES" & set /a provider_count+=1
-set "provider_list[4]=Nvidia" & set "provider_vars[4]=NVIDIA_NIM" & set /a provider_count+=1
-set "provider_list[5]=OpenAI" & set "provider_vars[5]=OPENAI" & set /a provider_count+=1
-set "provider_list[6]=Anthropic" & set "provider_vars[6]=ANTHROPIC" & set /a provider_count+=1
-set "provider_list[7]=Mistral" & set "provider_vars[7]=MISTRAL" & set /a provider_count+=1
-set "provider_list[8]=Groq" & set "provider_vars[8]=GROQ" & set /a provider_count+=1
-set "provider_list[9]=Cohere" & set "provider_vars[9]=COHERE" & set /a provider_count+=1
-set "provider_list[10]=Bedrock" & set "provider_vars[10]=BEDROCK" & set /a provider_count+=1
-
-
-:main
-cls
-echo =================================================================
-echo      Welcome to the API Key Setup for Your Proxy Server
-echo =================================================================
-echo.
-echo This script will help you set up your .env file.
-echo.
-
-REM --- Ensure .env file exists and has PROXY_API_KEY ---
-if not exist "%ENV_FILE%" (
-    echo Creating a new %ENV_FILE% file for you...
-    echo PROXY_API_KEY="%DEFAULT_PROXY_KEY%" > "%ENV_FILE%"
-    echo.
-) else (
-    findstr /C:"PROXY_API_KEY=" "%ENV_FILE%" >nul
-    if errorlevel 1 (
-        echo Adding the default proxy key to your .env file...
-        echo.>> "%ENV_FILE%"
-        echo PROXY_API_KEY="%DEFAULT_PROXY_KEY%" >> "%ENV_FILE%"
-        echo.
-    )
-)
-
-:get_provider
-echo -----------------------------------------------------------------
-echo Please choose a provider to add an API key for:
-echo -----------------------------------------------------------------
-echo.
-for /L %%i in (1,1,%provider_count%) do (
-    echo   %%i. !provider_list[%%i]!
-)
-echo.
-set /p "choice=Type the number of the provider and press Enter: "
-
-REM --- Validate Provider Choice ---
-set "VAR_NAME="
-set "provider_choice="
-if %choice% GTR 0 if %choice% LEQ %provider_count% (
-    set "VAR_NAME=!provider_vars[%choice%]!"
-    set "provider_choice=!provider_list[%choice%]!"
-)
-
-if not defined VAR_NAME (
-    cls
-    echo =================================================================
-    echo      INVALID SELECTION! Please try again.
-    echo =================================================================
-    echo.
-    pause
-    goto :get_provider
-)
-
-set "API_VAR_BASE=%VAR_NAME%_API_KEY"
-
-:get_key
-echo.
-echo -----------------------------------------------------------------
-set /p "api_key=Enter the API key for %provider_choice%: "
-if not defined api_key (
-    echo You must enter an API key.
-    goto :get_key
-)
-echo -----------------------------------------------------------------
-echo.
-
-REM --- Find the next available key number ---
-set /a key_index=1
-:find_next_key
-findstr /R /C:"^%API_VAR_BASE%_%key_index% *=" "%ENV_FILE%" >nul
-if %errorlevel% equ 0 (
-    set /a key_index+=1
-    goto :find_next_key
-)
-
-REM --- Append the new key to the .env file ---
-echo Adding your key to %ENV_FILE%...
-echo %API_VAR_BASE%_%key_index%="%api_key%" >> "%ENV_FILE%"
-echo.
-echo Successfully added %provider_choice% API key as %API_VAR_BASE%_%key_index%!
-echo.
-
-:ask_another
-set /p "another=Do you want to add another key? (yes/no): "
-if /i "%another%"=="yes" (
-    goto :main
-)
-if /i "%another%"=="y" (
-    goto :main
-)
-
-cls
-echo =================================================================
-echo      Setup Complete! Your .env file is ready.
-echo =================================================================
-echo.
-echo You can now run the proxy server.
-echo.
-pause
-exit /b

src/rotator_library/credential_tool.py

@@ -4,8 +4,10 @@ import asyncio
 import json
 import re
 from pathlib import Path
+from dotenv import set_key, get_key
 
 from .provider_factory import get_provider_auth_class, get_available_providers
+from .providers import PROVIDER_PLUGINS
 from rich.console import Console
 from rich.panel import Panel
 from rich.prompt import Prompt
@@ -13,9 +15,138 @@ from rich.text import Text
 
 OAUTH_BASE_DIR = Path.cwd() / "oauth_creds"
 OAUTH_BASE_DIR.mkdir(exist_ok=True)
+# Use a direct path to the .env file in the project root
+ENV_FILE = Path.cwd() / ".env"
+
 
 console = Console()
 
+def ensure_env_defaults():
+    """
+    Ensures the .env file exists and contains essential default values like PROXY_API_KEY.
+    """
+    if not ENV_FILE.is_file():
+        ENV_FILE.touch()
+        console.print(f"Creating a new [bold yellow]{ENV_FILE.name}[/bold yellow] file...")
+
+    # Check for PROXY_API_KEY, similar to setup_env.bat
+    if get_key(str(ENV_FILE), "PROXY_API_KEY") is None:
+        default_key = "VerysecretKey"
+        console.print(f"Adding default [bold cyan]PROXY_API_KEY[/bold cyan] to [bold yellow]{ENV_FILE.name}[/bold yellow]...")
+        set_key(str(ENV_FILE), "PROXY_API_KEY", default_key)
+
+async def setup_api_key():
+    """
+    Interactively sets up a new API key for a provider.
+    """
+    console.print(Panel("[bold cyan]API Key Setup[/bold cyan]", expand=False))
+
+    # Verified list of LiteLLM providers with their friendly names and API key variables
+    LITELLM_PROVIDERS = {
+        "OpenAI": "OPENAI_API_KEY", "Anthropic": "ANTHROPIC_API_KEY",
+        "Google AI Studio (Gemini)": "GEMINI_API_KEY", "Azure OpenAI": "AZURE_API_KEY",
+        "Vertex AI": "GOOGLE_API_KEY", "AWS Bedrock": "AWS_ACCESS_KEY_ID",
+        "Cohere": "COHERE_API_KEY", "Mistral AI": "MISTRAL_API_KEY",
+        "Codestral (Mistral)": "CODESTRAL_API_KEY", "Groq": "GROQ_API_KEY",
+        "Perplexity": "PERPLEXITYAI_API_KEY", "xAI": "XAI_API_KEY",
+        "Together AI": "TOGETHERAI_API_KEY", "Fireworks AI": "FIREWORKS_AI_API_KEY",
+        "Replicate": "REPLICATE_API_KEY", "Hugging Face": "HUGGINGFACE_API_KEY",
+        "Anyscale": "ANYSCALE_API_KEY", "NVIDIA NIM": "NVIDIA_NIM_API_KEY",
+        "Deepseek": "DEEPSEEK_API_KEY", "AI21": "AI21_API_KEY",
+        "Cerebras": "CEREBRAS_API_KEY", "Moonshot": "MOONSHOT_API_KEY",
+        "Ollama": "OLLAMA_API_KEY", "Xinference": "XINFERENCE_API_KEY",
+        "Infinity": "INFINITY_API_KEY", "OpenRouter": "OPENROUTER_API_KEY",
+        "Deepinfra": "DEEPINFRA_API_KEY", "Cloudflare": "CLOUDFLARE_API_KEY",
+        "Baseten": "BASETEN_API_KEY", "Modal": "MODAL_API_KEY",
+        "Databricks": "DATABRICKS_API_KEY", "AWS SageMaker": "AWS_ACCESS_KEY_ID",
+        "IBM watsonx.ai": "WATSONX_APIKEY", "Predibase": "PREDIBASE_API_KEY",
+        "Clarifai": "CLARIFAI_API_KEY", "NLP Cloud": "NLP_CLOUD_API_KEY",
+        "Voyage AI": "VOYAGE_API_KEY", "Jina AI": "JINA_API_KEY",
+        "Hyperbolic": "HYPERBOLIC_API_KEY", "Morph": "MORPH_API_KEY",
+        "Lambda AI": "LAMBDA_API_KEY", "Novita AI": "NOVITA_API_KEY",
+        "Aleph Alpha": "ALEPH_ALPHA_API_KEY", "SambaNova": "SAMBANOVA_API_KEY",
+        "FriendliAI": "FRIENDLI_TOKEN", "Galadriel": "GALADRIEL_API_KEY",
+        "CompactifAI": "COMPACTIFAI_API_KEY", "Lemonade": "LEMONADE_API_KEY",
+        "GradientAI": "GRADIENTAI_API_KEY", "Featherless AI": "FEATHERLESS_AI_API_KEY",
+        "Nebius AI Studio": "NEBIUS_API_KEY", "Dashscope (Qwen)": "DASHSCOPE_API_KEY",
+        "Bytez": "BYTEZ_API_KEY", "Oracle OCI": "OCI_API_KEY",
+        "DataRobot": "DATAROBOT_API_KEY", "OVHCloud": "OVHCLOUD_API_KEY",
+        "Volcengine": "VOLCENGINE_API_KEY", "Snowflake": "SNOWFLAKE_API_KEY",
+        "Nscale": "NSCALE_API_KEY", "Recraft": "RECRAFT_API_KEY",
+        "v0": "V0_API_KEY", "Vercel": "VERCEL_AI_GATEWAY_API_KEY",
+        "Topaz": "TOPAZ_API_KEY", "ElevenLabs": "ELEVENLABS_API_KEY",
+        "Deepgram": "DEEPGRAM_API_KEY", "Custom API": "CUSTOM_API_KEY",
+        "GitHub Models": "GITHUB_TOKEN", "GitHub Copilot": "GITHUB_COPILOT_API_KEY",
+    }
+
+    # Discover custom providers and add them to the list
+    oauth_providers = {'gemini_cli', 'qwen_code'}
+    discovered_providers = {
+        p.replace('_', ' ').title(): p.upper() + "_API_KEY"
+        for p in PROVIDER_PLUGINS.keys()
+        if p not in oauth_providers and p.replace('_', ' ').title() not in LITELLM_PROVIDERS
+    }
+    
+    combined_providers = {**LITELLM_PROVIDERS, **discovered_providers}
+    provider_display_list = sorted(combined_providers.keys())
+
+    provider_text = Text()
+    for i, provider_name in enumerate(provider_display_list):
+        provider_text.append(f"  {i + 1}. {provider_name}\n")
+
+    console.print(Panel(provider_text, title="Available Providers for API Key", style="bold blue"))
+
+    choice = Prompt.ask(
+        Text.from_markup("[bold]Please select a provider or type [red]'b'[/red] to go back[/bold]"),
+        choices=[str(i + 1) for i in range(len(provider_display_list))] + ["b"],
+        show_choices=False
+    )
+
+    if choice.lower() == 'b':
+        return
+
+    try:
+        choice_index = int(choice) - 1
+        if 0 <= choice_index < len(provider_display_list):
+            display_name = provider_display_list[choice_index]
+            api_var_base = combined_providers[display_name]
+
+            api_key = Prompt.ask(f"Enter the API key for {display_name}")
+
+            # Special handling for AWS
+            if display_name in ["AWS Bedrock", "AWS SageMaker"]:
+                console.print(Panel(
+                    Text.from_markup(
+                        "This provider requires both an Access Key ID and a Secret Access Key.\n"
+                        f"The key you entered will be saved as [bold yellow]{api_var_base}_1[/bold yellow].\n"
+                        "Please manually add the [bold cyan]AWS_SECRET_ACCESS_KEY_1[/bold cyan] to your .env file."
+                    ),
+                    title="[bold yellow]Additional Step Required[/bold yellow]",
+                    border_style="yellow"
+                ))
+
+            key_index = 1
+            while True:
+                key_name = f"{api_var_base}_{key_index}"
+                if ENV_FILE.is_file():
+                     with open(ENV_FILE, "r") as f:
+                        if not any(line.startswith(f"{key_name}=") for line in f):
+                            break
+                else:
+                    break
+                key_index += 1
+            
+            key_name = f"{api_var_base}_{key_index}"
+            set_key(str(ENV_FILE), key_name, api_key)
+            
+            success_text = Text.from_markup(f"Successfully added {display_name} API key as [bold yellow]'{key_name}'[/bold yellow].")
+            console.print(Panel(success_text, style="bold green", title="Success"))
+
+        else:
+            console.print("[bold red]Invalid choice. Please try again.[/bold red]")
+    except ValueError:
+        console.print("[bold red]Invalid input. Please enter a number or 'b'.[/bold red]")
+
 async def setup_new_credential(provider_name: str):
     """
     Interactively sets up a new OAuth credential for a given provider.
@@ -66,39 +197,65 @@ async def setup_new_credential(provider_name: str):
 
 async def main():
     """
-    An interactive CLI tool to add new OAuth credentials.
+    An interactive CLI tool to add new credentials.
     """
+    ensure_env_defaults()
     console.print(Panel("[bold cyan]Interactive Credential Setup[/bold cyan]", title="--- API Key Proxy ---", expand=False))
     
     while True:
-        available_providers = get_available_providers()
+        console.print(Panel(
+            Text.from_markup("1. Add OAuth Credential\n2. Add API Key"),
+            title="Choose credential type",
+            style="bold blue"
+        ))
         
-        provider_text = Text()
-        for i, provider in enumerate(available_providers):
-            provider_text.append(f"  {i + 1}. {provider.capitalize()}\n")
-        
-        console.print(Panel(provider_text, title="Available Providers", style="bold blue"))
-
-        choice = Prompt.ask(
-            Text.from_markup("[bold]Please select a provider or type [red]'q'[/red] to quit[/bold]"),
-            choices=[str(i + 1) for i in range(len(available_providers))] + ["q"],
+        setup_type = Prompt.ask(
+            Text.from_markup("[bold]Please select an option or type [red]'q'[/red] to quit[/bold]"),
+            choices=["1", "2", "q"],
             show_choices=False
         )
 
-        if choice.lower() == 'q':
+        if setup_type.lower() == 'q':
             break
         
-        try:
-            choice_index = int(choice) - 1
-            if 0 <= choice_index < len(available_providers):
-                provider_name = available_providers[choice_index]
-                console.print(f"\nStarting setup for [bold cyan]{provider_name.capitalize()}[/bold cyan]...")
-                await setup_new_credential(provider_name)
-            else:
-                console.print("[bold red]Invalid choice. Please try again.[/bold red]")
-        except ValueError:
-            console.print("[bold red]Invalid input. Please enter a number or 'q'.[/bold red]")
-        
+        if setup_type == "1":
+            available_providers = get_available_providers()
+            oauth_friendly_names = {
+                "gemini_cli": "Gemini CLI (OAuth)",
+                "qwen_code": "Qwen Code (OAuth)"
+            }
+            
+            provider_text = Text()
+            for i, provider in enumerate(available_providers):
+                display_name = oauth_friendly_names.get(provider, provider.replace('_', ' ').title())
+                provider_text.append(f"  {i + 1}. {display_name}\n")
+            
+            console.print(Panel(provider_text, title="Available Providers for OAuth", style="bold blue"))
+
+            choice = Prompt.ask(
+                Text.from_markup("[bold]Please select a provider or type [red]'b'[/red] to go back[/bold]"),
+                choices=[str(i + 1) for i in range(len(available_providers))] + ["b"],
+                show_choices=False
+            )
+
+            if choice.lower() == 'b':
+                continue
+            
+            try:
+                choice_index = int(choice) - 1
+                if 0 <= choice_index < len(available_providers):
+                    provider_name = available_providers[choice_index]
+                    display_name = oauth_friendly_names.get(provider_name, provider_name.replace('_', ' ').title())
+                    console.print(f"\nStarting OAuth setup for [bold cyan]{display_name}[/bold cyan]...")
+                    await setup_new_credential(provider_name)
+                else:
+                    console.print("[bold red]Invalid choice. Please try again.[/bold red]")
+            except ValueError:
+                console.print("[bold red]Invalid input. Please enter a number or 'b'.[/bold red]")
+
+        elif setup_type == "2":
+            await setup_api_key()
+
         console.print("\n" + "="*50 + "\n")
 
 def run_credential_tool():