bog-agents-cli 0.5.2__py3-none-any.whl

bog_agents_cli/__init__.py

@@ -0,0 +1,9 @@
+"""Bog Agents CLI - Interactive AI coding assistant."""
+
+from bog_agents_cli._version import __version__
+from bog_agents_cli.main import cli_main
+
+__all__ = [
+    "__version__",
+    "cli_main",
+]

bog_agents_cli/__main__.py

@@ -0,0 +1,6 @@
+"""Allow running the CLI as: python -m bog_agents.cli."""
+
+from bog_agents_cli.main import cli_main
+
+if __name__ == "__main__":
+    cli_main()

bog_agents_cli/_debug.py

@@ -0,0 +1,50 @@
+"""Shared debug-logging configuration for verbose file-based tracing.
+
+When the `BOG_AGENTS_DEBUG` environment variable is set, modules that handle
+streaming or remote communication can enable detailed file-based logging. This
+helper centralizes the setup so the env-var name, file path, and format are
+defined in one place.
+"""
+
+from __future__ import annotations
+
+import logging
+import os
+from pathlib import Path
+
+
+def configure_debug_logging(target: logging.Logger) -> None:
+    """Attach a file handler to *target* when `BOG_AGENTS_DEBUG` is set.
+
+    The log file defaults to `'/tmp/bog_agents_debug.log'` but can be overridden
+    with `BOG_AGENTS_DEBUG_FILE`. The handler appends so that multiple modules
+    share the same log file across a session.
+
+    Does nothing when `BOG_AGENTS_DEBUG` is not set.
+
+    Args:
+        target: Logger to configure.
+    """
+    if not os.environ.get("BOG_AGENTS_DEBUG"):
+        return
+
+    debug_path = Path(
+        os.environ.get(
+            "BOG_AGENTS_DEBUG_FILE",
+            "/tmp/bog_agents_debug.log",  # noqa: S108
+        )
+    )
+    try:
+        handler = logging.FileHandler(str(debug_path), mode="a")
+    except OSError as exc:
+        import sys
+
+        print(  # noqa: T201
+            f"Warning: could not open debug log file {debug_path}: {exc}",
+            file=sys.stderr,
+        )
+        return
+    handler.setLevel(logging.DEBUG)
+    handler.setFormatter(logging.Formatter("%(asctime)s %(name)s %(message)s"))
+    target.addHandler(handler)
+    target.setLevel(logging.DEBUG)

bog_agents_cli/_server_config.py

