agent-brain-cli 9.4.1__tar.gz → 9.6.0__tar.gz

{agent_brain_cli-9.4.1 → agent_brain_cli-9.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agent-brain-cli
-Version: 9.4.1
+Version: 9.6.0
 Summary: Agent Brain CLI - Command-line interface for managing AI agent memory and knowledge retrieval
 Home-page: https://github.com/SpillwaveSolutions/agent-brain
 License: MIT
@@ -15,7 +15,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: agent-brain-rag (>=9.4.1,<10.0.0)
+Requires-Dist: agent-brain-rag (>=9.6.0,<10.0.0)
 Requires-Dist: click (>=8.1.0,<9.0.0)
 Requires-Dist: httpx (>=0.28.0,<0.29.0)
 Requires-Dist: pydantic (>=2.10.0,<3.0.0)

{agent_brain_cli-9.4.1 → agent_brain_cli-9.6.0}/agent_brain_cli/__init__.py

@@ -1,3 +1,3 @@
 """Doc-Serve CLI - Command-line interface for managing Doc-Serve server."""
 
-__version__ = "9.4.1"
+__version__ = "9.6.0"

{agent_brain_cli-9.4.1 → agent_brain_cli-9.6.0}/agent_brain_cli/commands/config.py

@@ -12,6 +12,15 @@ import yaml
 from rich.console import Console
 from rich.table import Table
 
+from agent_brain_cli.config_migrate import (
+    MigrationResult,
+    diff_config_file,
+    migrate_config_file,
+)
+from agent_brain_cli.config_schema import (
+    format_validation_errors,
+    validate_config_file,
+)
 from agent_brain_cli.xdg_paths import get_xdg_config_dir
 
 console = Console()
@@ -152,9 +161,12 @@ def config_group() -> None:
 
     \b
     Commands:
-      show   - Display active configuration
-      path   - Show config file location
-      wizard - Create/update config interactively
+      show     - Display active configuration
+      path     - Show config file location
+      wizard   - Create/update config interactively
+      validate - Validate config against schema
+      migrate  - Upgrade config to current schema
+      diff     - Preview what migrate would change
     """
     pass
 
@@ -162,6 +174,17 @@ def config_group() -> None:
 @config_group.command("wizard")
 def wizard() -> None:
     """Interactive configuration wizard for Agent Brain providers."""
+    # Check existing config for validation issues before prompting
+    existing_config = _find_config_file()
+    if existing_config:
+        existing_errors = validate_config_file(existing_config)
+        if existing_errors:
+            console.print(
+                "\n[bold yellow]Warning:[/] Existing config has validation issues:"
+            )
+            console.print(format_validation_errors(existing_errors))
+            console.print()
+
     embed_provider: str = click.prompt(
         "Embedding provider",
         type=click.Choice(["openai", "ollama", "cohere"]),
@@ -288,6 +311,19 @@ def wizard() -> None:
 
     console.print(f"[green]Config written to {config_path}[/]")
 
+    # Validate the written config and warn if issues found
+    post_write_errors = validate_config_file(config_path)
+    if post_write_errors:
+        console.print(
+            "\n[bold yellow]Warning:[/] The generated config has validation issues:\n"
+        )
+        console.print(format_validation_errors(post_write_errors))
+        if not click.confirm("Continue with this config anyway?", default=False):
+            console.print(
+                "[red]Config wizard aborted." " Please fix the issues and try again.[/]"
+            )
+            sys.exit(1)
+
 
 @config_group.command("show")
 @click.option("--json", "json_output", is_flag=True, help="Output as JSON")
@@ -408,3 +444,169 @@ def config_path(json_output: bool) -> None:
         console.print(f"[green]{config_path}[/]")
     else:
         console.print("[yellow]No config file found[/]")
+
+
+@config_group.command("validate")
+@click.option(
+    "--file",
+    "config_file",
+    type=click.Path(exists=True),
+    default=None,
+    help="Path to config file (default: auto-detect)",
+)
+@click.option("--json", "json_output", is_flag=True, help="Output as JSON")
+def validate_config(config_file: str | None, json_output: bool) -> None:
+    """Validate config.yaml against the Agent Brain schema.
+
+    Checks for unknown keys, invalid provider values, deprecated fields,
+    and type errors. Reports line numbers and fix suggestions.
+
+    \b
+    Examples:
+      agent-brain config validate
+      agent-brain config validate --file ./my-config.yaml
+      agent-brain config validate --json
+    """
+    if config_file is not None:
+        path: Path | None = Path(config_file)
+    else:
+        path = _find_config_file()
+
+    if path is None:
+        if json_output:
+            click.echo(
+                json.dumps(
+                    {
+                        "valid": None,
+                        "config_file": None,
+                        "errors": [],
+                        "message": "No config file found",
+                    }
+                )
+            )
+        else:
+            console.print("[yellow]No config file found. Nothing to validate.[/]")
+        sys.exit(0)
+
+    errors = validate_config_file(path)
+
+    if json_output:
+        output: dict[str, Any] = {
+            "valid": len(errors) == 0,
+            "config_file": str(path),
+            "errors": [
+                {
+                    "field": e.field,
+                    "message": e.message,
+                    "line_number": e.line_number,
+                    "suggestion": e.suggestion,
+                }
+                for e in errors
+            ],
+        }
+        click.echo(json.dumps(output, indent=2))
+        if errors:
+            sys.exit(1)
+        return
+
+    if not errors:
+        console.print(f"[green]Config is valid[/] ({path})")
+        sys.exit(0)
+
+    console.print(format_validation_errors(errors))
+    sys.exit(1)
+
+
+@config_group.command("migrate")
+@click.option(
+    "--file",
+    "config_file",
+    type=click.Path(exists=True),
+    default=None,
+    help="Path to config file (default: auto-detect)",
+)
+@click.option(
+    "--dry-run", is_flag=True, help="Show what would change without modifying"
+)
+def migrate_config_cmd(config_file: str | None, dry_run: bool) -> None:
+    """Migrate config.yaml to the current schema version.
+
+    Upgrades deprecated keys and restructures config sections.
+    Use --dry-run to preview changes without modifying the file.
+
+    \b
+    Examples:
+      agent-brain config migrate
+      agent-brain config migrate --dry-run
+      agent-brain config migrate --file ./old-config.yaml
+    """
+    if config_file is not None:
+        path: Path | None = Path(config_file)
+    else:
+        path = _find_config_file()
+
+    if path is None:
+        console.print("[yellow]No config file found.[/]")
+        sys.exit(0)
+
+    if dry_run:
+        diff = diff_config_file(path)
+        if diff:
+            console.print(diff)
+        else:
+            console.print("[green]Config is already up to date. No changes needed.[/]")
+        sys.exit(0)
+
+    result: MigrationResult = migrate_config_file(path)
+    if result.already_current:
+        console.print("[green]Config is already up to date[/]")
+        sys.exit(0)
+
+    for change in result.changes:
+        console.print(f"  [cyan]->[/] {change}")
+    console.print(f"\n[green]Config migrated successfully[/] ({path})")
+
+
+@config_group.command("diff")
+@click.option(
+    "--file",
+    "config_file",
+    type=click.Path(exists=True),
+    default=None,
+    help="Path to config file (default: auto-detect)",
+)
+def diff_config_cmd(config_file: str | None) -> None:
+    """Show what 'config migrate' would change.
+
+    Displays a unified diff of the current config vs the migrated version.
+
+    \b
+    Examples:
+      agent-brain config diff
+      agent-brain config diff --file ./config.yaml
+    """
+    if config_file is not None:
+        path: Path | None = Path(config_file)
+    else:
+        path = _find_config_file()
+
+    if path is None:
+        console.print("[yellow]No config file found.[/]")
+        sys.exit(0)
+
+    diff = diff_config_file(path)
+    if not diff:
+        console.print("[green]Config is already up to date. No changes needed.[/]")
+        sys.exit(0)
+
+    for line in diff.splitlines():
+        if line.startswith("---") or line.startswith("+++"):
+            console.print(f"[bold]{line}[/]")
+        elif line.startswith("-"):
+            console.print(f"[red]{line}[/]")
+        elif line.startswith("+"):
+            console.print(f"[green]{line}[/]")
+        elif line.startswith("@@"):
+            console.print(f"[cyan]{line}[/]")
+        else:
+            console.print(line)

{agent_brain_cli-9.4.1 → agent_brain_cli-9.6.0}/agent_brain_cli/commands/types.py

@@ -24,7 +24,8 @@ FILE_TYPE_PRESETS: dict[str, list[str]] = {
     "rust": ["*.rs"],
     "java": ["*.java"],
     "csharp": ["*.cs"],
-    "pascal": ["*.pas", "*.pp", "*.lpr", "*.dpr"],
+    "pascal": ["*.pas", "*.pp", "*.lpr", "*.dpr", "*.dpk"],
+    "object-pascal": ["*.pas", "*.pp", "*.lpr", "*.dpr", "*.dpk"],
     "c": ["*.c", "*.h"],
     "cpp": ["*.cpp", "*.hpp", "*.cc", "*.hh"],
     "web": ["*.html", "*.css", "*.scss", "*.jsx", "*.tsx"],
@@ -49,6 +50,7 @@ FILE_TYPE_PRESETS: dict[str, list[str]] = {
         "*.pp",
         "*.lpr",
         "*.dpr",
+        "*.dpk",
         "*.c",
         "*.h",
         "*.cpp",

agent_brain_cli-9.6.0/agent_brain_cli/config_migrate.py

@@ -0,0 +1,198 @@
+"""Config migration engine for Agent Brain YAML config files.
+
+Provides versioned migration steps that upgrade deprecated config keys to the
+current schema, and a diff utility to preview changes before applying them.
+"""
+
+from __future__ import annotations
+
+import copy
+import difflib
+from collections.abc import Callable
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any
+
+import yaml
+
+# ---------------------------------------------------------------------------
+# MigrationResult dataclass
+# ---------------------------------------------------------------------------
+
+
+@dataclass
+class MigrationResult:
+    """Result of running migration steps against a config dict.
+
+    Attributes:
+        original: Deep copy of the original config dict before migration.
+        migrated: Config dict after all migration steps have been applied.
+        changes: Human-readable descriptions of each change applied.
+        already_current: True when no migration steps produced any changes.
+    """
+
+    original: dict[str, Any]
+    migrated: dict[str, Any]
+    changes: list[str]
+    already_current: bool
+
+
+# ---------------------------------------------------------------------------
+# Migration functions
+# ---------------------------------------------------------------------------
+
+
+def _migrate_use_llm_extraction(
+    config: dict[str, Any],
+) -> tuple[dict[str, Any], list[str]]:
+    """Migrate graphrag.use_llm_extraction -> graphrag.doc_extractor.
+
+    Phase 34 decision: `use_llm_extraction` (bool) was renamed to
+    `doc_extractor` (str: "langextract" | "none").
+
+    Args:
+        config: Config dict (will NOT be mutated; a deep copy is made).
+
+    Returns:
+        Tuple of (updated config dict, list of change descriptions).
+    """
+    changes: list[str] = []
+    config = copy.deepcopy(config)
+    graphrag = config.get("graphrag", {})
+
+    if "use_llm_extraction" in graphrag:
+        old_val = graphrag.pop("use_llm_extraction")
+        if old_val:
+            graphrag["doc_extractor"] = "langextract"
+            changes.append(
+                "graphrag.use_llm_extraction=True -> graphrag.doc_extractor=langextract"
+            )
+        else:
+            changes.append(
+                "graphrag.use_llm_extraction=False -> removed"
+                " (default: no doc extraction)"
+            )
+        config["graphrag"] = graphrag
+
+    return config, changes
+
+
+# ---------------------------------------------------------------------------
+# MIGRATIONS list — applied in order by migrate_config
+# ---------------------------------------------------------------------------
+
+MigrationFn = Callable[[dict[str, Any]], tuple[dict[str, Any], list[str]]]
+
+MIGRATIONS: list[MigrationFn] = [
+    _migrate_use_llm_extraction,
+]
+
+
+# ---------------------------------------------------------------------------
+# Public API
+# ---------------------------------------------------------------------------
+
+
+def migrate_config(config: dict[str, Any]) -> MigrationResult:
+    """Apply all migration steps to a config dict.
+
+    Does NOT read or write any files. Mutations are applied to a deep copy,
+    so the original dict is never modified.
+
+    Args:
+        config: Parsed YAML config dictionary.
+
+    Returns:
+        :class:`MigrationResult` describing the changes (or lack thereof).
+    """
+    original = copy.deepcopy(config)
+    current = copy.deepcopy(config)
+    all_changes: list[str] = []
+
+    for migration_fn in MIGRATIONS:
+        current, changes = migration_fn(current)
+        all_changes.extend(changes)
+
+    return MigrationResult(
+        original=original,
+        migrated=current,
+        changes=all_changes,
+        already_current=len(all_changes) == 0,
+    )
+
+
+def migrate_config_file(path: Path) -> MigrationResult:
+    """Apply all migrations to a YAML config file and write results to disk.
+
+    Reads the YAML from *path*, calls :func:`migrate_config`, and if any
+    changes were made, writes the migrated dict back to *path* using
+    ``yaml.safe_dump`` with ``default_flow_style=False, sort_keys=False``.
+
+    Args:
+        path: Path to the YAML config file.
+
+    Returns:
+        :class:`MigrationResult` describing the changes (or lack thereof).
+
+    Raises:
+        OSError: If the file cannot be read or written.
+        yaml.YAMLError: If the YAML cannot be parsed.
+    """
+    yaml_text = path.read_text(encoding="utf-8")
+    config: dict[str, Any] = yaml.safe_load(yaml_text) or {}
+
+    result = migrate_config(config)
+
+    if not result.already_current:
+        with open(path, "w", encoding="utf-8") as fh:
+            yaml.safe_dump(
+                result.migrated, fh, default_flow_style=False, sort_keys=False
+            )
+
+    return result
+
+
+def diff_config(config: dict[str, Any]) -> str:
+    """Return a unified diff of what :func:`migrate_config` would change.
+
+    Args:
+        config: Parsed YAML config dictionary.
+
+    Returns:
+        Unified diff string (empty if no changes needed).
+    """
+    result = migrate_config(config)
+    if result.already_current:
+        return ""
+
+    original_yaml = yaml.safe_dump(result.original, default_flow_style=False)
+    migrated_yaml = yaml.safe_dump(result.migrated, default_flow_style=False)
+
+    diff_lines = list(
+        difflib.unified_diff(
+            original_yaml.splitlines(keepends=True),
+            migrated_yaml.splitlines(keepends=True),
+            fromfile="config.yaml (current)",
+            tofile="config.yaml (migrated)",
+        )
+    )
+
+    return "".join(diff_lines)
+
+
+def diff_config_file(path: Path) -> str:
+    """Return a unified diff of what migrating *path* would change.
+
+    Args:
+        path: Path to the YAML config file.
+
+    Returns:
+        Unified diff string (empty if no changes needed).
+
+    Raises:
+        OSError: If the file cannot be read.
+        yaml.YAMLError: If the YAML cannot be parsed.
+    """
+    yaml_text = path.read_text(encoding="utf-8")
+    config: dict[str, Any] = yaml.safe_load(yaml_text) or {}
+    return diff_config(config)