ayechat 0.37.0__tar.gz → 0.39.0__tar.gz

{ayechat-0.37.0/src/ayechat.egg-info → ayechat-0.39.0}/PKG-INFO

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

{ayechat-0.37.0 → ayechat-0.39.0}/src/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']}[/]")
@@ -145,9 +145,28 @@ def handle_debug_command(tokens: list):
         rprint(f"[yellow]Debug mode is {current.title()}[/]")
 
 
+def handle_autodiff_command(tokens: list):
+    """Handle the 'autodiff' command for toggling automatic diff display.
+
+    When autodiff is enabled, diffs are automatically displayed for every
+    file modified by an LLM response.
+    """
+    if len(tokens) > 1:
+        val = tokens[1].lower()
+        if val in ("on", "off"):
+            set_user_config("autodiff", val)
+            rprint(f"[green]Autodiff set to {val.title()}[/]")
+        else:
+            rprint("[red]Usage: autodiff on|off[/]")
+    else:
+        current = get_user_config("autodiff", "off")
+        rprint(f"[yellow]Autodiff is {current.title()}[/]")
+        rprint("[dim]When on, diffs are shown automatically after each LLM file update.[/]")
+
+
 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 +188,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 +316,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 +353,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 +373,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 +391,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

{ayechat-0.37.0 → ayechat-0.39.0}/src/aye/controller/llm_handler.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Any, Optional
+from typing import Any, Optional, List
 
 from rich import print as rprint
 from rich.console import Console
@@ -11,10 +11,12 @@ from aye.presenter.repl_ui import (
     print_files_updated,
     print_error
 )
-from aye.model.snapshot import apply_updates
+from aye.presenter import diff_presenter
+from aye.model.snapshot import apply_updates, get_diff_base_for_file
 from aye.model.file_processor import filter_unchanged_files, make_paths_relative
 from aye.model.models import LLMResponse
 from aye.model.auth import get_user_config