@@ -0,0 +1,320 @@
+"""Typed configuration for the CLI-to-server subprocess communication channel.
+
+The CLI spawns a `langgraph dev` subprocess and passes configuration via
+environment variables prefixed with `DA_SERVER_`. This module provides a single
+`ServerConfig` dataclass that both sides share so that the set of variables,
+their serialization format, and their default values are defined in one place.
+The CLI writes config with `to_env()` and the server graph reads it back
+with `from_env()`.
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+import os
+from dataclasses import dataclass
+from pathlib import Path
+from typing import TYPE_CHECKING, Any
+
+from bog_agents_cli._server_constants import ENV_PREFIX as _ENV_PREFIX
+
+if TYPE_CHECKING:
+    from bog_agents_cli.project_utils import ProjectContext
+
+logger = logging.getLogger(__name__)
+
+_DEFAULT_ASSISTANT_ID = "agent"
+
+
+def _read_env_bool(suffix: str, *, default: bool = False) -> bool:
+    """Read a `DA_SERVER_*` boolean from the environment.
+
+    Boolean env vars use the `'true'` / `'false'` convention (case insensitive).
+    Missing variables fall back to *default*.
+
+    Args:
+        suffix: Variable name suffix after the `DA_SERVER_` prefix.
+        default: Value when the variable is absent.
+
+    Returns:
+        Parsed boolean.
+    """
+    raw = os.environ.get(f"{_ENV_PREFIX}{suffix}")
+    if raw is None:
+        return default
+    return raw.lower() == "true"
+
+
+def _read_env_json(suffix: str) -> Any:  # noqa: ANN401
+    """Read a JSON-encoded `DA_SERVER_*` variable.
+
+    Args:
+        suffix: Variable name suffix after the `DA_SERVER_` prefix.
+
+    Returns:
+        Parsed JSON value, or `None` if the variable is absent.
+
+    Raises:
+        ValueError: If the variable is present but not valid JSON.
+    """
+    raw = os.environ.get(f"{_ENV_PREFIX}{suffix}")
+    if raw is None:
+        return None
+    try:
+        return json.loads(raw)
+    except json.JSONDecodeError as exc:
+        msg = (
+            f"Failed to parse {_ENV_PREFIX}{suffix} as JSON: {exc}. "
+            f"Value was: {raw[:200]!r}"
+        )
+        raise ValueError(msg) from exc
+
+
+def _read_env_str(suffix: str) -> str | None:
+    """Read an optional `DA_SERVER_*` string variable.
+
+    Args:
+        suffix: Variable name suffix after the `DA_SERVER_` prefix.
+
+    Returns:
+        The string value, or `None` if absent.
+    """
+    return os.environ.get(f"{_ENV_PREFIX}{suffix}")
+
+
+def _read_env_optional_bool(suffix: str) -> bool | None:
+    """Read a tri-state `DA_SERVER_*` boolean (`True` / `False` / `None`).
+
+    Used for settings where `None` carries a distinct meaning (e.g. "not
+    specified, use default logic").
+
+    Args:
+        suffix: Variable name suffix after the `DA_SERVER_` prefix.
+
+    Returns:
+        `True`, `False`, or `None` when the variable is absent.
+    """
+    raw = os.environ.get(f"{_ENV_PREFIX}{suffix}")
+    if raw is None:
+        return None
+    return raw.lower() == "true"
+
+
+@dataclass(frozen=True)
+class ServerConfig:
+    """Full configuration payload passed from the CLI to the server subprocess.
+
+    Serialized to/from `DA_SERVER_*` environment variables so that the server
+    graph (which runs in a separate Python interpreter) can reconstruct the
+    CLI's intent without sharing memory.
+    """
+
+    model: str | None = None
+    model_params: dict[str, Any] | None = None
+    assistant_id: str = _DEFAULT_ASSISTANT_ID
+    system_prompt: str | None = None
+    auto_approve: bool = False
+    interactive: bool = True
+    enable_shell: bool = True
+    enable_ask_user: bool = False
+    enable_memory: bool = True
+    enable_skills: bool = True
+    sandbox_type: str | None = None
+    sandbox_id: str | None = None
+    sandbox_setup: str | None = None
+    cwd: str | None = None
+    project_root: str | None = None
+    mcp_config_path: str | None = None
+    no_mcp: bool = False
+    trust_project_mcp: bool | None = None
+
+    def __post_init__(self) -> None:
+        """Normalize fields that have canonical representations."""
+        if self.sandbox_type == "none":
+            object.__setattr__(self, "sandbox_type", None)
+
+    # ------------------------------------------------------------------
+    # Serialization
+    # ------------------------------------------------------------------
+
+    def to_env(self) -> dict[str, str | None]:
+        """Serialize this config to a `DA_SERVER_*` env-var mapping.
+
+        `None` values signal that the variable should be *cleared* from the
+        environment (rather than set to an empty string), so callers can
+        iterate and set or clear each variable in `os.environ`.
+
+        Returns:
+            Dict mapping env-var suffixes (without the prefix) to their
+                string values or `None`.
+        """
+        return {
+            "MODEL": self.model,
+            "MODEL_PARAMS": (
+                json.dumps(self.model_params) if self.model_params is not None else None
+            ),
+            "ASSISTANT_ID": self.assistant_id,
+            "SYSTEM_PROMPT": self.system_prompt,
+            "AUTO_APPROVE": str(self.auto_approve).lower(),
+            "INTERACTIVE": str(self.interactive).lower(),
+            "ENABLE_SHELL": str(self.enable_shell).lower(),
+            "ENABLE_ASK_USER": str(self.enable_ask_user).lower(),
+            "ENABLE_MEMORY": str(self.enable_memory).lower(),
+            "ENABLE_SKILLS": str(self.enable_skills).lower(),
+            "SANDBOX_TYPE": self.sandbox_type,
+            "SANDBOX_ID": self.sandbox_id,
+            "SANDBOX_SETUP": self.sandbox_setup,
+            "CWD": self.cwd,
+            "PROJECT_ROOT": self.project_root,
+            "MCP_CONFIG_PATH": self.mcp_config_path,
+            "NO_MCP": str(self.no_mcp).lower(),
+            "TRUST_PROJECT_MCP": (
+                str(self.trust_project_mcp).lower()
+                if self.trust_project_mcp is not None
+                else None
+            ),
+        }
+
+    @classmethod
+    def from_env(cls) -> ServerConfig:
+        """Reconstruct a `ServerConfig` from the current `DA_SERVER_*` env vars.
+
+        This is the inverse of `to_env()` and is called inside the server
+        subprocess to recover the CLI's configuration.
+
+        Returns:
+            A `ServerConfig` populated from the environment.
+        """
+        return cls(
+            model=_read_env_str("MODEL"),
+            model_params=_read_env_json("MODEL_PARAMS"),
+            assistant_id=_read_env_str("ASSISTANT_ID") or _DEFAULT_ASSISTANT_ID,
+            system_prompt=_read_env_str("SYSTEM_PROMPT"),
+            auto_approve=_read_env_bool("AUTO_APPROVE"),
+            interactive=_read_env_bool("INTERACTIVE", default=True),
+            enable_shell=_read_env_bool("ENABLE_SHELL", default=True),
+            enable_ask_user=_read_env_bool("ENABLE_ASK_USER"),
+            enable_memory=_read_env_bool("ENABLE_MEMORY", default=True),
+            enable_skills=_read_env_bool("ENABLE_SKILLS", default=True),
+            sandbox_type=_read_env_str("SANDBOX_TYPE"),
+            sandbox_id=_read_env_str("SANDBOX_ID"),
+            sandbox_setup=_read_env_str("SANDBOX_SETUP"),
+            cwd=_read_env_str("CWD"),
+            project_root=_read_env_str("PROJECT_ROOT"),
+            mcp_config_path=_read_env_str("MCP_CONFIG_PATH"),
+            no_mcp=_read_env_bool("NO_MCP"),
+            trust_project_mcp=_read_env_optional_bool("TRUST_PROJECT_MCP"),
+        )
+
+    # ------------------------------------------------------------------
+    # Factory
+    # ------------------------------------------------------------------
+
+    @classmethod
+    def from_cli_args(
+        cls,
+        *,
+        project_context: ProjectContext | None,
+        model_name: str | None,
+        model_params: dict[str, Any] | None,
+        assistant_id: str,
+        auto_approve: bool,
+        sandbox_type: str,
+        sandbox_id: str | None,
+        sandbox_setup: str | None,
+        enable_shell: bool,
+        enable_ask_user: bool,
+        mcp_config_path: str | None,
+        no_mcp: bool,
+        trust_project_mcp: bool | None,
+        interactive: bool,
+    ) -> ServerConfig:
+        """Build a `ServerConfig` from parsed CLI arguments.
+
+        Handles path normalization (e.g. resolving relative MCP config paths
+        against the user's working directory) so that the raw serialized values
+        are always absolute and unambiguous.
+
+        Args:
+            project_context: Explicit user/project path context.
+            model_name: Model spec string.
+            model_params: Extra model kwargs.
+            assistant_id: Agent identifier.
+            auto_approve: Auto-approve all tools.
+            sandbox_type: Sandbox type.
+            sandbox_id: Existing sandbox ID to reuse.
+            sandbox_setup: Path to setup script for the sandbox.
+            enable_shell: Enable shell execution tools.
+            enable_ask_user: Enable ask_user tool.
+            mcp_config_path: Path to MCP config.
+            no_mcp: Disable MCP.
+            trust_project_mcp: Trust project MCP servers.
+            interactive: Whether the agent is interactive.
+
+        Returns:
+            A fully resolved `ServerConfig`.
+        """
+        normalized_mcp = _normalize_path(mcp_config_path, project_context, "MCP config")
+
+        return cls(
+            model=model_name,
+            model_params=model_params,
+            assistant_id=assistant_id,
+            auto_approve=auto_approve,
+            interactive=interactive,
+            enable_shell=enable_shell,
+            enable_ask_user=enable_ask_user,
+            sandbox_type=sandbox_type,
+            sandbox_id=sandbox_id,
+            sandbox_setup=_normalize_path(
+                sandbox_setup, project_context, "sandbox setup"
+            ),
+            cwd=(
+                str(project_context.user_cwd) if project_context is not None else None
+            ),
+            project_root=(
+                str(project_context.project_root)
+                if project_context is not None
+                and project_context.project_root is not None
+                else None
+            ),
+            mcp_config_path=normalized_mcp,
+            no_mcp=no_mcp,
+            trust_project_mcp=trust_project_mcp,
+        )
+
+
+def _normalize_path(
+    raw_path: str | None,
+    project_context: ProjectContext | None,
+    label: str,
+) -> str | None:
+    """Resolve a possibly-relative path to absolute.
+
+    The server subprocess runs in a different working directory, so relative
+    paths must be resolved against the user's original cwd before serialization.
+
+    Args:
+        raw_path: Path from CLI arguments (may be relative).
+        project_context: User/project context for path resolution.
+        label: Human-readable label for error messages (e.g. "MCP config").
+
+    Returns:
+        Absolute path string, or `None` when *raw_path* is `None` or empty.
+
+    Raises:
+        ValueError: If the path cannot be resolved.
+    """
+    if not raw_path:
+        return None
+    try:
+        if project_context is not None:
+            return str(project_context.resolve_user_path(raw_path))
+        return str(Path(raw_path).expanduser().resolve())
+    except OSError as exc:
+        msg = (
+            f"Could not resolve {label} path {raw_path!r}: {exc}. "
+            "Ensure the path exists and is accessible."
+        )
+        raise ValueError(msg) from exc

bog_agents_cli/_server_constants.py

@@ -0,0 +1,4 @@
+"""Shared constants for server communication between CLI and server graph."""
+
+ENV_PREFIX = "DA_SERVER_"
+"""Environment variable prefix used to pass CLI config to the server subprocess."""

bog_agents_cli/_testing_models.py

@@ -0,0 +1,144 @@
+"""Internal chat models used by local integration tests."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from langchain_core.language_models.fake_chat_models import GenericFakeChatModel
+from langchain_core.messages import AIMessage, BaseMessage
+from langchain_core.outputs import ChatGeneration, ChatResult
+from pydantic import Field
+
+if TYPE_CHECKING:
+    from collections.abc import Callable, Sequence
+
+    from langchain_core.callbacks import CallbackManagerForLLMRun
+    from langchain_core.language_models import LanguageModelInput
+    from langchain_core.runnables import Runnable
+    from langchain_core.tools import BaseTool
+
+
+class DeterministicIntegrationChatModel(GenericFakeChatModel):
+    """Deterministic chat model for CLI integration tests.
+
+    This subclasses LangChain's `GenericFakeChatModel` so the implementation
+    stays aligned with the core fake-chat-model test surface, while overriding
+    generation to remain prompt-driven and restart-safe for real CLI server
+    integration tests.
+
+    Why the existing `langchain_core` fakes cannot be reused here:
+
+    1. Every core fake (`GenericFakeChatModel`, `FakeListChatModel`,
+        `FakeMessagesListChatModel`) pops from an iterator or cycles an index —
+        the actual prompt is ignored. CLI integration tests start and stop the
+        server process, which resets in-memory state. An iterator-based model
+        either raises `StopIteration` or replays from the beginning after a
+        restart, producing wrong or missing responses. This model derives output
+        solely from the prompt text, so identical input always produces
+        identical output regardless of process lifecycle.
+
+    2. The agent runtime calls `model.bind_tools(schemas)` during
+        initialization. None of the core fakes implement `bind_tools`, so they
+        raise `AttributeError` in any agent-loop context. This model provides a
+        no-op passthrough.
+
+    3. The CLI server reads `model.profile` for capability negotiation (e.g.
+        `tool_calling`, `max_input_tokens`). Core fakes have no such attribute,
+        causing `AttributeError` or silent misconfiguration at runtime.
+
+    Additionally, the compact middleware issues summarization prompts mid-
+    conversation. A list-based model cannot distinguish these from normal user
+    turns without pre-knowledge of exact call ordering, whereas this model
+    detects summary requests by inspecting the prompt content.
+    """
+
+    model: str = "fake"
+    # Required by `GenericFakeChatModel`, but our override does not consume it.
+    messages: object = Field(default_factory=lambda: iter(()))
+    profile: dict[str, Any] | None = Field(
+        default_factory=lambda: {
+            "tool_calling": True,
+            "max_input_tokens": 8000,
+        }
+    )
+
+    def bind_tools(
+        self,
+        tools: Sequence[dict[str, Any] | type | Callable | BaseTool],  # noqa: ARG002
+        *,
+        tool_choice: str | None = None,  # noqa: ARG002
+        **kwargs: Any,  # noqa: ARG002
+    ) -> Runnable[LanguageModelInput, AIMessage]:
+        """Return self so the agent can bind tool schemas during tests."""
+        return self
+
+    def _generate(
+        self,
+        messages: list[BaseMessage],
+        stop: list[str] | None = None,  # noqa: ARG002
+        run_manager: CallbackManagerForLLMRun | None = None,  # noqa: ARG002
+        **kwargs: Any,  # noqa: ARG002
+    ) -> ChatResult:
+        """Produce a deterministic reply derived from the prompt text.
+
+        Returns:
+            A single-message `ChatResult` with deterministic content.
+        """
+        prompt = "\n".join(
+            text
+            for message in messages
+            if (text := self._stringify_message(message)).strip()
+        )
+        if self._looks_like_summary_request(prompt):
+            content = "integration summary"
+        else:
+            excerpt = " ".join(prompt.split()[-18:])
+            if excerpt:
+                content = f"integration reply: {excerpt}"
+            else:
+                content = "integration reply"
+
+        return ChatResult(
+            generations=[ChatGeneration(message=AIMessage(content=content))]
+        )
+
+    @property
+    def _llm_type(self) -> str:
+        """Return the LangChain model type identifier."""
+        return "deterministic-integration"
+
+    @staticmethod
+    def _stringify_message(message: BaseMessage) -> str:
+        """Flatten message content into plain text for deterministic responses.
+
+        Returns:
+            Plain-text content extracted from the message.
+        """
+        content = message.content
+        if isinstance(content, str):
+            return content
+        if isinstance(content, list):
+            parts: list[str] = []
+            for block in content:
+                if isinstance(block, str):
+                    parts.append(block)
+                elif isinstance(block, dict) and block.get("type") == "text":
+                    text = block.get("text")
+                    if isinstance(text, str):
+                        parts.append(text)
+            return " ".join(parts)
+        return str(content)
+
+    @staticmethod
+    def _looks_like_summary_request(prompt: str) -> bool:
+        """Detect the middleware's summary-generation prompt.
+
+        Returns:
+            `True` when the prompt appears to be a summarization request.
+        """
+        lowered = prompt.lower()
+        return (
+            "messages to summarize" in lowered
+            or "condense the following conversation" in lowered
+            or "<summary>" in lowered
+        )

bog_agents_cli/_version.py

@@ -0,0 +1,3 @@
+"""Version information for `bog-agents-cli`."""
+
+__version__ = "0.5.2"  # x-release-please-version