kolega-code 0.14.0__tar.gz → 0.16.0__tar.gz

{kolega_code-0.14.0 → kolega_code-0.16.0}/.gitignore

@@ -15,3 +15,4 @@ build/
 coverage.xml
 htmlcov/
 docs/kolega_reskin_project/
+docs/agent-harness-frontier-roadmap.md

{kolega_code-0.14.0 → kolega_code-0.16.0}/CHANGELOG.md

@@ -6,6 +6,51 @@ This project uses GitHub Releases for detailed generated release notes. This fil
 
 ## Unreleased
 
+## 0.16.0 - 2026-07-09
+
+### Added
+
+- Added support for xAI Grok 4.5 (`grok-4.5`): 500K-token context, vision input,
+  and `low`/`medium`/`high` reasoning effort. Selectable in the Settings UI,
+  `/model` picker, CLI flags, and env vars; now the default `xai` model.
+- Send a stable `x-grok-conv-id` header on xAI Chat Completions requests so
+  multi-turn agent loops pin to a cache-warm server (xAI prompt-caching guidance).
+- Added session-scoped workspace snapshots for agent file mutations: `snapshot`
+  (list/show/create/restore, including `snapshot_id='latest'` undo) and `resolve`
+  for applying or discarding pending `lsp_edit` preview actions. Restore refuses
+  on post-snapshot drift unless `force=true`.
+- Improved TUI usability inside tmux/screen when Shift-modified keys never reach
+  the app: `/attach` with no path pastes an image from the system clipboard,
+  `Alt+V` is a portable image-paste binding alongside `Ctrl+Shift+V`, startup
+  help highlights `Ctrl+J` / `/plan` / `/build` fallbacks, and a one-time
+  startup hint appears under tmux. Documented optional tmux extended-keys config
+  under Troubleshooting.
+
+### Fixed
+
+- Prevented a TUI crash when pasting long multi-line text into the composer
+  (`OSError: File name too long` from treating paste content as a path).
+- Stopped advertising host lifecycle `initialize` as a model-callable tool while
+  keeping it available for CLI/TUI startup (LSP setup).
+- Bumped `lxml-html-clean` to 0.4.5 (and transitive `lxml` to 6.1.1) to clear
+  GHSA-4jhm-jv67-739f from the dependency audit.
+
+## 0.15.0 - 2026-07-07
+
+### Added
+
+- Added Language Server Protocol (LSP) integration with an agent-callable
+  read-only `lsp` tool (diagnostics, go-to-definition, references, hover,
+  document/workspace symbols, and status) and a trusted `lsp_edit` tool for
+  rename, formatting, and code actions. The `edit`, `multi_edit`, and `write`
+  tools now append LSP diagnostics to their results, with a `/lsp` status
+  command, detected-language status in the TUI wordmark, and a Settings toggle.
+- Added bounded skill metadata rendering with a context-aware token budget.
+  The prompt catalog now uses skill name and description only, descriptions are
+  truncated before skills are omitted, and `list_skills` is queryable with
+  `max_results`.
+- Documented composable workflow shapes in the Gigacode orchestration guide.
+
 ## 0.14.0 - 2026-07-03
 
 ### Added

