mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-27 04:20:09 +00:00
zeekay
95d8461196
- Update quality-gate.yml to only lint hanzo_mcp and tests directories
- Add ruff config to pyproject.toml:
- Exclude vscode-extension and node_modules
- Ignore lint rules that conflict with test patterns:
B023 (loop variable binding), B904 (raise from),
E741 (ambiguous vars), F401/F821/F841 (unused/undefined in tests)
- Apply per-file ignores for tests/ and __init__.py
- Apply ruff fixes for import sorting and format
141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
"""Tool mode loader for dynamic tool configuration."""
|
|
|
|
import os
|
|
from typing import Dict, Optional, Set
|
|
|
|
from hanzo_mcp.tools.common.mode import (
|
|
ModeRegistry,
|
|
activate_mode_from_env,
|
|
register_default_modes,
|
|
)
|
|
|
|
# Essential system tools that are ALWAYS enabled regardless of mode
|
|
# These are core infrastructure tools that should never be disabled
|
|
# NOTE: llm and consensus are NOT essential (heavy litellm dependency)
|
|
ESSENTIAL_SYSTEM_TOOLS: Set[str] = {
|
|
# Reasoning (lightweight)
|
|
"think", # Structured thinking
|
|
"critic", # Critical analysis
|
|
"agent", # Lightweight agent spawning
|
|
# Configuration and mode management
|
|
"config", # System configuration
|
|
"mode", # Mode switching
|
|
# Tool management (unified command)
|
|
"tool", # Unified tool management (install, enable, disable, list)
|
|
# Core system
|
|
"version", # Version info
|
|
"stats", # Statistics
|
|
# Memory (always available)
|
|
"memory", # Unified memory operations
|
|
}
|
|
|
|
# Heavy tools that are disabled by default (opt-in)
|
|
# These require large dependencies like litellm
|
|
HEAVY_TOOLS: Set[str] = {
|
|
"llm", # Requires litellm (~100MB+ deps)
|
|
"consensus", # Requires litellm
|
|
}
|
|
|
|
|
|
class ModeLoader:
|
|
"""Loads and manages tool modes for dynamic configuration."""
|
|
|
|
@staticmethod
|
|
def initialize_modes() -> None:
|
|
"""Initialize the mode system with defaults."""
|
|
# Initialize modes
|
|
register_default_modes()
|
|
|
|
# Check for mode from environment
|
|
activate_mode_from_env()
|
|
|
|
# If no mode set, use default
|
|
if not ModeRegistry.get_active():
|
|
default_mode = os.environ.get("HANZO_DEFAULT_MODE", "hanzo")
|
|
if ModeRegistry.get(default_mode):
|
|
ModeRegistry.set_active(default_mode)
|
|
|
|
@staticmethod
|
|
def get_enabled_tools_from_mode(
|
|
base_enabled_tools: Optional[Dict[str, bool]] = None,
|
|
force_mode: Optional[str] = None,
|
|
) -> Dict[str, bool]:
|
|
"""Get enabled tools configuration from active mode.
|
|
|
|
Args:
|
|
base_enabled_tools: Base configuration to merge with
|
|
force_mode: Force a specific mode (overrides active)
|
|
|
|
Returns:
|
|
Dictionary of tool enable states
|
|
"""
|
|
# Initialize if needed
|
|
if not ModeRegistry.list():
|
|
ModeLoader.initialize_modes()
|
|
|
|
# Get mode to use
|
|
tools_list = None
|
|
|
|
if force_mode:
|
|
# Set and get mode
|
|
if ModeRegistry.get(force_mode):
|
|
ModeRegistry.set_active(force_mode)
|
|
mode = ModeRegistry.get_active()
|
|
tools_list = mode.tools if mode else None
|
|
else:
|
|
# Check active mode
|
|
mode = ModeRegistry.get_active()
|
|
if mode:
|
|
tools_list = mode.tools
|
|
|
|
if not tools_list:
|
|
# No active mode, return base config
|
|
return base_enabled_tools or {}
|
|
|
|
# Start with base configuration
|
|
result = base_enabled_tools.copy() if base_enabled_tools else {}
|
|
|
|
# Get all possible tools from registry
|
|
from hanzo_mcp.config.tool_config import TOOL_REGISTRY
|
|
|
|
all_possible_tools = set(TOOL_REGISTRY.keys())
|
|
|
|
# Disable all tools first (clean slate for mode)
|
|
# EXCEPT essential system tools which are always enabled
|
|
for tool in all_possible_tools:
|
|
if tool in ESSENTIAL_SYSTEM_TOOLS:
|
|
result[tool] = True # Essential tools always enabled
|
|
else:
|
|
result[tool] = False
|
|
|
|
# Enable tools from mode
|
|
for tool in tools_list:
|
|
result[tool] = True
|
|
|
|
# Ensure all essential system tools are enabled (in case mode tried to disable them)
|
|
for tool in ESSENTIAL_SYSTEM_TOOLS:
|
|
result[tool] = True
|
|
|
|
return result
|
|
|
|
@staticmethod
|
|
def get_environment_from_mode() -> Dict[str, str]:
|
|
"""Get environment variables from active mode.
|
|
|
|
Returns:
|
|
Dictionary of environment variables
|
|
"""
|
|
# Check mode
|
|
mode = ModeRegistry.get_active()
|
|
if mode and mode.environment:
|
|
return mode.environment.copy()
|
|
|
|
return {}
|
|
|
|
@staticmethod
|
|
def apply_environment_from_mode() -> None:
|
|
"""Apply environment variables from active mode."""
|
|
env_vars = ModeLoader.get_environment_from_mode()
|
|
for key, value in env_vars.items():
|
|
os.environ[key] = value
|