+from aye.model.autodiff_config import is_autodiff_enabled
 from aye.model.write_validator import (
     check_files_against_ignore_patterns,
     is_strict_mode_enabled,
@@ -52,6 +54,51 @@ def _maybe_print_restore_tip(conf: Any, console: Console) -> None:
     console.print(Padding(msg, (0, 4, 0, 4)))
 
 
+def _run_autodiff(updated_files: List[dict], batch_id: str, conf: Any, console: Console) -> None:
+    """Display diffs for all updated files against their snapshot versions.
+
+    Args:
+        updated_files: List of file dicts with 'file_name' keys
+        batch_id: The batch identifier from apply_updates()
+        conf: Configuration object with root path
+        console: Rich console for output
+    """
+    verbose = getattr(conf, 'verbose', False)
+    debug = get_user_config("debug", "off").lower() == "on"
+
+    console.print(Padding("[dim]───── Auto-diff (autodiff=on) ─────[/]", (1, 0, 0, 0)))
+
+    for item in updated_files:
+        file_name = item.get("file_name")
+        if not file_name:
+            continue
+
+        file_path = Path(file_name)
+
+        # Get the snapshot reference for this file
+        diff_base = get_diff_base_for_file(batch_id, file_path)
+
+        if diff_base is None:
+            if verbose or debug:
+                rprint(f"[yellow]Warning: Could not find snapshot for {file_name}, skipping autodiff[/]")
+            continue
+
+        snapshot_ref, is_git_ref = diff_base
+
+        # Print file header
+        console.print(f"\n[bold cyan]{file_name}[/]")
+
+        try:
+            # show_diff expects: (current_file, snapshot_ref, is_stash_ref)
+            # For autodiff, we diff the current (new) file against the snapshot (old)
+            diff_presenter.show_diff(file_path, snapshot_ref, is_stash_ref=is_git_ref)
+        except Exception as e:
+            if verbose or debug:
+                rprint(f"[yellow]Warning: Could not show diff for {file_name}: {e}[/]")
+
+    console.print(Padding("[dim]───── End auto-diff ─────[/]", (1, 0, 0, 0)))
+
+
 def process_llm_response(
     response: LLMResponse,
     conf: Any,
@@ -118,12 +165,17 @@ def process_llm_response(
         else:
             # Apply updates to the model (Model update)
             try:
-                apply_updates(updated_files, prompt)
+                batch_id = apply_updates(updated_files, prompt)
                 file_names = [item.get("file_name") for item in updated_files if "file_name" in item]
                 if file_names:
                     # Update the view
                     print_files_updated(console, file_names)
                     _maybe_print_restore_tip(conf, console)
+
+                    # Run autodiff if enabled
+                    if is_autodiff_enabled():
+                        _run_autodiff(updated_files, batch_id, conf, console)
+
             except Exception as e:
                 rprint(f"[red]Error applying updates:[/] {e}")
 

{ayechat-0.37.0 → ayechat-0.39.0}/src/aye/controller/llm_invoker.py

@@ -13,12 +13,16 @@ from aye.model.source_collector import collect_sources
 from aye.model.auth import get_user_config
 from aye.model.offline_llm_manager import is_offline_model
 from aye.controller.util import is_truncated_json
-from aye.model.config import SYSTEM_PROMPT, MODELS, DEFAULT_MAX_OUTPUT_TOKENS, DEFAULT_CONTEXT_TARGET_KB
+from aye.model.config import SYSTEM_PROMPT, MODELS, DEFAULT_MAX_OUTPUT_TOKENS, DEFAULT_CONTEXT_TARGET_KB, CONTEXT_HARD_LIMIT_KB
 from aye.model import telemetry
 
 import os
 
 
+def _is_verbose():
+    return get_user_config("verbose", "off").lower() == "on"
+
+
 def _is_debug():
     return get_user_config("debug", "off").lower() == "on"
 
@@ -77,7 +81,7 @@ def _get_context_hard_limit(model_id: str) -> int:
     model_config = _get_model_config(model_id)
     if model_config and "max_prompt_kb" in model_config:
         return model_config["max_prompt_kb"] * 1024
-    return 170 * 1024
+    return CONTEXT_HARD_LIMIT_KB * 1024
 
 
 def _filter_ground_truth(files: Dict[str, str], conf: Any, verbose: bool) -> Dict[str, str]:
@@ -134,6 +138,8 @@ def _get_rag_context_files(prompt: str, conf: Any, verbose: bool) -> Dict[str, s
 
     context_target_size = _get_context_target_size(conf.selected_model)
     context_hard_limit = _get_context_hard_limit(conf.selected_model)
+    #context_target_size = DEFAULT_CONTEXT_TARGET_KB #_get_context_target_size(conf.selected_model)
+    #context_hard_limit = CONTEXT_HARD_LIMIT_KB # _get_context_hard_limit(conf.selected_model)
 
     if _is_debug():
         rprint(f"[yellow]Context target: {context_target_size / 1024:.1f}KB, hard limit: {context_hard_limit / 1024:.1f}KB[/]")
@@ -298,50 +304,54 @@ def invoke_llm(
     model_config = _get_model_config(conf.selected_model)
     max_output_tokens = model_config.get("max_output_tokens", DEFAULT_MAX_OUTPUT_TOKENS) if model_config else DEFAULT_MAX_OUTPUT_TOKENS
 
-    # 1. Try local/offline model plugins first (no streaming UI for local models)
-    local_response = plugin_manager.handle_command("local_model_invoke", {
-        "prompt": prompt,
-        "model_id": conf.selected_model,
-        "source_files": source_files,
-        "chat_id": chat_id,
-        "root": conf.root,
-        "system_prompt": system_prompt,
-        "max_output_tokens": max_output_tokens
-    })
-
-    if local_response is not None:
-        return LLMResponse(
-            summary=local_response.get("summary", ""),
-            updated_files=local_response.get("updated_files", []),
-            chat_id=None,
-            source=LLMSource.LOCAL
-        )
-
-    # 2. API call with spinner + streaming display
-    if _is_debug():
-        print(f"[DEBUG] Processing chat message with chat_id={chat_id or -1}, model={conf.selected_model}")
-
-    telemetry_payload = telemetry.build_payload(top_n=20) if telemetry.is_enabled() else None
-
-    # Create spinner - will be stopped when streaming starts
+    # Create spinner - will be shown for ALL model types (local, databricks, API)
     spinner = StoppableSpinner(
         console,
         messages=DEFAULT_THINKING_MESSAGES,
         interval=15.0
     )
     
+    # For API calls, we also have streaming display
+    streaming_display: Optional[StreamingResponseDisplay] = None
+    
     def stop_spinner():
-        """Callback to stop spinner when first content arrives."""
+        """Callback to stop spinner when first content arrives (for streaming API)."""
         spinner.stop()
-    
-    # Create streaming display with callback to stop spinner on first content
-    streaming_display = StreamingResponseDisplay(on_first_content=stop_spinner)
-    stream_callback = create_streaming_callback(streaming_display)
 
     try:
-        # Start the spinner before the API call
+        # Start the spinner before ANY LLM call (local or API)
         spinner.start()
         
+        # 1. Try local/offline model plugins first
+        local_response = plugin_manager.handle_command("local_model_invoke", {
+            "prompt": prompt,
+            "model_id": conf.selected_model,
+            "source_files": source_files,
+            "chat_id": chat_id,
+            "root": conf.root,
+            "system_prompt": system_prompt,
+            "max_output_tokens": max_output_tokens
+        })
+
+        if local_response is not None:
+            # Local model handled the request - spinner will be stopped in finally block
+            return LLMResponse(
+                summary=local_response.get("summary", ""),
+                updated_files=local_response.get("updated_files", []),
+                chat_id=None,
+                source=LLMSource.LOCAL
+            )
+
+        # 2. API call with streaming display
+        if _is_debug():
+            print(f"[DEBUG] Processing chat message with chat_id={chat_id or -1}, model={conf.selected_model}")
+
+        telemetry_payload = telemetry.build_payload(top_n=20) if telemetry.is_enabled() else None
+
+        # Create streaming display with callback to stop spinner on first content
+        streaming_display = StreamingResponseDisplay(on_first_content=stop_spinner)
+        stream_callback = create_streaming_callback(streaming_display)
+
         api_resp = cli_invoke(
             message=prompt,
             chat_id=chat_id or -1,
@@ -352,29 +362,29 @@ def invoke_llm(
             telemetry=telemetry_payload,
             on_stream_update=stream_callback
         )
-    finally:
-        # Ensure spinner is stopped (in case no streaming content was received)
-        spinner.stop()
-        
-        # Always stop the streaming display when done
-        if streaming_display.is_active():
-            streaming_display.stop()
 
-    if telemetry_payload is not None:
-        telemetry.reset()
+        if telemetry_payload is not None:
+            telemetry.reset()
 
-    if _is_debug():
-        print(f"[DEBUG] Chat message processed, response keys: {api_resp.keys() if api_resp else 'None'}")
+        if _is_debug():
+            print(f"[DEBUG] Chat message processed, response keys: {api_resp.keys() if api_resp else 'None'}")
 
-    # Check if we already displayed the response via streaming
-    streamed_summary = bool(api_resp.get("_streamed_summary")) if isinstance(api_resp, dict) else False
+        # Check if we already displayed the response via streaming
+        streamed_summary = bool(api_resp.get("_streamed_summary")) if isinstance(api_resp, dict) else False
 
-    # 3. Parse API response
-    assistant_resp, new_chat_id = _parse_api_response(api_resp)
+        # 3. Parse API response
+        assistant_resp, new_chat_id = _parse_api_response(api_resp)
 
-    return LLMResponse(
-        summary="" if streamed_summary else assistant_resp.get("answer_summary", ""),
-        updated_files=assistant_resp.get("source_files", []),
-        chat_id=new_chat_id,
-        source=LLMSource.API
-    )
+        return LLMResponse(
+            summary="" if streamed_summary else assistant_resp.get("answer_summary", ""),
+            updated_files=assistant_resp.get("source_files", []),
+            chat_id=new_chat_id,
+            source=LLMSource.API
+        )
+    finally:
+        # Ensure spinner is stopped for ALL code paths (local model, API, or error)
+        spinner.stop()
+        
+        # Stop the streaming display if it was created and is active
+        if streaming_display is not None and streaming_display.is_active():
+            streaming_display.stop()

{ayechat-0.37.0 → ayechat-0.39.0}/src/aye/controller/repl.py

@@ -41,6 +41,8 @@ from aye.controller.command_handlers import (
     handle_completion_command,
     handle_with_command,
     handle_blog_command,
+    handle_llm_command,
+    handle_autodiff_command,
 )
 
 DEBUG = False
@@ -284,7 +286,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", "autodiff", "completion", "exit", "quit", ":q", "help", "cd", "db", "llm"]
 
     # Get the completion style setting
     completion_style = get_user_config("completion_style", "readline").lower()
@@ -415,6 +417,9 @@ def chat_repl(conf: Any) -> None:
                 elif lowered_first == "debug":
                     telemetry.record_command("debug", has_args=len(tokens) > 1, prefix=_AYE_PREFIX)
                     handle_debug_command(tokens)
+                elif lowered_first == "autodiff":
+                    telemetry.record_command("autodiff", has_args=len(tokens) > 1, prefix=_AYE_PREFIX)
+                    handle_autodiff_command(tokens)
                 elif lowered_first == "completion":
                     telemetry.record_command("completion", has_args=len(tokens) > 1, prefix=_AYE_PREFIX)
                     new_style = handle_completion_command(tokens)
@@ -429,6 +434,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>")

{ayechat-0.37.0 → ayechat-0.39.0}/src/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)
-

ayechat-0.39.0/src/aye/model/autodiff_config.py

@@ -0,0 +1,32 @@
+"""Autodiff configuration for automatic diff display after LLM changes.
+
+This module provides functionality to check if autodiff mode is enabled.
+When enabled, diffs are automatically displayed for every file modified
+by an LLM response.
+
+See: autodiff.md for the full design plan.
+"""
+
+from aye.model.auth import get_user_config
+
+
+# Config key for autodiff mode
+AUTODIFF_KEY = "autodiff"
+
+
+def is_autodiff_enabled() -> bool:
+    """Check if autodiff mode is enabled.
+
+    When enabled, diffs are automatically displayed for every file
+    modified by an LLM response, immediately after the optimistic
+    write is applied.
+
+    Can be set via:
+    - Environment variable: AYE_AUTODIFF=on
+    - Config file (~/.ayecfg): autodiff=on
+
+    Returns:
+        True if autodiff mode is enabled, False otherwise (default)
+    """
+    value = get_user_config(AUTODIFF_KEY, "off")
+    return str(value).lower() in ("on", "true", "1", "yes")