{kolega_code-0.14.0 → kolega_code-0.16.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kolega-code
-Version: 0.14.0
+Version: 0.16.0
 Summary: Multi-agent coding in the terminal
 Project-URL: Homepage, https://kolega-ai.github.io/kolega-code/
 Project-URL: Documentation, https://kolega-ai.github.io/kolega-code/
@@ -234,7 +234,7 @@ Requires-Dist: google-genai==2.0.1
 Requires-Dist: httpx==0.28.1
 Requires-Dist: jinja2==3.1.6
 Requires-Dist: langfuse==3.14.5
-Requires-Dist: lxml-html-clean==0.4.4
+Requires-Dist: lxml-html-clean==0.4.5
 Requires-Dist: mcp>=1.21.2
 Requires-Dist: nest-asyncio==1.6.0
 Requires-Dist: openai==1.109.1

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/astro.config.mjs

@@ -68,6 +68,10 @@ export default defineConfig({
               label: "MCP Servers",
               slug: "configuration/mcp",
             },
+            {
+              label: "Language Servers",
+              slug: "configuration/lsp",
+            },
             {
               label: "Environment Variables",
               slug: "configuration/environment-variables",
@@ -116,6 +120,10 @@ export default defineConfig({
               label: "Diagnostics & Bug Reports",
               slug: "troubleshooting/diagnostics",
             },
+            {
+              label: "Terminal & tmux shortcuts",
+              slug: "troubleshooting/terminal-tmux",
+            },
           ],
         },
         {

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/assets/coverage.svg

@@ -1 +1 @@
-<svg xmlns="http://www.w3.org/2000/svg" width="114" height="20" role="img" aria-label="coverage: 80.35%"><title>coverage: 80.35%</title><filter id="blur"><feGaussianBlur stdDeviation="16"/></filter><linearGradient id="s" x2="0" y2="100%"><stop offset="0" stop-color="#bbb" stop-opacity=".1"/><stop offset="1" stop-opacity=".1"/></linearGradient><clipPath id="r"><rect width="114" height="20" rx="3"/></clipPath><g clip-path="url(#r)"><rect width="61" height="20" fill="#555"/><rect x="61" width="53" height="20" fill="#67ac09"/><rect width="114" height="20" fill="url(#s)"/></g><g fill="#fff" text-anchor="middle" font-family="Verdana,Geneva,DejaVu Sans,sans-serif" text-rendering="geometricPrecision" font-size="110"><g transform="scale(.1)"><g aria-hidden="true" fill="#010101"><text x="315" y="150" fill-opacity=".8" filter="url(#blur)" textLength="510">coverage</text><text x="315" y="150" fill-opacity=".3" textLength="510">coverage</text></g><text x="315" y="140" textLength="510">coverage</text></g><g transform="scale(.1)"><g aria-hidden="true" fill="#010101"><text x="865" y="150" fill-opacity=".8" filter="url(#blur)" textLength="430">80.35%</text><text x="865" y="150" fill-opacity=".3" textLength="430">80.35%</text></g><text x="865" y="140" textLength="430">80.35%</text></g></g></svg>
+<svg xmlns="http://www.w3.org/2000/svg" width="114" height="20" role="img" aria-label="coverage: 79.93%"><title>coverage: 79.93%</title><filter id="blur"><feGaussianBlur stdDeviation="16"/></filter><linearGradient id="s" x2="0" y2="100%"><stop offset="0" stop-color="#bbb" stop-opacity=".1"/><stop offset="1" stop-opacity=".1"/></linearGradient><clipPath id="r"><rect width="114" height="20" rx="3"/></clipPath><g clip-path="url(#r)"><rect width="61" height="20" fill="#555"/><rect x="61" width="53" height="20" fill="#67ac09"/><rect width="114" height="20" fill="url(#s)"/></g><g fill="#fff" text-anchor="middle" font-family="Verdana,Geneva,DejaVu Sans,sans-serif" text-rendering="geometricPrecision" font-size="110"><g transform="scale(.1)"><g aria-hidden="true" fill="#010101"><text x="315" y="150" fill-opacity=".8" filter="url(#blur)" textLength="510">coverage</text><text x="315" y="150" fill-opacity=".3" textLength="510">coverage</text></g><text x="315" y="140" textLength="510">coverage</text></g><g transform="scale(.1)"><g aria-hidden="true" fill="#010101"><text x="865" y="150" fill-opacity=".8" filter="url(#blur)" textLength="430">79.93%</text><text x="865" y="150" fill-opacity=".3" textLength="430">79.93%</text></g><text x="865" y="140" textLength="430">79.93%</text></g></g></svg>

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/content/docs/concepts/tools.md

@@ -18,6 +18,9 @@ Read, search, and edit your project:
 - `read_entire_file`, `read_file_section` — read file contents.
 - `find_files_by_pattern` — glob-based file search.
 - `search_codebase` — search the codebase by regular expression (ripgrep/grep), e.g. `foo|bar`.
+- `lsp` — read diagnostics, symbols, definitions,
+  references, hover text, call hierarchy, and code action metadata from
+  [configured language servers](../../configuration/lsp/).
 - Create and edit files — create new files and apply precise edits.
 
 ### Terminal

kolega_code-0.16.0/docs/src/content/docs/configuration/lsp.md

@@ -0,0 +1,121 @@
+---
+title: Language servers
+description: Configure LSP diagnostics, code intelligence, and trusted LSP edits.
+---
+
+Kolega Code can start local Language Server Protocol (LSP) servers for detected
+project languages. LSP is enabled by default and is used for:
+
+- diagnostics after `edit`, `multi_edit`, and `write`;
+- the generic read-only `lsp` tool;
+- the permission-gated `lsp_edit` tool;
+- `/lsp` and the Settings tab status display.
+
+The `lsp` tool remains read-only. Mutating LSP operations are exposed through
+`lsp_edit`, which is treated like `edit`, `multi_edit`, and `write` for
+permission prompts. Kolega Code does not expose raw arbitrary LSP requests.
+
+## Tool operations
+
+The generic `lsp` tool supports:
+
+- `diagnostics` for file errors, warnings, and hints;
+- `definition`, `type_definition`, `implementation`, `references`, and `hover`;
+- `document_symbols` and `workspace_symbols`;
+- `call_hierarchy` for incoming and outgoing calls;
+- `code_actions` to list available quick fixes and refactors without applying them;
+- `status`, `capabilities`, and `reload`.
+
+Position operations use `path`, a 1-based `line`, and a `symbol` on that line.
+Use `symbol#2` to select the second occurrence of a symbol on the same line.
+`code_actions` includes an `action_id` for each action so a later `lsp_edit`
+call can apply the exact server-provided action.
+
+The mutating `lsp_edit` tool supports:
+
+- `rename` for symbol rename;
+- `rename_file` for file moves with `workspace/willRenameFiles` and
+  `workspace/didRenameFiles` notifications;
+- `format_document` and `format_range`;
+- `apply_code_action` by `action_id`, title/query, or index.
+
+`lsp_edit` applies server-provided `WorkspaceEdit` payloads only. If the server
+does not advertise or return an edit for an operation, Kolega Code reports that
+the operation is unsupported instead of constructing a manual fallback edit.
+Pass `apply: false` to preview the LSP edit without writing files.
+
+## Project configuration
+
+Create `<project>/.kolega/lsp.json` to override LSP behavior for a repository.
+Project LSP config is trusted input because custom servers can execute local
+binaries. Kolega Code ignores this file unless the session is started with
+`--trust-lsp` or the project is listed in trusted LSP projects in user settings.
+
+```json
+{
+  "enabled": true,
+  "auto_diagnostics_on_edit": true,
+  "max_diagnostics": 20,
+  "auto_fallback": true,
+  "prompt_on_missing": true,
+  "disabled_languages": [],
+  "preferences": {
+    "python": "pyright"
+  },
+  "diagnostic_servers": ["ruff-lsp"],
+  "initialization_options": {
+    "pyright": {
+      "python": {
+        "analysis": {
+          "typeCheckingMode": "basic"
+        }
+      }
+    }
+  },
+  "workspace_configuration": {
+    "pyright": {
+      "python": {
+        "analysis": {
+          "typeCheckingMode": "basic"
+        }
+      }
+    }
+  },
+  "servers": {
+    "custom-python-lsp": {
+      "bin": "custom-lsp",
+      "args": ["--stdio"],
+      "languages": ["python"],
+      "env": {
+        "CUSTOM_LSP_MODE": "project"
+      }
+    }
+  }
+}
+```
+
+The project file cannot change the user's global LSP kill switch: `enabled` is
+accepted in user settings, but ignored from `.kolega/lsp.json`.
+
+`preferences` chooses the preferred server for a language. If
+`auto_fallback` is `false`, Kolega Code checks only the preferred server before
+reporting the language as missing. If it is `true`, installed alternatives can be
+used automatically.
+
+`diagnostic_servers` starts additional servers for diagnostics alongside the
+primary server. This is useful for setups such as Pyright plus Ruff.
+
+`initialization_options` is sent in the LSP `initialize` request. Some servers
+also ask for `workspace/configuration`; use `workspace_configuration` for those
+responses, keyed first by server name and then by requested section.
+
+## Status and troubleshooting
+
+Run `/lsp` in the TUI to see detected languages, missing servers, active
+sessions, and recent diagnostic counts. The Settings tab has a global LSP toggle;
+changing it requires restarting the agent session.
+
+If a server is missing, `/lsp` shows the server name and install guidance from
+Kolega Code's bundled registry. You can suppress a language with
+`disabled_languages` or point it at a custom server with `servers` and
+`preferences`.

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/content/docs/configuration/providers-and-models.mdx

@@ -184,6 +184,8 @@ models in the TUI.
 | Ollama Cloud `gpt-oss:*` | `low`, `medium`, `high` | `medium` |
 | Google `gemini-2.5-pro` | `auto`, `low`, `medium`, `high` | `medium` |
 | OpenAI reasoning models | `low`, `medium`, `high` | `medium` |
+| xAI `grok-4.5` | `low`, `medium`, `high` | `medium` |
+| xAI `grok-4.3` | `none`, `low`, `medium`, `high` | `low` |
 
 ## Check what's active
 

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/content/docs/gigacode.mdx

@@ -34,6 +34,9 @@ can:
   modules, services, issues, or review targets.
 - **Run judge panels** — ask several agents to evaluate a change or plan from
   different angles, then reconcile disagreements.
+- **Compose workflow shapes** — use sweeps, matrices, funnels, pipelines, triage
+  loops, and adversarial reviews when the task needs more structure than simple
+  parallelism.
 - **Preserve artifacts** — save the workflow script, result files, transcripts, raw
   JSONL, and resume journal so the run is inspectable after the fact.
 
@@ -99,6 +102,13 @@ findings = await parallel([
 return findings
 ```
 
+The stage names in a workflow are not fixed runtime types. The agent composes
+shapes from the task: a broad audit might map surfaces, sweep reviewers across
+them, verify findings adversarially, and synthesize; a planning task might funnel
+unknowns through research and option comparison; a debugging task might cluster
+failures, investigate root causes, test targeted fixes, and run a regression
+review. All of those shapes are built from the same primitives above.
+
 <Aside type="note">
 Workflows are deterministic and **resumable**. Each run is saved under Kolega
 Code's state directory (`workflows/<run-id>/`) with its script, full result files,

kolega_code-0.16.0/docs/src/content/docs/troubleshooting/terminal-tmux.md

@@ -0,0 +1,67 @@
+---
+title: Terminal & tmux shortcuts
+description: Why Shift chords fail in tmux/screen and the portable fallbacks Kolega Code supports.
+---
+
+Kolega Code’s TUI uses some Shift-modified shortcuts (`Shift+Enter`, `Shift+Tab`,
+`Ctrl+Shift+V`). In a modern terminal with the [Kitty keyboard protocol](https://sw.kovidgoyal.net/kitty/keyboard-protocol/),
+those chords reach the app. Inside **tmux** or **GNU screen**, the multiplexer
+often sits between the terminal and the app and either drops modifiers or
+never forwards the extended key sequences — so Shift chords look like plain
+keys, or never arrive at all.
+
+Kolega Code cannot invent modifier information the terminal never sends. Instead
+it provides **portable fallbacks** that work without Shift, plus optional tmux
+config if you want the original chords restored.
+
+## Portable shortcuts
+
+| Action | Preferred (capable terminals) | Portable fallback |
+| --- | --- | --- |
+| Send prompt | `Enter` | — |
+| Insert newline | `Shift+Enter` | `Ctrl+J` or `Ctrl+Enter` |
+| Toggle Plan ⇄ Build | `Shift+Tab` | `/plan` or `/build` |
+| Paste image from system clipboard | `Ctrl+Shift+V` | `Alt+V` (Option+V on macOS when the terminal treats Option as Meta) or `/attach` with no path |
+| Attach image from a file | — | `/attach <path>` or `@image.png` |
+
+On session start inside tmux/screen, the startup block prints a short reminder
+of these fallbacks.
+
+### Image paste notes
+
+- `/attach` with **no path** reads an image from the **system** clipboard (same
+  path as `Ctrl+Shift+V` / `Alt+V`). This is the most reliable option under
+  plain tmux.
+- Clipboard image reading uses host tools independent of tmux:
+  - **macOS:** `osascript` (built-in), or `pngpaste` if installed
+  - **Linux:** `xclip` or `wl-paste`
+  - **Windows / WSL:** PowerShell clipboard APIs
+- If Option+V types a special character (e.g. `√`) instead of pasting an image,
+  your terminal is not sending Alt as a modifier. Prefer `/attach`, or enable
+  “Option as Meta” / “Left Option key acts as +Esc” in the terminal settings.
+
+## Optional: restore Shift+Enter in tmux
+
+With **tmux 3.2+** and an outer terminal that supports extended keys, add to
+`~/.tmux.conf`:
+
+```tmux
+set -g extended-keys always
+set -g extended-keys-format csi-u
+set -as terminal-features 'xterm*:extkeys'
+```
+
+Then reload (`tmux source-file ~/.tmux.conf`) or restart tmux.
+
+Also make sure `default-terminal` / `terminal-overrides` match the outer
+terminal’s `$TERM` family (`xterm-256color`, `xterm-ghostty`, etc.) so the
+`extkeys` feature actually applies.
+
+After that, `Shift+Enter` and similar chords often work again inside Kolega Code.
+If they still do not, keep using the portable fallbacks above.
+
+## Related
+
+- [Chat Composer](../../tui/composer/) — full composer key reference
+- [Interface Tour](../../tui/interface/) — key bindings at a glance
+- [Slash Commands](../../tui/slash-commands/) — `/attach`, `/plan`, `/build`

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/content/docs/tui/composer.md

@@ -13,17 +13,24 @@ multi-line input, and drives the option lists the agent shows you.
 | --- | --- |
 | `Enter` | Send the prompt |
 | `Shift+Enter` | Insert a newline (keep typing) |
-| `Ctrl+J` | Insert a newline (alternative) |
+| `Ctrl+J` / `Ctrl+Enter` | Insert a newline (portable alternatives; prefer these inside tmux) |
 | `Ctrl+L` | Select all composer text |
 | `Cmd+A` / `⌘A` | Select all composer text on macOS, when forwarded by the terminal |
 | `Ctrl+A` | Move to the start of the current line (intentionally not select-all) |
 | `Ctrl+U` | Delete backward to the start of the line; at line start, delete the previous line |
+| `Ctrl+Shift+V` | Paste an image from the system clipboard |
+| `Alt+V` | Paste an image from the system clipboard (portable alternative when Shift chords fail) |
 
 `Ctrl+L` is the portable select-all shortcut because `Ctrl+A` is preserved for
 terminal-style line navigation. Some macOS terminal emulators reserve `Cmd+A` for
 terminal scrollback selection before the TUI can see it; `Ctrl+L` still works in
 the composer.
 
+Inside **tmux** or **screen**, Shift-modified keys often never reach the app.
+Use `Ctrl+J` for newlines and `Alt+V` or `/attach` for images. See
+[Terminal & tmux shortcuts](../../troubleshooting/terminal-tmux/) for optional
+tmux config to restore Shift+Enter.
+
 ## Queuing follow-up prompts
 
 You can keep typing while the agent is still generating. Pressing `Enter` during
@@ -45,6 +52,19 @@ If you cancel the active turn with `Ctrl+C` or `Escape`, queued follow-ups are
 restored into the composer instead of being silently sent. To discard pending
 follow-ups without stopping the active turn, run [`/queue-clear`](../slash-commands/).
 
+## Attaching images
+
+You can attach images for vision-capable models in several ways:
+
+| Method | How |
+| --- | --- |
+| System clipboard | `Ctrl+Shift+V` or `Alt+V`, or run `/attach` with no path |
+| File path | `/attach path/to/image.png` |
+| File mention | `@screenshot.png` (supported image extensions only) |
+
+Use `/detach` to clear pending attachments before sending. On a non-vision
+model, Kolega Code warns and blocks the send until you detach or switch models.
+
 ## File mentions with `@`
 
 Type `@` to reference a file. A fuzzy-searchable dropdown of project files appears

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/content/docs/tui/interface.md

@@ -82,14 +82,16 @@ the turn keeps running while you look around.
 
 | Keys | Action |
 | --- | --- |
-| `Shift+Tab` | Toggle Build ⇄ Plan mode |
+| `Shift+Tab` | Toggle Build ⇄ Plan mode (`/plan` / `/build` if Shift is unavailable) |
 | `Ctrl+P` | Toggle shell/edit permissions between Ask ⇄ Auto |
 | `Ctrl+O` | Show or hide the side panel |
 | `Ctrl+G` | Open the sub-agent inspector |
 | `Enter` | Send the prompt |
 | `Shift+Enter` / `Ctrl+J` | Insert a newline |
+| `Ctrl+Shift+V` / `Alt+V` | Paste an image from the system clipboard (`/attach` also works) |
 | `Ctrl+C` / `Escape` | Cancel the current generation |
 | `Ctrl+Q` | Save the session and quit |
 
 A complete composer-and-completion key reference is in
-[Chat Composer](../composer/).
+[Chat Composer](../composer/). If Shift chords fail inside tmux, see
+[Terminal & tmux shortcuts](../../troubleshooting/terminal-tmux/).

{kolega_code-0.14.0 → kolega_code-0.16.0}/docs/src/content/docs/tui/slash-commands.md

@@ -34,6 +34,8 @@ These control the app and your session.
 | --- | --- |
 | `/skills` | List available Agent Skills |
 | `/init` | Create or update `AGENTS.md` for this repository |
+| `/attach` | Attach an image: clipboard if no path, or `/attach <path>` for a file |
+| `/detach` | Remove pending image attachments |
 | `/plan` | Switch to [Plan mode](../modes/) |
 | `/build` | Switch to [Build mode](../modes/) |
 | `/sidebar` | Show or hide the side panel |

{kolega_code-0.14.0 → kolega_code-0.16.0}/kolega_code/__init__.py

@@ -151,4 +151,4 @@ __all__ = [
     "parse_git_status_output",
 ]
 
-__version__ = "0.14.0"
+__version__ = "0.16.0"

{kolega_code-0.14.0 → kolega_code-0.16.0}/kolega_code/agent/__init__.py

@@ -44,4 +44,4 @@ __all__ = [
     "ToolExtension",
 ]
 
-__version__ = "0.14.0"
+__version__ = "0.16.0"

{kolega_code-0.14.0 → kolega_code-0.16.0}/kolega_code/agent/orchestration/guide.py

@@ -88,6 +88,23 @@ without running the script. Required keys: `name`, `description`. Optional `phas
 
 ### Quality patterns to compose
 
+- Surface map: before a broad run, map the real work surface — files, modules,
+  services, risks, unknowns, owner boundaries, or test targets — then fan out over
+  that map instead of guessing stages from the user's wording.
+- Shard-and-sweep: split a large surface by package/module/service/concern, send
+  focused agents across the shards, then deduplicate and synthesize.
+- Cross-cut matrix: review the same target set across dimensions such as correctness,
+  security, performance, UX, compatibility, test coverage, and migration risk.
+- Research funnel: map unknowns → research in parallel → compare options → produce
+  a decision-ready plan. In plan mode this is the main workflow shape.
+- Hypothesis tournament: generate several plausible explanations or approaches,
+  gather evidence for each, score them with parallel judges, then choose.
+- Implementation pipeline: map disjoint workstreams → implement with coder agents →
+  verify each stream → integrate and run final checks yourself.
+- Migration factory: inventory affected areas → classify risk → batch independent
+  edits → run targeted checks → compatibility review.
+- Failure triage loop: collect failures → cluster by likely root cause → investigate
+  clusters → fix isolated causes → rerun targeted tests.
 - Adversarial verify: for each finding, spawn N skeptics prompted to REFUTE it; keep it
   only if a majority fail to refute. Prevents plausible-but-wrong findings surviving.
 - Loop-until-dry: for unknown-size discovery, keep spawning finders until K consecutive
@@ -96,6 +113,8 @@ without running the script. Required keys: `name`, `description`. Optional `phas
   judges, synthesize from the winner.
 - Loop-until-budget: `while budget.total and budget.remaining() > 50_000: ...` to scale
   depth to the token ceiling.
+- Synthesis gate: after any broad fan-out, merge duplicates, rank confidence, identify
+  unresolved gaps, and decide whether another loop is worth the remaining budget.
 
 ### Implementing a plan (build mode)
 

{kolega_code-0.14.0 → kolega_code-0.16.0}/kolega_code/agent/tool_backend/edit_tool.py

@@ -1,10 +1,16 @@
+from __future__ import annotations
+
 import re
 from dataclasses import dataclass
-from typing import Callable
+from typing import TYPE_CHECKING, Callable, Optional
 
 from .base_tool import BaseTool
 from .edit_preview import build_diff_preview, build_head_preview
 
+if TYPE_CHECKING:
+    from kolega_code.services.lsp import LspManager
+    from kolega_code.services.snapshots import SnapshotService
+
 
 _BLOCK_PATTERN = re.compile(r"<<<<<<< SEARCH\r?\n(.*?)\r?\n=======\r?\n(.*?)\r?\n>>>>>>> REPLACE", re.DOTALL)
 _SMART_PUNCTUATION_TRANSLATION = str.maketrans(
@@ -38,6 +44,17 @@ class ResolvedReplacement:
 
 
 class EditTool(BaseTool):
+    def __init__(
+        self,
+        *args,
+        lsp_manager: Optional["LspManager"] = None,
+        snapshot_service: Optional["SnapshotService"] = None,
+        **kwargs,
+    ) -> None:
+        super().__init__(*args, **kwargs)
+        self._lsp_manager = lsp_manager
+        self._snapshot_service = snapshot_service
+
     async def edit(self, path: str, block: str) -> str:
         """
         Edit a file using a single search and replace block.
@@ -54,7 +71,11 @@ class EditTool(BaseTool):
             error_msg = "Multiple search and replace blocks provided. Use multi_edit for multiple blocks."
             await self.log_error(error_msg, sender=self.caller.agent_name)
             raise ValueError(error_msg)
-        return await self._edit_blocks(path, parsed_blocks, tool_name="edit")
+        result = await self._edit_blocks(path, parsed_blocks, tool_name="edit")
+        lsp_diags = await self._maybe_append_lsp_diagnostics(path)
+        if lsp_diags:
+            result += lsp_diags
+        return result
 
     async def multi_edit(self, path: str, blocks: str) -> str:
         """
@@ -71,7 +92,11 @@ class EditTool(BaseTool):
             A short summary of the update
         """
         parsed_blocks = self._parse_blocks(blocks)
-        return await self._edit_blocks(path, parsed_blocks, tool_name="multi_edit")
+        result = await self._edit_blocks(path, parsed_blocks, tool_name="multi_edit")
+        lsp_diags = await self._maybe_append_lsp_diagnostics(path)
+        if lsp_diags:
+            result += lsp_diags
+        return result
 
     async def write(self, path: str, content: str) -> str:
         """
@@ -98,8 +123,6 @@ class EditTool(BaseTool):
                     old_content = None
 
             parent_dir = self.filesystem.get_parent(path)
-            if parent_dir and parent_dir != "." and not self.filesystem.exists(parent_dir):
-                self.filesystem.create_directory(parent_dir)
 
             # Preserve the file's dominant line ending on overwrite; new files
             # keep the line endings the caller provided (typically LF).
@@ -107,7 +130,17 @@ class EditTool(BaseTool):
                 line_ending = self._detect_dominant_line_ending(path, old_content)
                 content = self._normalize_line_endings(content, line_ending)
 
-            self.filesystem.write_text(path, content)
+            def _write() -> None:
+                if parent_dir and parent_dir != "." and not self.filesystem.exists(parent_dir):
+                    self.filesystem.create_directory(parent_dir)
+                self.filesystem.write_text(path, content)
+
+            self._record_snapshot_mutation(
+                tool_name="write",
+                reason=f"write {path}",
+                paths=self._snapshot_paths_for_write(path),
+                mutate=_write,
+            )
 
             preview = (
                 build_diff_preview(old_content, content, path)
@@ -119,7 +152,11 @@ class EditTool(BaseTool):
                 tool_call_id=getattr(self.caller, "current_tool_execution_id", None),
                 tool_name="write",
             )
-            return f"Wrote {path}"
+            result = f"Wrote {path}"
+            lsp_diags = await self._maybe_append_lsp_diagnostics(path)
+            if lsp_diags:
+                result += lsp_diags
+            return result
         except PermissionError:
             error_msg = f"Permission denied when writing to file: {path}"
             await self.log_error(error_msg, sender=self.caller.agent_name)
@@ -159,7 +196,12 @@ class EditTool(BaseTool):
             blocked_msg = self._enforce_vibe_edit_policy(path)
             if blocked_msg:
                 return blocked_msg
-            self.filesystem.write_text(path, normalized_updated)
+            self._record_snapshot_mutation(
+                tool_name=tool_name,
+                reason=f"{tool_name} {path}",
+                paths=[path],
+                mutate=lambda: self.filesystem.write_text(path, normalized_updated),
+            )
 
             await self.send_edit_preview(
                 build_diff_preview(original_content, normalized_updated, path),
@@ -202,6 +244,33 @@ class EditTool(BaseTool):
             )
         return parsed_blocks
 
+    def _record_snapshot_mutation(
+        self,
+        *,
+        tool_name: str,
+        reason: str,
+        paths: list[str],
+        mutate: Callable[[], None],
+    ) -> None:
+        if self._snapshot_service is None or not self._snapshot_service.can_snapshot_paths(paths):
+            mutate()
+            return
+        self._snapshot_service.record_mutation(
+            tool_name=tool_name,
+            tool_call_id=getattr(self.caller, "current_tool_execution_id", None),
+            reason=reason,
+            paths=paths,
+            mutate=mutate,
+        )
+
+    def _snapshot_paths_for_write(self, path: str) -> list[str]:
+        paths = [path]
+        parent = self.filesystem.get_parent(path)
+        while parent and parent != "." and not self.filesystem.exists(parent):
+            paths.append(parent)
+            parent = self.filesystem.get_parent(parent)
+        return paths
+
     def _resolve_replacement(self, content: str, block: SearchReplaceBlock) -> ResolvedReplacement:
         passes: list[tuple[str, Callable[[str, str], list[tuple[int, int]]]]] = [
             ("exact", self._exact_spans),
@@ -379,3 +448,35 @@ class EditTool(BaseTool):
         if target == "\n":
             return text
         return text.replace("\n", target)
+
+    async def _maybe_append_lsp_diagnostics(self, path: str) -> str:
+        """Query LSP diagnostics for *path* and format them for appending to the tool result.
+
+        Returns an empty string if LSP is unavailable, disabled, or produces no diagnostics.
+        """
+        if self._lsp_manager is None or not self._lsp_manager.enabled:
+            return ""
+
+        if not self._lsp_manager._config.auto_diagnostics_on_edit:
+            return ""
+
+        if not self._lsp_manager._initialized:
+            await self._lsp_manager.initialize()
+
+        server_name = self._lsp_manager.server_for_path(path)
+        if server_name is None:
+            return ""
+
+        try:
+            diagnostics = await self._lsp_manager.get_fresh_diagnostics(path)
+        except Exception:
+            return ""
+
+        if not diagnostics:
+            return ""
+
+        from kolega_code.services.lsp import format_diagnostics
+
+        # Blank-line separator so the diagnostics block reads as a distinct section
+        # after the result line (e.g. "Edited foo.py\n\nLSP diagnostics (...)").
+        return "\n\n" + format_diagnostics(diagnostics, path, source=server_name)