ayechat-dev 0.36.9.20260201141756__py3-none-any.whl → 0.36.9.20260204003405__py3-none-any.whl

aye/controller/command_handlers.py

@@ -7,7 +7,7 @@ from prompt_toolkit import PromptSession
 from rich import print as rprint
 from rich.console import Console
 
-from aye.model.auth import get_user_config, set_user_config
+from aye.model.auth import get_user_config, set_user_config, delete_user_config
 from aye.model.config import MODELS
 from aye.presenter.repl_ui import print_error
 from aye.controller.llm_invoker import invoke_llm
@@ -37,7 +37,7 @@ def handle_model_command(session: Optional[PromptSession], models: list, conf: A
             num = int(tokens[1])
             if 1 <= num <= len(models):
                 selected_id = models[num - 1]["id"]
-                
+
                 # Check if this is an offline model and trigger download if needed
                 selected_model = models[num - 1]
                 if selected_model.get("type") == "offline":
@@ -49,7 +49,7 @@ def handle_model_command(session: Optional[PromptSession], models: list, conf: A
                     if download_response and not download_response.get("success", True):
                         rprint(f"[red]Failed to download model: {download_response.get('error', 'Unknown error')}[/]")
                         return
-                
+
                 conf.selected_model = selected_id
                 set_user_config("selected_model", selected_id)
                 rprint(f"[green]Selected model: {models[num - 1]['name']}[/]")
@@ -81,7 +81,7 @@ def handle_model_command(session: Optional[PromptSession], models: list, conf: A
             num = int(choice)
             if 1 <= num <= len(models):
                 selected_id = models[num - 1]["id"]
-                
+
                 # Check if this is an offline model and trigger download if needed
                 selected_model = models[num - 1]
                 if selected_model.get("type") == "offline":
@@ -93,7 +93,7 @@ def handle_model_command(session: Optional[PromptSession], models: list, conf: A
                     if download_response and not download_response.get("success", True):
                         rprint(f"[red]Failed to download model: {download_response.get('error', 'Unknown error')}[/]")
                         return
-                
+
                 conf.selected_model = selected_id
                 set_user_config("selected_model", selected_id)
                 rprint(f"[green]Selected: {models[num - 1]['name']}[/]")
@@ -147,7 +147,7 @@ def handle_debug_command(tokens: list):
 
 def handle_completion_command(tokens: list) -> Optional[str]:
     """Handle the 'completion' command for switching completion styles.
-    
+
     Returns:
         The new completion style if changed ('readline' or 'multi'), None otherwise.
     """
@@ -169,25 +169,125 @@ def handle_completion_command(tokens: list) -> Optional[str]:
         return None
 
 
+def handle_llm_command(session: Optional[PromptSession], tokens: list[str]) -> None:
+    """Handle the 'llm' command for configuring OpenAI-compatible local model endpoint.
+
+    Usage:
+        llm         - Interactively configure URL, key, and model
+        llm clear   - Remove all LLM config values
+
+    Config keys stored in ~/.ayecfg:
+        llm_api_url
+        llm_api_key
+        llm_model
+    """
+    # Handle 'llm clear' subcommand
+    if len(tokens) > 1 and tokens[1].lower() == "clear":
+        delete_user_config("llm_api_url")
+        delete_user_config("llm_api_key")
+        delete_user_config("llm_model")
+        rprint("[green]LLM config cleared.[/]")
+        return
+
+    # Interactive configuration
+    current_url = get_user_config("llm_api_url", "")
+    current_key = get_user_config("llm_api_key", "")
+    current_model = get_user_config("llm_model", "")
+
+    # Show current status
+    rprint("\n[bold cyan]LLM Endpoint Configuration[/]")
+    rprint("[dim]Press Enter to keep current value, or type a new value.[/]\n")
+
+    if not session:
+        rprint("[red]Error: Interactive session not available.[/]")
+        return
+
+    try:
+        # Prompt for URL (explicitly non-password; some prompt_toolkit versions may reuse app state)
+        url_display = current_url if current_url else "not set"
+        new_url = session.prompt(
+            f"LLM API URL (current: {url_display}): ",
+            is_password=False,
+        ).strip()
+        final_url = new_url if new_url else current_url
+
+        # Prompt for API key (hidden input)
+        key_display = "set" if current_key else "not set"
+        new_key = session.prompt(
+            f"LLM API KEY (current: {key_display}): ",
+            is_password=True,
+        ).strip()
+        final_key = new_key if new_key else current_key
+
+        # Prompt for model (explicitly non-password)
+        model_display = current_model if current_model else "not set"
+        new_model = session.prompt(
+            f"LLM MODEL (current: {model_display}): ",
+            is_password=False,
+        ).strip()
+        final_model = new_model if new_model else current_model
+
+    except (EOFError, KeyboardInterrupt):
+        rprint("\n[yellow]Configuration cancelled.[/]")
+        return
+
+    # Save values (only if they have content)
+    if final_url:
+        set_user_config("llm_api_url", final_url)
+    elif current_url and not new_url:
+        # Keep existing
+        pass
+    else:
+        delete_user_config("llm_api_url")
+
+    if final_key:
+        set_user_config("llm_api_key", final_key)
+    elif current_key and not new_key:
+        # Keep existing
+        pass
+    else:
+        delete_user_config("llm_api_key")
+
+    if final_model:
+        set_user_config("llm_model", final_model)
+    elif current_model and not new_model:
+        # Keep existing
+        pass
+    else:
+        delete_user_config("llm_model")
+
+    # Print confirmation
+    rprint("\n[bold cyan]LLM Configuration Updated[/]")
+    rprint(f"  URL:   {final_url if final_url else '[dim]not set[/]'}")
+    rprint(f"  KEY:   {'[dim]set (hidden)[/]' if final_key else '[dim]not set[/]'}")
+    rprint(f"  MODEL: {final_model if final_model else '[dim]not set[/]'}")
+
+    # Show status message
+    if final_url and final_key:
+        rprint("\n[green] OpenAI-compatible endpoint is configured and active.[/]")
+    else:
+        rprint("\n[yellow] Both URL and KEY are required for the local LLM endpoint to be active.[/]")
+
+
 def _expand_file_patterns(patterns: list[str], conf: Any) -> list[str]:
     """Expand wildcard patterns and return a list of existing file paths."""
     expanded_files = []
-    
+
     for pattern in patterns:
         pattern = pattern.strip()
         if not pattern:
             continue
-            
+
         # Check if it's a direct file path first
         direct_path = conf.root / pattern
         if direct_path.is_file():
             expanded_files.append(pattern)
             continue
-            
+
         # Use glob to expand wildcards
         # Search relative to the project root
         matched_paths = list(conf.root.glob(pattern))
-        
+
         # Add relative paths of matched files
         for matched_path in matched_paths:
             if matched_path.is_file():
@@ -197,26 +297,26 @@ def _expand_file_patterns(patterns: list[str], conf: Any) -> list[str]:
                 except ValueError:
                     # If we can't make it relative, use the original pattern
                     expanded_files.append(pattern)
-    
+
     return expanded_files
 
 
 def handle_with_command(
-    prompt: str, 
-    conf: Any, 
-    console: Console, 
-    chat_id: int, 
+    prompt: str,
+    conf: Any,
+    console: Console,
+    chat_id: int,
     chat_id_file: Path
 ) -> Optional[int]:
     """Handle the 'with' command for file-specific prompts with wildcard support.
-    
+
     Args:
         prompt: The full prompt string starting with 'with'
         conf: Configuration object
         console: Rich console for output
         chat_id: Current chat ID
         chat_id_file: Path to chat ID file
-        
+
     Returns:
         New chat_id if available, None otherwise
     """
@@ -234,16 +334,16 @@ def handle_with_command(
 
         # Parse file patterns (can include wildcards)
         file_patterns = [f.strip() for f in file_list_str.replace(",", " ").split() if f.strip()]
-        
+
         # Expand wildcards to get actual file paths
         expanded_files = _expand_file_patterns(file_patterns, conf)
-        
+
         if not expanded_files:
             rprint("[red]Error: No files found matching the specified patterns.[/red]")
             return None
-        
+
         explicit_source_files = {}
-        
+
         for file_name in expanded_files:
             file_path = conf.root / file_name
             if not file_path.is_file():
@@ -254,11 +354,11 @@ def handle_with_command(
             except Exception as e:
                 rprint(f"[red]Could not read file '{file_name}': {e}[/red]")
                 continue  # Continue with other files instead of breaking
-        
+
         if not explicit_source_files:
             rprint("[red]Error: No readable files found.[/red]")
             return None
-        
+
         # Show which files were included
         if conf.verbose or len(explicit_source_files) != len(expanded_files):
             rprint(f"[cyan]Including {len(explicit_source_files)} file(s): {', '.join(explicit_source_files.keys())}[/cyan]")
@@ -272,20 +372,20 @@ def handle_with_command(
             verbose=conf.verbose,
             explicit_source_files=explicit_source_files
         )
-        
+
         if llm_response:
             new_chat_id = process_llm_response(
-                response=llm_response, 
-                conf=conf, 
-                console=console, 
-                prompt=new_prompt_str.strip(), 
+                response=llm_response,
+                conf=conf,
+                console=console,
+                prompt=new_prompt_str.strip(),
                 chat_id_file=chat_id_file if llm_response.chat_id else None
             )
             return new_chat_id
         else:
             rprint("[yellow]No response from LLM.[/]")
             return None
-            
+
     except Exception as exc:
         handle_llm_error(exc)
         return None

aye/controller/repl.py

@@ -41,6 +41,7 @@ from aye.controller.command_handlers import (
     handle_completion_command,
     handle_with_command,
     handle_blog_command,
+    handle_llm_command,
 )
 
 DEBUG = False
@@ -284,7 +285,7 @@ def _execute_forced_shell_command(command: str, args: List[str], conf: Any) -> N
 def chat_repl(conf: Any) -> None:
     is_first_run = run_first_time_tutorial_if_needed()
 
-    BUILTIN_COMMANDS = ["with", "blog", "new", "history", "diff", "restore", "undo", "keep", "model", "verbose", "debug", "completion", "exit", "quit", ":q", "help", "cd", "db"]
+    BUILTIN_COMMANDS = ["with", "blog", "new", "history", "diff", "restore", "undo", "keep", "model", "verbose", "debug", "completion", "exit", "quit", ":q", "help", "cd", "db", "llm"]
 
     # Get the completion style setting
     completion_style = get_user_config("completion_style", "readline").lower()
@@ -429,6 +430,9 @@ def chat_repl(conf: Any) -> None:
                         # Recreate the session with the new completer
                         session = create_prompt_session(completer, new_style)
                         rprint(f"[green]Completion style is now active.[/]")
+                elif lowered_first == "llm":
+                    telemetry.record_command("llm", has_args=len(tokens) > 1, prefix=_AYE_PREFIX)
+                    handle_llm_command(session, tokens)
                 elif lowered_first == "blog":
                     telemetry.record_command("blog", has_args=len(tokens) > 1, prefix=_AYE_PREFIX)
                     telemetry.record_llm_prompt("LLM <blog>")

aye/model/auth.py

@@ -58,6 +58,27 @@ def set_user_config(key: str, value: Any) -> None:
     TOKEN_FILE.chmod(0o600)
 
 
+def delete_user_config(key: str) -> None:
+    """Delete a user config key from the [default] section.
+    
+    If the key doesn't exist, this is a no-op.
+    Preserves other settings and maintains file permissions.
+    """
+    config = _parse_user_config()
+    if key not in config:
+        return
+    config.pop(key, None)
+    if not config:
+        # If no config left, remove the file entirely
+        TOKEN_FILE.unlink(missing_ok=True)
+    else:
+        new_content = "[default]\n"
+        for k, v in config.items():
+            new_content += f"{k}={v}\n"
+        TOKEN_FILE.write_text(new_content, encoding="utf-8")
+        TOKEN_FILE.chmod(0o600)
+
+
 def store_token(token: str) -> None:
     """Persist the token in ~/.ayecfg or value from AYE_TOKEN_FILE environment variable (unless AYE_TOKEN is set)."""
     token = token.strip()
@@ -122,4 +143,3 @@ def login_flow() -> None:
     token = typer.prompt("Paste your token", hide_input=True)
     store_token(token.strip())
     typer.secho("✅ Token saved.", fg=typer.colors.GREEN)
-

aye/plugins/local_model.py

@@ -8,6 +8,7 @@ from rich import print as rprint
 
 from .plugin_base import Plugin
 from aye.model.config import SYSTEM_PROMPT, MODELS, DEFAULT_MAX_OUTPUT_TOKENS
+from aye.model.auth import get_user_config
 from aye.controller.util import is_truncated_json
 
 LLM_TIMEOUT = 600.0
@@ -179,9 +180,17 @@ class LocalModelPlugin(Plugin):
             return self._create_error_response(f"Error calling Databricks API: {str(e)}")
 
     def _handle_openai_compatible(self, prompt: str, source_files: Dict[str, str], chat_id: Optional[int] = None, system_prompt: Optional[str] = None, max_output_tokens: int = DEFAULT_MAX_OUTPUT_TOKENS) -> Optional[Dict[str, Any]]:
-        api_url = os.environ.get("AYE_LLM_API_URL")
-        api_key = os.environ.get("AYE_LLM_API_KEY")
-        model_name = os.environ.get("AYE_LLM_MODEL", "gpt-3.5-turbo")
+        """Handle OpenAI-compatible API endpoints.
+        
+        Reads configuration from:
+        - get_user_config("llm_api_url") / AYE_LLM_API_URL
+        - get_user_config("llm_api_key") / AYE_LLM_API_KEY  
+        - get_user_config("llm_model") / AYE_LLM_MODEL (default: gpt-3.5-turbo)
+        """
+        # Read from config (supports both ~/.ayecfg and AYE_LLM_* env vars)
+        api_url = get_user_config("llm_api_url")
+        api_key = get_user_config("llm_api_key")
+        model_name = get_user_config("llm_model", "gpt-3.5-turbo")
         
         if not api_url or not api_key:
             return None

aye/presenter/repl_ui.py

@@ -89,6 +89,7 @@ def print_help_message():
         ("@filename", "Include a file in your prompt inline (e.g., \"explain @main.py\"). Supports wildcards (e.g., @*.py, @src/*.js)."),
         ("!command", "Force shell execution (e.g., \"!echo hello\")."),
         ("model", "Select a different model. Selection will persist between sessions."),
+        ("llm", "Configure OpenAI-compatible LLM endpoint (URL, key, model). Use 'llm clear' to reset."),
         (r"verbose \[on|off]", "Toggle verbose mode to increase or decrease chattiness (on/off, persists between sessions)"),
         (r"completion \[readline|multi]", "Switch auto-completion style (readline or multi, persists between sessions)"),
         ("new", "Start a new chat session (if you want to change the subject)"),

{ayechat_dev-0.36.9.20260201141756.dist-info → ayechat_dev-0.36.9.20260204003405.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ayechat-dev
-Version: 0.36.9.20260201141756
+Version: 0.36.9.20260204003405
 Summary: Aye Chat: Terminal-first AI Code Generator
 Author-email: "Acrotron, Inc." <info@acrotron.com>
 License: MIT

{ayechat_dev-0.36.9.20260201141756.dist-info → ayechat_dev-0.36.9.20260204003405.dist-info}/RECORD

@@ -3,18 +3,18 @@ aye/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aye/__main__.py,sha256=fj7pl0i_mLvpKAbpKtotU3zboLBTivsILSLKuH5M5Sg,5377
 aye/__main_chat__.py,sha256=R6RaidxG3Px5TaYxcoWAuIleE5KUZlceneUB6u_9UVU,1066
 aye/controller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aye/controller/command_handlers.py,sha256=C-6beHXVhW8vR_AbH4dCPl794ycFNVGuMBQiHnH4naI,13438
+aye/controller/command_handlers.py,sha256=72dA9x4vHOxsSuWb1wEX2xTxL4uCFuiSPL4-VpEPwyA,16728
 aye/controller/commands.py,sha256=sXmK_sgNBrw9Fs7mKcr93-wsu740ZlvWSisQfS-1EUE,12278
 aye/controller/llm_handler.py,sha256=BSBab6onF9BYiFndYW1647eEy377Vt52trzu0Qjm4bQ,5075
 aye/controller/llm_invoker.py,sha256=p_Vk2a3YrWKwDupLfSVRinR5llDfq1Fb_f7WrYozK6M,14127
 aye/controller/plugin_manager.py,sha256=9ZuITyA5sQJJJU-IntLQ1SsxXsDnbgZKPOF4e9VmsEU,3018
-aye/controller/repl.py,sha256=UCgH4bdL3AUYjgD6KPtIOeysCNaYbs0TvEwS3Zql8kY,26552
+aye/controller/repl.py,sha256=fTYEdX1PAds49pRBVgvyi-4lIfArR8gjBTp_shwx-K0,26782
 aye/controller/tutorial.py,sha256=lc92jOcJOYCVrrjTEF0Suk4-8jn-ku98kTJEIL8taUA,7254
 aye/controller/util.py,sha256=gBmktDEaY63OKhgzZHA2IFrgcWUN_Iphn3e1daEeUBI,2828
 aye/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aye/model/api.py,sha256=HhSMQQ_szdC2ZPOSfNsJRbs1FRwb6WyYIeLejB2ScbA,13272
 aye/model/ast_chunker.py,sha256=rVcDdynVUXXyxWVgtUcsee_STqB7SAwP776ktWTlYig,4462
-aye/model/auth.py,sha256=FDTUGubrIzVQ5tdk97JFOwUGQNfDDcvRaqYOA9MTI_w,4144
+aye/model/auth.py,sha256=ozV_uQxdqXtUoWO3nZwpzVnDOIfnRAmSMC6W0N724vE,4800
 aye/model/config.py,sha256=o6bQhj5gqhSqtWD6DLow7NNy6Hdaede02h_xb7uPLXo,9280
 aye/model/download_plugins.py,sha256=6omyFGdxlEIb7tKPLq4rRVrRYeCPUUCE8aZHvJAKGSc,4442
 aye/model/file_processor.py,sha256=b7YGvHAmhGto9JbtzcfrsdkFtksHbosYt-42EnR22Uo,2131
@@ -42,7 +42,7 @@ aye/plugins/__init__.py,sha256=dSTxs461ICx0O1tbCBCca0W_7QIAa0Yt9PQhHiT5uZQ,173
 aye/plugins/at_file_completer.py,sha256=uNS4gWpfKvn9_nGxZbhQVjVg_S82g977gfBR-pL3XrQ,19582
 aye/plugins/auto_detect_mask.py,sha256=gZKH4qkR-A73uKpMkPXhlgI452Ae_2YG1nHtaIkOvwM,6864
 aye/plugins/completer.py,sha256=qhxke5Q76P2u0LojSIL3V48RTNG5tWL-5-TK5tNutrE,13893
-aye/plugins/local_model.py,sha256=Jj4bHiPYaLMx6zTrKamBPrkiGQQ787jWz0F4ojRCjlQ,14394
+aye/plugins/local_model.py,sha256=q0RjSjLhEQcDMOCLAK6k1YCW5ECrvdT_g0lKRHMX-AE,14810
 aye/plugins/offline_llm.py,sha256=qFmd1e8Lbl7yiMgXpXjOQkQTNxOk0_WXU7km2DTKXGY,13357
 aye/plugins/plugin_base.py,sha256=t5hTOnA0dZC237BnseAgdXbOqErlSCNLUo_Uul09TSw,1673
 aye/plugins/shell_executor.py,sha256=a0mlZnQeURONdtPM7iageTcQ8PiNLQbjxoY54EsS32o,7502
@@ -50,12 +50,12 @@ aye/plugins/slash_completer.py,sha256=MyrDTC_KwVWhtD_kpHbu0WjSjmSAWp36PwOBQczSuX
 aye/presenter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aye/presenter/cli_ui.py,sha256=8oHqQiMHO8hHXCTdqWoquMkJBshl2_3-YWN6SQnlbKg,8449
 aye/presenter/diff_presenter.py,sha256=cbxfOEqGomPTDvQpKdybfYeNUD2DYVAl85j1uy5--Ww,12512
-aye/presenter/repl_ui.py,sha256=J6QlcFa_qvV00NP1OS5xyhDuJcfNcCrADZq9ZFY2GKQ,7917
+aye/presenter/repl_ui.py,sha256=5nAv8qLo3azDuoGYAxGdK2SEwowXPoHSEruupvS6jy8,8023
 aye/presenter/streaming_ui.py,sha256=_3tBEuNH9UQ9Gyq2yuvRfX4SWVkcGMYirEUGj-MXVJ0,12768
 aye/presenter/ui_utils.py,sha256=6KXR4_ZZZUdF5pCHrPqO8yywlQk7AOzWe-2B4Wj_-ZQ,5441
-ayechat_dev-0.36.9.20260201141756.dist-info/licenses/LICENSE,sha256=U1ou6lkMKmPo16-E9YowIu3goU7sOWKUprGo0AOA72s,1065
-ayechat_dev-0.36.9.20260201141756.dist-info/METADATA,sha256=1whf_VoyU8Vt49wLxnbCTCtt6vLGS1NtXCsJDaRqFQI,7718
-ayechat_dev-0.36.9.20260201141756.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ayechat_dev-0.36.9.20260201141756.dist-info/entry_points.txt,sha256=KGsOma6szoefNN6vHozg3Pbf1fjZ7ZbmwrOiVwBd0Ik,41
-ayechat_dev-0.36.9.20260201141756.dist-info/top_level.txt,sha256=7WZL0LOx4-GKKvgU1mtI5s4Dhk2OdieVZZvVnxFJHr8,4
-ayechat_dev-0.36.9.20260201141756.dist-info/RECORD,,
+ayechat_dev-0.36.9.20260204003405.dist-info/licenses/LICENSE,sha256=U1ou6lkMKmPo16-E9YowIu3goU7sOWKUprGo0AOA72s,1065
+ayechat_dev-0.36.9.20260204003405.dist-info/METADATA,sha256=EHTmVsIzzf2ki5cb1N9-x4eWLHLFbp3eWxtU3pUKe4g,7718
+ayechat_dev-0.36.9.20260204003405.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ayechat_dev-0.36.9.20260204003405.dist-info/entry_points.txt,sha256=KGsOma6szoefNN6vHozg3Pbf1fjZ7ZbmwrOiVwBd0Ik,41
+ayechat_dev-0.36.9.20260204003405.dist-info/top_level.txt,sha256=7WZL0LOx4-GKKvgU1mtI5s4Dhk2OdieVZZvVnxFJHr8,4
+ayechat_dev-0.36.9.20260204003405.dist-info/RECORD,,