ayechat-dev 0.36.9.20260204003405__tar.gz → 0.36.9.20260204011001__tar.gz

{ayechat_dev-0.36.9.20260204003405/src/ayechat_dev.egg-info → ayechat_dev-0.36.9.20260204011001}/PKG-INFO

@@ -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 → ayechat_dev-0.36.9.20260204011001}/pyproject.toml

@@ -8,7 +8,7 @@ description = "Aye Chat: Terminal-first AI Code Generator"
 readme = "README.md"
 requires-python = ">=3.10, <3.14"
 dependencies = [ "typer>=0.20.0", "httpx>=0.28.1", "keyring>=25.7.0", "prompt-toolkit>=3.0.52", "pathspec>=0.12.1", "chromadb>=1.3.5", "rapidfuzz",]
-version = "0.36.9.20260204003405"
+version = "0.36.9.20260204011001"
 [[project.authors]]
 name = "Acrotron, Inc."
 email = "info@acrotron.com"

{ayechat_dev-0.36.9.20260204003405 → ayechat_dev-0.36.9.20260204011001}/src/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]:

{ayechat_dev-0.36.9.20260204003405 → ayechat_dev-0.36.9.20260204011001}/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_dev-0.36.9.20260204003405 → ayechat_dev-0.36.9.20260204011001}/src/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)

ayechat_dev-0.36.9.20260204011001/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")

{ayechat_dev-0.36.9.20260204003405 → ayechat_dev-0.36.9.20260204011001}/src/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)
 # ------------------------------------------------------------------

{ayechat_dev-0.36.9.20260204003405 → ayechat_dev-0.36.9.20260204011001}/src/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 → ayechat_dev-0.36.9.20260204011001/src/ayechat_dev.egg-info}/PKG-INFO

@@ -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 → ayechat_dev-0.36.9.20260204011001}/src/ayechat_dev.egg-info/SOURCES.txt

@@ -49,6 +49,7 @@ src/aye/model/__init__.py
 src/aye/model/api.py
 src/aye/model/ast_chunker.py
 src/aye/model/auth.py
+src/aye/model/autodiff_config.py
 src/aye/model/config.py
 src/aye/model/download_plugins.py
 src/aye/model/file_processor.py