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

aye/controller/command_handlers.py

@@ -145,6 +145,25 @@ 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.
 
@@ -264,9 +283,9 @@ def handle_llm_command(session: Optional[PromptSession], tokens: list[str]) -> N
 
     # Show status message
     if final_url and final_key:
-        rprint("\n[green] OpenAI-compatible endpoint is configured and active.[/]")
+        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.[/]")
+        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]:

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}")
 

aye/controller/repl.py

@@ -42,6 +42,7 @@ from aye.controller.command_handlers import (
     handle_with_command,
     handle_blog_command,
     handle_llm_command,
+    handle_autodiff_command,
 )
 
 DEBUG = False
@@ -285,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", "llm"]
+    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()
@@ -416,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)

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")

aye/model/snapshot/__init__.py

@@ -7,6 +7,7 @@ Note:
 - Git stash snapshots (GitStashBackend) are intentionally NOT used.
 """
 
+import json
 import subprocess
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple, Union
@@ -32,6 +33,7 @@ __all__ = [
     "delete_snapshot",
     "prune_snapshots",
     "cleanup_snapshots",
+    "get_diff_base_for_file",
     # Utilities
     "get_backend",
     "reset_backend",
@@ -145,6 +147,102 @@ def cleanup_snapshots(older_than_days: int = 30) -> int:
     return get_backend().cleanup_snapshots(older_than_days)
 
 
+def get_diff_base_for_file(batch_id: str, file_path: Path) -> Optional[Tuple[str, bool]]:
+    """Return the snapshot reference for a file in a given batch.
+
+    This provides a backend-agnostic way to get a reference suitable for
+    diff_presenter.show_diff().
+
+    Args:
+        batch_id: The batch identifier returned by apply_updates()
+        file_path: The file path to get the snapshot reference for
+
+    Returns:
+        Tuple of (snapshot_ref, is_git_ref) where:
+        - snapshot_ref: For FileBasedBackend, a filesystem path to the snapshot file.
+                        For GitRefBackend, a 'refname:repo_rel_path' string.
+        - is_git_ref: True if snapshot_ref is a git ref format, False for filesystem path.
+        Returns None if the file is not found in the snapshot.
+    """
+    backend = get_backend()
+
+    if isinstance(backend, FileBasedBackend):
+        return _get_diff_base_file_backend(backend, batch_id, file_path)
+    elif isinstance(backend, GitRefBackend):
+        return _get_diff_base_git_backend(backend, batch_id, file_path)
+
+    return None
+
+
+def _get_diff_base_file_backend(
+    backend: FileBasedBackend, batch_id: str, file_path: Path
+) -> Optional[Tuple[str, bool]]:
+    """Get diff base for FileBasedBackend.
+
+    Reads the metadata.json from the snapshot batch directory to find
+    the snapshot path for the given file.
+    """
+    # Find the batch directory matching the batch_id
+    # batch_id format is like "001_20231201T120000"
+    batch_dir = None
+    if backend.snap_root.is_dir():
+        for dir_path in backend.snap_root.iterdir():
+            if dir_path.is_dir() and dir_path.name == batch_id:
+                batch_dir = dir_path
+                break
+
+    if batch_dir is None:
+        return None
+
+    meta_file = batch_dir / "metadata.json"
+    if not meta_file.is_file():
+        return None
+
+    try:
+        meta = json.loads(meta_file.read_text(encoding="utf-8"))
+    except (json.JSONDecodeError, OSError):
+        return None
+
+    # Resolve the target file path for comparison
+    file_resolved = file_path.resolve()
+
+    # Find the matching entry in metadata
+    for entry in meta.get("files", []):
+        original_path = entry.get("original")
+        snapshot_path = entry.get("snapshot")
+
+        if not original_path or not snapshot_path:
+            continue
+
+        if Path(original_path).resolve() == file_resolved:
+            # Return the snapshot path as a string, not a git ref
+            return (snapshot_path, False)
+
+    return None
+
+
+def _get_diff_base_git_backend(
+    backend: GitRefBackend, batch_id: str, file_path: Path
+) -> Optional[Tuple[str, bool]]:
+    """Get diff base for GitRefBackend.
+
+    Returns a ref:path format suitable for git show.
+    """
+    # Construct the refname from batch_id
+    refname = f"{backend.REF_NAMESPACE}/{batch_id}"
+
+    # Get repo-relative path
+    try:
+        file_resolved = file_path.resolve()
+        rel_path = file_resolved.relative_to(backend.git_root).as_posix()
+    except ValueError:
+        # File is outside git root
+        return None
+
+    # Return the git ref format
+    return (f"{refname}:{rel_path}", True)
+
+
 # ------------------------------------------------------------------
 # Legacy helper functions (for backward compatibility with tests)
 # ------------------------------------------------------------------

aye/presenter/repl_ui.py

@@ -88,14 +88,15 @@ def print_help_message():
         # Some commands are intentionally undocumented: keep them as such.
         ("@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\")."),
+        ("new", "Start a new chat session (if you want to change the subject)"),
+        (r"restore, undo \[id] \[file]", "Revert changes to the last state, a specific snapshot `id`, or for a single `file`."),
+        ("history", "Show snapshot history"),
+        (r"diff <file> \[snapshot_id]", "Show diff of file with the latest snapshot, or a specified snapshot"),
         ("model", "Select a different model. Selection will persist between sessions."),
+        (r"autodiff \[on|off]", "Toggle automatic diff display after LLM file updates (off by default, persists 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)"),
-        ("history", "Show snapshot history"),
-        (r"diff <file> \[snapshot_id]", "Show diff of file with the latest snapshot, or a specified snapshot"),
-        (r"restore, undo \[id] \[file]", "Revert changes to the last state, a specific snapshot `id`, or for a single `file`."),
         ("keep [N]", "Keep only N most recent snapshots (10 by default)"),
         ("exit, quit, Ctrl+D", "Exit the chat session"),
         ("help", "Show this help message"),

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

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

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

@@ -3,18 +3,19 @@ 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=72dA9x4vHOxsSuWb1wEX2xTxL4uCFuiSPL4-VpEPwyA,16728
+aye/controller/command_handlers.py,sha256=S_HOYY91lbLkGtCkby28T8qUFPNtfy-CbOGn9VGZtTo,17446
 aye/controller/commands.py,sha256=sXmK_sgNBrw9Fs7mKcr93-wsu740ZlvWSisQfS-1EUE,12278
-aye/controller/llm_handler.py,sha256=BSBab6onF9BYiFndYW1647eEy377Vt52trzu0Qjm4bQ,5075
+aye/controller/llm_handler.py,sha256=gY3X2rHcvhPp8iqp_Vor7QpbwHltmcm_Uu9MWF1Z598,7120
 aye/controller/llm_invoker.py,sha256=p_Vk2a3YrWKwDupLfSVRinR5llDfq1Fb_f7WrYozK6M,14127
 aye/controller/plugin_manager.py,sha256=9ZuITyA5sQJJJU-IntLQ1SsxXsDnbgZKPOF4e9VmsEU,3018
-aye/controller/repl.py,sha256=fTYEdX1PAds49pRBVgvyi-4lIfArR8gjBTp_shwx-K0,26782
+aye/controller/repl.py,sha256=nSlzAHD8MQlQ7vnpBTLVzA1jQ8jkNjxUnUoqE31chxY,27028
 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=ozV_uQxdqXtUoWO3nZwpzVnDOIfnRAmSMC6W0N724vE,4800
+aye/model/autodiff_config.py,sha256=b8pyudkJFYXF5JWPxft0bH5uazeCab9i-A11dNt1d7U,931
 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
@@ -34,7 +35,7 @@ aye/model/index_manager/index_manager_executor.py,sha256=rSI8OX9bFld5ta-ajfT8eWU
 aye/model/index_manager/index_manager_file_ops.py,sha256=petgGU0ZtXOvLkHAQuzSbzqNIew5HygBOAy4yG6rTlk,7656
 aye/model/index_manager/index_manager_state.py,sha256=LflyC8UbE8fT7Tyag1ueh7TBisPa15cRhs0JI1-jVEw,17587
 aye/model/index_manager/index_manager_utils.py,sha256=B4NCYuScxQcG46WoUEx8mxfWKDjTlrSf3coRQCydiBM,4615
-aye/model/snapshot/__init__.py,sha256=Lqq-W5RBGxu3_7O5DzW_qY5iOF9MhogA46BAqEdu6Mw,6958
+aye/model/snapshot/__init__.py,sha256=XyEaYF0WS0wuGZI9t2vcA1UUxWbwj5CEkwKFYUSQwKc,10152
 aye/model/snapshot/base.py,sha256=5fzxM_85avxHwocv-w6PwlxDPzvMdTbuoUE_9P6D844,2009
 aye/model/snapshot/file_backend.py,sha256=hTlqaBCiMe9YHFZs7gcyXMVRTsR0pY23LQN15D0ZOY0,10757
 aye/model/snapshot/git_ref_backend.py,sha256=mY4yJiZ0ER4pgMzCgiTPQiKoVGCgWravNrPFue_zwLw,21356
@@ -50,12 +51,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=5nAv8qLo3azDuoGYAxGdK2SEwowXPoHSEruupvS6jy8,8023
+aye/presenter/repl_ui.py,sha256=PVENlAQM_tm_k2dANsmQH6I8ATMVXhrdj_hNzc38pSw,8156
 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.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,,
+ayechat_dev-0.36.9.20260204011001.dist-info/licenses/LICENSE,sha256=U1ou6lkMKmPo16-E9YowIu3goU7sOWKUprGo0AOA72s,1065
+ayechat_dev-0.36.9.20260204011001.dist-info/METADATA,sha256=c6g026ZNTxo1IUHsgXnRCWdyJZnX_ddUTwwo4i4mvtc,7718
+ayechat_dev-0.36.9.20260204011001.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ayechat_dev-0.36.9.20260204011001.dist-info/entry_points.txt,sha256=KGsOma6szoefNN6vHozg3Pbf1fjZ7ZbmwrOiVwBd0Ik,41
+ayechat_dev-0.36.9.20260204011001.dist-info/top_level.txt,sha256=7WZL0LOx4-GKKvgU1mtI5s4Dhk2OdieVZZvVnxFJHr8,4
+ayechat_dev-0.36.9.20260204011001.dist-info/RECORD,,