agent-brain-cli 9.2.0__tar.gz → 9.4.0__tar.gz

{agent_brain_cli-9.2.0 → agent_brain_cli-9.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agent-brain-cli
-Version: 9.2.0
+Version: 9.4.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.2.0,<10.0.0)
+Requires-Dist: agent-brain-rag (>=9.4.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.2.0 → agent_brain_cli-9.4.0}/agent_brain_cli/__init__.py

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

{agent_brain_cli-9.2.0 → agent_brain_cli-9.4.0}/agent_brain_cli/commands/config.py

@@ -2,11 +2,13 @@
 
 import json
 import os
+import socket
 import sys
 from pathlib import Path
 from typing import Any
 
 import click
+import yaml
 from rich.console import Console
 from rich.table import Table
 
@@ -15,6 +17,33 @@ from agent_brain_cli.xdg_paths import get_xdg_config_dir
 console = Console()
 
 
+EMBEDDING_DEFAULT_MODELS = {
+    "openai": "text-embedding-3-large",
+    "ollama": "nomic-embed-text",
+    "cohere": "embed-english-v3.0",
+}
+
+SUMMARIZATION_DEFAULT_MODELS = {
+    "anthropic": "claude-haiku-4-5-20251001",
+    "openai": "gpt-5-mini",
+    "ollama": "llama3.2:latest",
+    "gemini": "gemini-3-flash",
+}
+
+
+def _find_available_api_port(start: int = 8000, end: int = 8300) -> int:
+    """Find an available TCP port in an inclusive range."""
+    for port in range(start, end + 1):
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
+            sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+            try:
+                sock.bind(("127.0.0.1", port))
+            except OSError:
+                continue
+            return port
+    raise click.ClickException(f"No available ports found in range {start}-{end}")
+
+
 def _find_config_file() -> Path | None:
     """Find the configuration file in standard locations.
 
@@ -91,12 +120,32 @@ def _find_config_file() -> Path | None:
 
 def _load_yaml(path: Path) -> dict[str, Any]:
     """Load YAML configuration file."""
-    import yaml
-
     with open(path) as f:
         return yaml.safe_load(f) or {}
 
 
+def _resolve_wizard_config_path() -> Path:
+    """Resolve output path for config wizard.
+
+    Returns:
+        Path to .agent-brain/config.yaml in nearest existing state dir,
+        or creates .agent-brain/config.yaml in current working directory.
+    """
+    current = Path.cwd()
+    root = Path(current.anchor)
+
+    while True:
+        state_dir = current / ".agent-brain"
+        if state_dir.is_dir():
+            return state_dir / "config.yaml"
+
+        if current == root:
+            break
+        current = current.parent
+
+    return Path.cwd() / ".agent-brain" / "config.yaml"
+
+
 @click.group("config")
 def config_group() -> None:
     """View and manage Agent Brain configuration.
@@ -105,10 +154,141 @@ def config_group() -> None:
     Commands:
       show   - Display active configuration
       path   - Show config file location
