code-muse 0.0.1__py3-none-any.whl

code_muse/config_agent.py

@@ -0,0 +1,168 @@
+"""Config: agent settings."""
+
+import configparser
+
+import code_muse.config as _config
+
+PACK_AGENT_NAMES = frozenset(
+    [
+        "pack-leader",
+        "bloodhound",
+        "shepherd",
+        "terrier",
+        "watchdog",
+        "retriever",
+    ]
+)
+UC_AGENT_NAMES = frozenset(["helios"])
+
+
+def get_pack_agents_enabled() -> bool:
+    """Return True if pack agents are enabled (default False).
+
+    When False (default), pack agents (pack-leader, bloodhound, shepherd,
+    terrier, watchdog, retriever) are hidden from `list_agents` tool and `/agents`
+    command. They cannot be invoked by other agents or selected by users.
+
+    When True, pack agents are available for use.
+    """
+    cfg_val = _config.get_value("enable_pack_agents")
+    if cfg_val is None:
+        return False
+    return str(cfg_val).strip().lower() in {"1", "true", "yes", "on"}
+
+
+def get_universal_constructor_enabled() -> bool:
+    """Return True if the Universal Constructor is enabled (default True).
+
+    The Universal Constructor allows agents to dynamically create, manage,
+    and execute custom tools at runtime. When enabled, agents can extend
+    their capabilities by writing Python code that becomes callable tools.
+
+    When False, the universal_constructor tool is not registered with agents.
+    """
+    cfg_val = _config.get_value("enable_universal_constructor")
+    if cfg_val is None:
+        return True  # Enabled by default
+    return str(cfg_val).strip().lower() in {"1", "true", "yes", "on"}
+
+
+def set_universal_constructor_enabled(enabled: bool) -> None:
+    """Enable or disable the Universal Constructor.
+
+    Args:
+        enabled: True to enable, False to disable
+    """
+    _config.set_value("enable_universal_constructor", "true" if enabled else "false")
+
+
+def get_user_agents_directory() -> str:
+    """Get the user's agents directory path.
+
+    Returns:
+        Path to the user's Muse agents directory.
+    """
+    # Ensure the agents directory exists
+    _config.AGENTS_DIR.mkdir(parents=True, exist_ok=True)
+    return str(_config.AGENTS_DIR)
+
+
+def get_project_agents_directory() -> str | None:
+    """Get the project-local agents directory path.
+
+    Looks for a .muse/agents/ directory in the current working directory.
+    Unlike get_user_agents_directory(), this does NOT create the directory
+    if it doesn't exist -- the team must create it intentionally.
+
+    Returns:
+        Path to the project's agents directory if it exists, or None.
+    """
+    project_agents_dir = _config.Path.cwd() / ".muse" / "agents"
+    if project_agents_dir.is_dir():
+        return str(project_agents_dir)
+    return None
+
+
+def get_agent_pinned_model(agent_name: str) -> str:
+    """Get the pinned model for a specific agent.
+
+    Args:
+        agent_name: Name of the agent to get the pinned model for.
+
+    Returns:
+        Pinned model name, or None if no model is pinned for this agent.
+    """
+    return _config.get_value(f"agent_model_{agent_name}")
+
+
+def set_agent_pinned_model(agent_name: str, model_name: str):
+    """Set the pinned model for a specific agent.
+
+    Args:
+        agent_name: Name of the agent to pin the model for.
+        model_name: Model name to pin to this agent.
+    """
+    _config.set_config_value(f"agent_model_{agent_name}", model_name)
+
+
+def clear_agent_pinned_model(agent_name: str):
+    """Clear the pinned model for a specific agent.
+
+    Args:
+        agent_name: Name of the agent to clear the pinned model for.
+    """
+    # We can't easily delete keys from configparser, so set to empty string
+    # which will be treated as None by get_agent_pinned_model
+    _config.set_config_value(f"agent_model_{agent_name}", "")
+
+
+def get_all_agent_pinned_models() -> dict:
+    """Get all agent-to-model pinnings from config.
+
+    Returns:
+        Dict mapping agent names to their pinned model names.
+        Only includes agents that have a pinned model (non-empty value).
+    """
+    config = configparser.ConfigParser()
+    config.read(_config.CONFIG_FILE)
+
+    pinnings = {}
+    if _config.DEFAULT_SECTION in config:
+        for key, value in config[_config.DEFAULT_SECTION].items():
+            if key.startswith("agent_model_") and value:
+                agent_name = key[len("agent_model_") :]
+                pinnings[agent_name] = value
+    return pinnings
+
+
+def get_agents_pinned_to_model(model_name: str) -> list:
+    """Get all agents that are pinned to a specific model.
+
+    Args:
+        model_name: The model name to look up.
+
+    Returns:
+        List of agent names pinned to this model.
+    """
+    all_pinnings = get_all_agent_pinned_models()
+    return [agent for agent, model in all_pinnings.items() if model == model_name]
+
+
+def get_default_agent() -> str:
+    """
+    Get the default agent name from muse.cfg.
+
+    Returns:
+        str: The default agent name, or "planning-agent" if not set.
+    """
+    return _config.get_value("default_agent") or "planning-agent"
+
+
+def set_default_agent(agent_name: str) -> None:
+    """
+    Set the default agent name in muse.cfg.
+
+    Args:
+        agent_name: The name of the agent to set as default.
+    """
+    _config.set_config_value("default_agent", agent_name)