+      wizard - Create/update config interactively
     """
     pass
 
 
+@config_group.command("wizard")
+def wizard() -> None:
+    """Interactive configuration wizard for Agent Brain providers."""
+    embed_provider: str = click.prompt(
+        "Embedding provider",
+        type=click.Choice(["openai", "ollama", "cohere"]),
+        default="openai",
+    )
+    embed_model: str = click.prompt(
+        "Embedding model",
+        default=EMBEDDING_DEFAULT_MODELS[embed_provider],
+    )
+
+    batch_size: str | None = None
+    request_delay_ms: int | None = None
+    if embed_provider == "ollama":
+        batch_size = click.prompt(
+            "Batch size",
+            type=click.Choice(["1", "5", "10", "20", "50", "100"]),
+            default="10",
+        )
+        request_delay_ms = click.prompt(
+            "Request delay between batches (ms, 0=none)",
+            type=click.IntRange(min=0),
+            default=0,
+        )
+
+    summ_provider: str = click.prompt(
+        "Summarization provider",
+        type=click.Choice(["anthropic", "openai", "ollama", "gemini"]),
+        default="anthropic",
+    )
+    summ_model: str = click.prompt(
+        "Summarization model",
+        default=SUMMARIZATION_DEFAULT_MODELS[summ_provider],
+    )
+
+    graphrag_mode: str = click.prompt(
+        "GraphRAG mode\n"
+        "1) Disabled\n"
+        "2) AST for code + LangExtract for docs (recommended for mixed repos)\n"
+        "3) Kuzu + AST for code + LangExtract for docs\n"
+        "4) AST for code only",
+        type=click.Choice(["1", "2", "3", "4"]),
+        default="1",
+    )
+
+    suggested_port = _find_available_api_port(8000, 8300)
+    click.echo(f"Discovered available API port in 8000-8300 range: {suggested_port}")
+
+    deployment_mode: str = click.prompt(
+        "Deployment mode\n"
+        "1) Localhost only (127.0.0.1)\n"
+        "2) Network accessible (0.0.0.0 or custom host)\n"
+        "3) Custom port on localhost",
+        type=click.Choice(["1", "2", "3"]),
+        default="1",
+    )
+
+    api_host = "127.0.0.1"
+    if deployment_mode == "2":
+        host_mode: str = click.prompt(
+            "Host selection\n" "1) 0.0.0.0\n" "2) Custom host/IP",
+            type=click.Choice(["1", "2"]),
+            default="1",
+        )
+        if host_mode == "1":
+            api_host = "0.0.0.0"
+        else:
+            api_host = click.prompt("Custom host", default="0.0.0.0")
+
+    api_port: int = click.prompt(
+        "API port",
+        type=click.IntRange(min=1, max=65535),
+        default=suggested_port,
+    )
+
+    config: dict[str, Any] = {
+        "embedding": {
+            "provider": embed_provider,
+            "model": embed_model,
+        },
+        "summarization": {
+            "provider": summ_provider,
+            "model": summ_model,
+        },
+        "graphrag": {
+            "enabled": False,
+        },
+        "api": {
+            "host": api_host,
+            "port": api_port,
+        },
+    }
+
+    if embed_provider == "ollama":
+        config["embedding"]["params"] = {
+            "batch_size": int(batch_size or "10"),
+            "request_delay_ms": int(request_delay_ms or 0),
+        }
+
+    if graphrag_mode == "2":
+        config["graphrag"] = {
+            "enabled": True,
+            "store_type": "simple",
+            "use_code_metadata": True,
+            "doc_extractor": "langextract",
+        }
+    elif graphrag_mode == "3":
+        config["graphrag"] = {
+            "enabled": True,
+            "store_type": "kuzu",
+            "use_code_metadata": True,
+            "doc_extractor": "langextract",
+        }
+    elif graphrag_mode == "4":
+        config["graphrag"] = {
+            "enabled": True,
+            "store_type": "simple",
+            "use_code_metadata": True,
+        }
+
+    config_path = _resolve_wizard_config_path()
+    config_path.parent.mkdir(parents=True, exist_ok=True)
+    with open(config_path, "w", encoding="utf-8") as f:
+        yaml.safe_dump(config, f, default_flow_style=False, sort_keys=False)
+
+    console.print(f"[green]Config written to {config_path}[/]")
+
+
 @config_group.command("show")
 @click.option("--json", "json_output", is_flag=True, help="Output as JSON")
 def show_config(json_output: bool) -> None:

{agent_brain_cli-9.2.0 → agent_brain_cli-9.4.0}/agent_brain_cli/commands/start.py

@@ -196,8 +196,8 @@ def update_registry(project_root: Path, state_dir: Path) -> None:
 @click.option(
     "--timeout",
     type=int,
-    default=30,
-    help="Startup timeout in seconds (default: 30)",
+    default=120,
+    help="Startup timeout in seconds (default: 120)",
 )
 @click.option("--json", "json_output", is_flag=True, help="Output as JSON")
 @click.option(

{agent_brain_cli-9.2.0 → agent_brain_cli-9.4.0}/agent_brain_cli/commands/types.py

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

{agent_brain_cli-9.2.0 → agent_brain_cli-9.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agent-brain-cli"
-version = "9.2.0"
+version = "9.4.0"
 description = "Agent Brain CLI - Command-line interface for managing AI agent memory and knowledge retrieval"
 authors = ["Spillwave Solutions"]
 readme = "README.md"
@@ -27,7 +27,7 @@ httpx = "^0.28.0"
 rich = "^13.9.0"
 pyyaml = "^6.0.0"
 pydantic = "^2.10.0"
-agent-brain-rag = "^9.2.0"
+agent-brain-rag = "^9.4.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.0"