code_muse/config_appearance.py

@@ -0,0 +1,241 @@
+"""Config: appearance settings."""
+
+import code_muse.config as _config
+
+
+def set_diff_highlight_style(style: str):
+    """Set the diff highlight style.
+
+    Note: Text mode has been removed. This function is kept for backwards compatibility
+    but does nothing. All diffs use beautiful syntax highlighting now!
+
+    Args:
+        style: Ignored (always uses 'highlight' mode)
+    """
+    # Do nothing - we always use highlight mode now!
+    pass
+
+
+_DEFAULT_DIFF_ADDITION_HEX = "#0b1f0b"  # darker green
+_DEFAULT_DIFF_DELETION_HEX = "#390e1a"  # wine
+
+
+def _coerce_to_hex(value: str | None, fallback: str) -> str:
+    """Normalize any color string to '#RRGGBB'.
+
+    Accepts:
+      - '#RRGGBB' hex strings (any case) — returned lowercased.
+      - Rich color names like 'green', 'orange1', 'bright_red'.
+      - 'rgb(r,g,b)' forms that Rich understands.
+
+    Anything Rich can't parse (including None/empty) falls back to ``fallback``.
+    This keeps downstream consumers like ``brighten_hex`` happy — they only
+    ever see a well-formed #RRGGBB string.
+    """
+    if not value:
+        return fallback
+    candidate = value.strip()
+    # Fast-path: already a valid #RRGGBB.
+    if (
+        len(candidate) == 7
+        and candidate.startswith("#")
+        and all(c in "0123456789abcdefABCDEF" for c in candidate[1:])
+    ):
+        return candidate.lower()
+    # Otherwise try Rich's parser (handles named colors, rgb(), etc.).
+    try:
+        from rich.color import Color  # local import keeps module import cheap
+
+        triplet = Color.parse(candidate).get_truecolor()
+        return f"#{triplet.red:02x}{triplet.green:02x}{triplet.blue:02x}"
+    except Exception:
+        return fallback
+
+
+def get_diff_addition_color() -> str:
+    """Get the base color for diff additions, always as a valid '#RRGGBB' hex.
+
+    Falls back to the default darker green if the configured value is missing
+    or unparseable.
+    """
+    return _coerce_to_hex(
+        _config.get_value("highlight_addition_color"), _DEFAULT_DIFF_ADDITION_HEX
+    )
+
+
+def set_diff_addition_color(color: str):
+    """Set the color for diff additions.
+
+    Accepts '#RRGGBB' hex, Rich color names ('green', 'bright_green', ...), or
+    'rgb(r,g,b)'. The value is normalized to '#RRGGBB' before being written so
+    downstream renderers never see a raw name.
+    """
+    _config.set_config_value(
+        "highlight_addition_color",
+        _coerce_to_hex(color, _DEFAULT_DIFF_ADDITION_HEX),
+    )
+
+
+def get_diff_deletion_color() -> str:
+    """Get the base color for diff deletions, always as a valid '#RRGGBB' hex.
+
+    Falls back to the default wine if the configured value is missing or
+    unparseable.
+    """
+    return _coerce_to_hex(
+        _config.get_value("highlight_deletion_color"), _DEFAULT_DIFF_DELETION_HEX
+    )
+
+
+def set_diff_deletion_color(color: str):
+    """Set the color for diff deletions.
+
+    Accepts '#RRGGBB' hex, Rich color names ('red', 'orange1', ...), or
+    'rgb(r,g,b)'. The value is normalized to '#RRGGBB' before being written so
+    downstream renderers never see a raw name.
+    """
+    _config.set_config_value(
+        "highlight_deletion_color",
+        _coerce_to_hex(color, _DEFAULT_DIFF_DELETION_HEX),
+    )
+
+
+DEFAULT_BANNER_COLORS = {
+    "thinking": "deep_sky_blue4",  # Sapphire - contemplation
+    "agent_response": "medium_purple4",  # Amethyst - main AI output
+    "shell_command": "dark_orange3",  # Amber - system commands
+    "read_file": "steel_blue",  # Steel - reading files
+    "edit_file": "dark_goldenrod",  # Gold - modifications (legacy)
+    "create_file": "dark_goldenrod",  # Gold - file creation
+    "replace_in_file": "dark_goldenrod",  # Gold - file modifications
+    "delete_snippet": "dark_goldenrod",  # Gold - snippet removal
+    "grep": "grey37",  # Silver - search results
+    "directory_listing": "dodger_blue2",  # Sky - navigation
+    "agent_reasoning": "dark_violet",  # Violet - deep thought
+    "invoke_agent": "deep_pink4",  # Ruby - agent invocation
+    "subagent_response": "sea_green3",  # Emerald - sub-agent success
+    "list_agents": "dark_slate_gray3",  # Slate - neutral listing
+    "universal_constructor": "dark_cyan",  # Teal - constructing tools
+    # Browser/Terminal tools - same color as edit_file (gold)
+    "terminal_tool": "dark_goldenrod",  # Gold - browser terminal operations
+    # User-initiated shell pass-through (! prefix) - distinct from agent's shell_command
+    "shell_passthrough": "medium_sea_green",  # Green - user's own shell commands
+}
+
+
+def get_banner_color(banner_name: str) -> str:
+    """Get the background color for a specific banner.
+
+    Args:
+        banner_name: The banner identifier (e.g., 'thinking', 'agent_response')
+
+    Returns:
+        Rich color name or hex code for the banner background
+    """
+    config_key = f"banner_color_{banner_name}"
+    val = _config.get_value(config_key)
+    if val:
+        return val
+    return DEFAULT_BANNER_COLORS.get(banner_name, "blue")
+
+
+def set_banner_color(banner_name: str, color: str):
+    """Set the background color for a specific banner.
+
+    Args:
+        banner_name: The banner identifier (e.g., 'thinking', 'agent_response')
+        color: Rich color name or hex code
+    """
+    config_key = f"banner_color_{banner_name}"
+    _config.set_config_value(config_key, color)
+
+
+def get_all_banner_colors() -> dict:
+    """Get all banner colors (configured or default).
+
+    Returns:
+        Dict mapping banner names to their colors
+    """
+    return {name: get_banner_color(name) for name in DEFAULT_BANNER_COLORS}
+
+
+def reset_banner_color(banner_name: str):
+    """Reset a banner color to its default.
+
+    Args:
+        banner_name: The banner identifier to reset
+    """
+    default_color = DEFAULT_BANNER_COLORS.get(banner_name, "blue")
+    set_banner_color(banner_name, default_color)
+
+
+def reset_all_banner_colors():
+    """Reset all banner colors to their defaults."""
+    for name, color in DEFAULT_BANNER_COLORS.items():
+        set_banner_color(name, color)
+
+
+def get_diff_context_lines() -> int:
+    """
+    Returns the user-configured number of context lines for diff display.
+    This controls how many lines of surrounding context are shown in diffs.
+    Defaults to 6 if unset or misconfigured.
+    Configurable by 'diff_context_lines' key.
+    """
+    val = _config.get_value("diff_context_lines")
+    try:
+        context_lines = int(val) if val else 6
+        # Apply reasonable bounds: minimum 0, maximum 50
+        return max(0, min(context_lines, 50))
+    except ValueError, TypeError:
+        return 6
+
+
+def get_suppress_thinking_messages() -> bool:
+    """
+    Checks muse.cfg for 'suppress_thinking_messages' (case-insensitive in value only).
+    Defaults to False if not set.
+    Allowed values for ON: 1, '1', 'true', 'yes', 'on' (all case-insensitive for value).
+    When enabled, thinking messages (agent_reasoning, planned_next_steps) will be hidden.
+    """
+    true_vals = {"1", "true", "yes", "on"}
+    cfg_val = _config.get_value("suppress_thinking_messages")
+    if cfg_val is not None:
+        return str(cfg_val).strip().lower() in true_vals
+    return False
+
+
+def set_suppress_thinking_messages(enabled: bool):
+    """Sets the suppress_thinking_messages configuration value.
+
+    Args:
+        enabled: Whether to suppress thinking messages
+    """
+    _config.set_config_value(
+        "suppress_thinking_messages", "true" if enabled else "false"
+    )
+
+
+def get_suppress_informational_messages() -> bool:
+    """
+    Checks muse.cfg for 'suppress_informational_messages' (case-insensitive in value only).
+    Defaults to False if not set.
+    Allowed values for ON: 1, '1', 'true', 'yes', 'on' (all case-insensitive for value).
+    When enabled, informational messages (info, success, warning) will be hidden.
+    """
+    true_vals = {"1", "true", "yes", "on"}
+    cfg_val = _config.get_value("suppress_informational_messages")
+    if cfg_val is not None:
+        return str(cfg_val).strip().lower() in true_vals
+    return False
+
+
+def set_suppress_informational_messages(enabled: bool):
+    """Sets the suppress_informational_messages configuration value.
+
+    Args:
+        enabled: Whether to suppress informational messages
+    """
+    _config.set_config_value(
+        "suppress_informational_messages", "true" if enabled else "false"
+    )