hanzoai/python-sdk
2
0
Fork 0
mirror of https://github.com/hanzoai/python-sdk.git synced 2026-10-23 20:23:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix CI Quality Gate: exclude vscode-extension from ruff, relax lint rules

Browse Source 
- 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
This commit is contained in:
zeekay
2026-01-17 21:49:47 -08:00
parent e450a908b1
commit 95d8461196
128 changed files with 527 additions and 491 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/quality-gate.yml
+1 -1
View File
@@ -157,7 +157,7 @@ jobs:
- name: "🔍 Lint with ruff"
run: |
cd pkg/hanzo-mcp
ruff check . --fix --exit-non-zero-on-fix
ruff check hanzo_mcp tests --fix --exit-non-zero-on-fix
- name: "🔍 Type check with mypy"
run: |
pkg/hanzo-mcp/hanzo_mcp/analytics/posthog_analytics.py
+6 -6
View File
@@ -7,16 +7,16 @@ This module provides analytics tracking for:
- A/B testing and feature flags
"""
import os
import time
import asyncio
import platform
import functools
import os
import platform
import time
import traceback
from typing import Any, Dict, TypeVar, Callable, Optional
from datetime import datetime
from dataclasses import dataclass
from datetime import datetime
from importlib.metadata import PackageNotFoundError, version
from typing import Any, Callable, Dict, Optional, TypeVar
# Try to import PostHog, but make it optional
try:
@@ -77,8 +77,8 @@ class Analytics:
return distinct_id
# Use hostname + username as fallback
import socket
import getpass
import socket
hostname = socket.gethostname()
username = getpass.getuser()
pkg/hanzo-mcp/hanzo_mcp/backends/sqlite_plugin.py
+4 -6
View File
@@ -1,13 +1,12 @@
"""SQLite backend plugin implementation."""
import asyncio
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, List, Optional
from hanzo_memory.models.memory import MemoryCreate
from hanzo_memory.db.sqlite_client import SQLiteMemoryClient
from hanzo_memory.models.memory import MemoryCreate
from ..plugin_interface import Capability, MemoryBackendPlugin
from ..plugin_interface import Capability
class SQLiteBackendPlugin:
@@ -58,7 +57,7 @@ class SQLiteBackendPlugin:
# For now, we'll store without embeddings to avoid dependency issues
# In a real implementation, we'd generate embeddings
memory_create = MemoryCreate(
MemoryCreate(
content=content,
memory_type=metadata.get("type", "general"),
importance=metadata.get("importance", 0.5),
@@ -69,7 +68,6 @@ class SQLiteBackendPlugin:
# Add memory directly to the database
import uuid
from datetime import datetime
memory_id = str(uuid.uuid4())
pkg/hanzo-mcp/hanzo_mcp/bridge.py
+5 -10
View File
@@ -4,21 +4,16 @@ This module provides MCP server functionality that allows Claude instances
to communicate with each other, enabling peer-to-peer agent networks.
"""
import os
import sys
import json
import argparse
import asyncio
import logging
import argparse
import os
from dataclasses import dataclass
from typing import Any, Dict, List, Optional
from dataclasses import asdict, dataclass
import mcp.server.fastmcp as mcp
from mcp import tool
from mcp.types import INTERNAL_ERROR, Tool, TextContent
from mcp.server.stdio import stdio_server
from mcp.server.models import InitializationOptions
from mcp.server.fastmcp import FastMCP
from mcp.server.models import InitializationOptions
from mcp.server.stdio import stdio_server
logger = logging.getLogger(__name__)
pkg/hanzo-mcp/hanzo_mcp/cli.py
+6 -6
View File
@@ -6,14 +6,14 @@ This prevents any stdout/stderr noise from imports that would corrupt the
MCP stdio transport used by Claude Desktop and other MCP clients.
"""
import os
import sys
import json
import signal
import logging
import argparse
from typing import Any, cast
import json
import logging
import os
import signal
import sys
from pathlib import Path
from typing import Any, cast
# Configure uvloop early (before any async imports) for better performance
# This is a no-op on Windows or if uvloop is not installed
pkg/hanzo-mcp/hanzo_mcp/cli_enhanced.py
+4 -4
View File
@@ -1,8 +1,8 @@
"""Enhanced command-line interface for the Hanzo AI server with full tool configuration."""
import os
import logging
import argparse
import logging
import os
from typing import Any, Dict
from hanzo_mcp.config import (
@@ -101,7 +101,7 @@ Examples:
# Add CLI flags for each tool
for tool_name, tool_config in TOOL_REGISTRY.items():
flag_name = tool_config.cli_flag.lstrip("-")
tool_config.cli_flag.lstrip("-")
help_text = f"{tool_config.description}"
if tool_config.enabled:
@@ -269,7 +269,7 @@ def apply_cli_overrides(args: argparse.Namespace) -> Dict[str, Any]:
# Handle individual tool flags
for tool_name, tool_config in TOOL_REGISTRY.items():
flag_name = tool_name.replace("_", "-")
tool_name.replace("_", "-")
if tool_config.enabled:
# Check for disable flag
pkg/hanzo-mcp/hanzo_mcp/cli_plugin.py
+5 -5
View File
@@ -1,8 +1,8 @@
#!/usr/bin/env python3
"""CLI for managing Hanzo MCP plugins."""
import sys
import argparse
import sys
from pathlib import Path
from hanzo_mcp.tools.common.plugin_loader import (
@@ -42,7 +42,7 @@ Examples:
)
# List command
list_parser = subparsers.add_parser("list", help="List installed plugins")
subparsers.add_parser("list", help="List installed plugins")
args = parser.parse_args()
@@ -51,10 +51,10 @@ Examples:
output_dir = args.output / args.name
try:
create_plugin_template(output_dir, args.name)
print(f"\n✅ Plugin template created successfully!")
print(f"\nTo use your plugin:")
print("\n✅ Plugin template created successfully!")
print("\nTo use your plugin:")
print(f"1. Edit the tool implementation in {output_dir / f'{args.name}_tool.py'}")
print(f"2. Restart Hanzo MCP to load the plugin")
print("2. Restart Hanzo MCP to load the plugin")
print(f"3. Add '{args.name}' to your mode's tool list")
except Exception as e:
print(f"❌ Error creating plugin: {e}", file=sys.stderr)
pkg/hanzo-mcp/hanzo_mcp/compute_nodes.py
+1 -1
View File
@@ -54,7 +54,7 @@ class ComputeNodeDetector:
if result.returncode == 0 and "Metal" in result.stdout:
# Parse GPU info from system_profiler
lines = result.stdout.split("\n")
for i, line in enumerate(lines):
for _i, line in enumerate(lines):
if "Chipset Model:" in line:
gpu_name = line.split(":")[1].strip()
gpus.append(
pkg/hanzo-mcp/hanzo_mcp/config.py
+2 -2
View File
@@ -1,10 +1,10 @@
"""Configuration system for the modular plugin architecture."""
import json
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, List, Optional
from pydantic import Field, BaseModel
from pydantic import BaseModel, Field
class BackendConfig(BaseModel):
pkg/hanzo-mcp/hanzo_mcp/core/base_agent.py
+14 -5
View File
@@ -6,16 +6,25 @@ following DRY principles and ensuring consistent behavior across all agents.
from __future__ import annotations
import os
import asyncio
import logging
import os
from abc import ABC, abstractmethod
from typing import Any, Dict, List, Generic, TypeVar, Optional, Protocol, runtime_checkable
from pathlib import Path
from dataclasses import dataclass, field
from datetime import datetime
from dataclasses import field, dataclass
from pathlib import Path
from typing import (
Any,
Dict,
Generic,
List,
Optional,
Protocol,
TypeVar,
runtime_checkable,
)
from .model_registry import ModelConfig, registry
from .model_registry import registry
logger = logging.getLogger(__name__)
pkg/hanzo-mcp/hanzo_mcp/core/model_registry.py
+2 -2
View File
@@ -8,9 +8,9 @@ Thread-safe singleton implementation.
from __future__ import annotations
import threading
from dataclasses import dataclass, field
from enum import Enum
from typing import Any, Set, Dict, List, Optional
from dataclasses import field, dataclass
from typing import Any, Dict, List, Optional, Set
class ModelProvider(Enum):
pkg/hanzo-mcp/hanzo_mcp/dev_cli.py
+1 -1
View File
@@ -3,8 +3,8 @@
Main CLI entry point for hanzo-dev command
"""
import sys
import asyncio
import sys
from hanzo_mcp.dev_tools import DevToolsCore
pkg/hanzo-mcp/hanzo_mcp/dev_server.py
+4 -4
View File
@@ -1,10 +1,10 @@
"""Development server with hot reload for Hanzo AI."""
import time
import asyncio
import logging
from typing import Set, Optional
import time
from pathlib import Path
from typing import Optional, Set
import watchdog.events
import watchdog.observers
@@ -167,7 +167,7 @@ class DevServer:
self.running = True
logger = logging.getLogger(__name__)
logger.info(f"\n🚀 Starting Hanzo AI in development mode...")
logger.info("\n🚀 Starting Hanzo AI in development mode...")
# Show compute nodes
try:
@@ -180,7 +180,7 @@ class DevServer:
# Silently ignore if compute node detection fails
pass
logger.info(f"🔧 Hot reload enabled - watching for file changes")
logger.info("🔧 Hot reload enabled - watching for file changes")
logger.info(f"📁 Project: {self.project_dir or 'current directory'}")
logger.info(f"🌐 Transport: {transport}\n")
pkg/hanzo-mcp/hanzo_mcp/dev_tools.py
+4 -14
View File
@@ -15,20 +15,11 @@ Features:
- Unified backend with SQLite vector storage
"""
import os
import json
import uuid
import asyncio
import logging
import sqlite3
import subprocess
from typing import Any, Dict, List, Union, Literal, Optional
from pathlib import Path
from datetime import datetime
from dataclasses import asdict, dataclass
from pydantic import Field, BaseModel
from hanzo_tools.lsp.lsp_tool import LSPTool
import uuid
from dataclasses import asdict
from typing import Any, Dict, List, Literal, Optional
from .unified_backend import TargetSpec, ToolResult, UnifiedBackend
@@ -171,7 +162,7 @@ class DevToolsCore:
# Fix all and organize imports
edit_result = await self.edit(
target=f"pkg:./...", op="code_action", only=["source.fixAll", "source.organizeImports"], language="go"
target="pkg:./...", op="code_action", only=["source.fixAll", "source.organizeImports"], language="go"
)
results.append(edit_result)
@@ -355,7 +346,6 @@ async def mcp_guard(
def main():
"""CLI entry point"""
import sys
import asyncio
from .dev_cli import main as cli_main
pkg/hanzo-mcp/hanzo_mcp/exact_mcp_server.py
+19 -8
View File
@@ -6,16 +6,27 @@ Enhanced MCP Server with Exact 6-Tool Implementation
Precise implementation of the 6 universal tools according to specification.
"""
import json
import asyncio
from typing import Any, Dict, List, Optional, Sequence
from typing import List
from pydantic import AnyUrl
from mcp.types import Tool, Resource, TextContent, ImageContent, LoggingLevel, EmbeddedResource
from mcp.server import Server, NotificationOptions
from mcp.server import NotificationOptions, Server
from mcp.server.models import InitializationOptions
from mcp.types import (
TextContent,
Tool,
)
from .exact_tools import FmtArgs, EditArgs, LintArgs, TestArgs, BuildArgs, GuardArgs, GuardRule, TargetSpec, tools
from .exact_tools import (
BuildArgs,
EditArgs,
FmtArgs,
GuardArgs,
GuardRule,
LintArgs,
TargetSpec,
TestArgs,
tools,
)
class ExactHanzoMCPServer:
@@ -366,7 +377,7 @@ class ExactHanzoMCPServer:
output += f"Execution time: {result.execution_time:.3f}s\n"
if result.touched_files:
output += f"\nModified files:\n"
output += "\nModified files:\n"
for f in result.touched_files:
output += f" - {f}\n"
@@ -377,7 +388,7 @@ class ExactHanzoMCPServer:
output += f"\nSTDERR:\n{result.stderr}\n"
if result.errors:
output += f"\nErrors:\n"
output += "\nErrors:\n"
for error in result.errors:
output += f" - {error}\n"
pkg/hanzo-mcp/hanzo_mcp/exact_tools.py
+9 -14
View File
@@ -7,22 +7,17 @@ Implements the precise specification for edit, fmt, test, build, lint, guard too
with proper target resolution, workspace detection, and backend selection.
"""
import os
import glob
import json
import uuid
import shutil
import asyncio
import logging
import sqlite3
import json
import os
import subprocess
from typing import Any, Dict, List, Union, Literal, Optional
from pathlib import Path
from dataclasses import asdict, dataclass, field
from datetime import datetime
from dataclasses import field, asdict, dataclass
from pathlib import Path
from typing import Any, Dict, List, Literal, Optional, Union
from pydantic import Field, BaseModel
from hanzo_tools.lsp.lsp_tool import LSPTool
from pydantic import BaseModel, Field
try:
from pydantic import ConfigDict
@@ -316,7 +311,7 @@ class TargetResolver:
def _resolve_go_package(self, pkg_spec: str, workspace: Dict, target_spec: TargetSpec) -> Dict[str, Any]:
"""Resolve Go package specification"""
root_path = Path(workspace["root"])
Path(workspace["root"])
if pkg_spec == "./...":
# All packages in workspace
@@ -1469,9 +1464,9 @@ tools = HanzoTools()
# CLI interface for testing
if __name__ == "__main__":
import sys
import asyncio
import argparse
import asyncio
import sys
async def main():
parser = argparse.ArgumentParser(description="Hanzo MCP Tools")
pkg/hanzo-mcp/hanzo_mcp/mcp_server.py
+8 -7
View File
@@ -8,17 +8,18 @@ Now includes the new unified development tools (edit, fmt, test, build, lint, gu
plus the existing tool suite.
"""
import json
import asyncio
from typing import Any, Dict, List, Optional, Sequence
from typing import List
from pydantic import AnyUrl
from mcp.types import Tool, Resource, TextContent, ImageContent, LoggingLevel, EmbeddedResource
from mcp.server import Server, NotificationOptions
from mcp.server import NotificationOptions, Server
from mcp.server.models import InitializationOptions
from mcp.types import (
TextContent,
Tool,
)
from .unified_backend import TargetSpec, ToolResult, backend
from .tools.dev_tools_mcp import dev_tools_server
from .unified_backend import TargetSpec, ToolResult, backend
class HanzoMCPServer:
@@ -488,7 +489,7 @@ class HanzoMCPServer:
output += f"\nErrors:\n{result.stderr}\n"
if result.errors:
output += f"\nIssues:\n"
output += "\nIssues:\n"
for error in result.errors:
output += f" - {error}\n"
pkg/hanzo-mcp/hanzo_mcp/memory_service.py
+2 -3
View File
@@ -1,11 +1,10 @@
"""Memory service with plugin support."""
import asyncio
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, List, Optional
from .plugin_interface import Capability, PluginRegistry, MemoryBackendPlugin
from .backends.sqlite_plugin import SQLiteBackendPlugin
from .plugin_interface import Capability, PluginRegistry
class PluginMemoryService:
pkg/hanzo-mcp/hanzo_mcp/plugin_interface.py
-3
View File
@@ -1,10 +1,7 @@
"""Plugin interface and registry for memory backends."""
import asyncio
from abc import ABC, abstractmethod
from enum import Enum
from typing import Any, Dict, List, Optional, Protocol, runtime_checkable
from dataclasses import dataclass
class Capability(str, Enum):
pkg/hanzo-mcp/hanzo_mcp/prompts/__init__.py
+20 -20
View File
@@ -2,37 +2,37 @@ import os
from mcp.server import FastMCP
from hanzo_mcp.prompts.utils import (
get_os_info,
get_git_info,
get_directory_structure,
)
from hanzo_mcp.prompts.tool_explorer import (
AGENT_TOOLS_HELP,
SHELL_TOOLS_HELP,
BATCH_TOOL_EXAMPLES,
TOOL_EXPLORER_PROMPT,
FILESYSTEM_TOOLS_HELP,
)
from hanzo_mcp.prompts.compact_conversation import COMPACT_CONVERSATION_PROMPT
from hanzo_mcp.prompts.create_release import CREATE_RELEASE_PROMPT
from hanzo_mcp.prompts.project_system import PROJECT_SYSTEM_PROMPT
from hanzo_mcp.prompts.enhanced_prompts import (
QUICK_START_PROMPT,
LSP_TOOLS_HELP_PROMPT,
PAGINATION_GUIDE_PROMPT,
PERFORMANCE_TIPS_PROMPT,
MEMORY_VECTOR_HELP_PROMPT,
CONFIGURATION_GUIDE_PROMPT,
DATABASE_TOOLS_HELP_PROMPT,
LSP_TOOLS_HELP_PROMPT,
MEMORY_VECTOR_HELP_PROMPT,
NETWORK_AGENT_GUIDE_PROMPT,
TROUBLESHOOTING_GUIDE_PROMPT,
PAGINATION_GUIDE_PROMPT,
PERFORMANCE_TIPS_PROMPT,
QUICK_START_PROMPT,
SECURITY_BEST_PRACTICES_PROMPT,
TROUBLESHOOTING_GUIDE_PROMPT,
)
from hanzo_mcp.prompts.compact_conversation import COMPACT_CONVERSATION_PROMPT
from hanzo_mcp.prompts.project_system import PROJECT_SYSTEM_PROMPT
from hanzo_mcp.prompts.project_todo_reminder import (
PROJECT_TODO_EMPTY_REMINDER,
get_project_todo_reminder,
)
from hanzo_mcp.prompts.tool_explorer import (
AGENT_TOOLS_HELP,
BATCH_TOOL_EXAMPLES,
FILESYSTEM_TOOLS_HELP,
SHELL_TOOLS_HELP,
TOOL_EXPLORER_PROMPT,
)
from hanzo_mcp.prompts.utils import (
get_directory_structure,
get_git_info,
get_os_info,
)
CONTINUE_FROM_LAST_SESSION_PROMPT = """<system-reminder>
This is a reminder that your todo list is currently empty. DO NOT mention this to the user explicitly because they are already aware. If you are working on tasks that would benefit from a todo list please use the TodoWrite tool to create one. If not, please feel free to ignore. Again do not mention this message to the user.
pkg/hanzo-mcp/hanzo_mcp/server.py
+3 -6
View File
@@ -6,22 +6,19 @@ Heavy imports happen only when the server is actually created, not at module loa
from __future__ import annotations
import os
import atexit
import signal
import logging
import os
import secrets
import warnings
import signal
import threading
import warnings
from typing import TYPE_CHECKING, Literal, cast, final
# Type-only imports - don't execute at runtime
if TYPE_CHECKING:
from mcp.server import FastMCP
from hanzo_tools.shell.session_storage import SessionStorage
from hanzo_mcp.server_enhanced import EnhancedFastMCP
from hanzo_mcp.tools.common.permissions import PermissionManager
# Suppress litellm deprecation warnings about event loop
warnings.filterwarnings("ignore", message="There is no current event loop", category=DeprecationWarning)
pkg/hanzo-mcp/hanzo_mcp/tools/__init__.py
+1 -1
View File
@@ -16,8 +16,8 @@ import logging
from typing import TYPE_CHECKING, Any
if TYPE_CHECKING:
from mcp.server import FastMCP
from hanzo_tools.core import BaseTool, PermissionManager
from mcp.server import FastMCP
logger = logging.getLogger(__name__)
pkg/hanzo-mcp/hanzo_mcp/tools/build_tool.py
+1 -1
View File
@@ -19,8 +19,8 @@ Backends:
- sol: forge build / hardhat compile
"""
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, Optional
from .dev_tools import DevResult, DevToolBase, create_dev_result
pkg/hanzo-mcp/hanzo_mcp/tools/common/auto_timeout.py
+11 -9
View File
@@ -4,20 +4,20 @@ This module provides automatic timeout and backgrounding for any MCP tool operat
that takes longer than the configured threshold (default: 2 minutes).
"""
import os
import json
import time
import uuid
import asyncio
import functools
from typing import Any, Tuple, Callable, Optional
from pathlib import Path
import json
import os
import time
import uuid
from collections.abc import Awaitable
from pathlib import Path
from typing import Any, Callable, Optional
from hanzo_async import append_file
from mcp.server.fastmcp import Context as MCPContext
from .timeout_parser import parse_timeout, format_timeout
from .timeout_parser import format_timeout, parse_timeout
class MCPToolTimeoutManager:
@@ -153,11 +153,13 @@ def with_auto_timeout(tool_name: str, timeout_manager: Optional[MCPToolTimeoutMa
# Method call: self, ctx, **params
self_or_ctx = args[0]
ctx = args[1]
call_func = lambda: func(self_or_ctx, ctx, **params)
def call_func():
return func(self_or_ctx, ctx, **params)
elif len(args) == 1:
# Function call: ctx, **params
ctx = args[0]
call_func = lambda: func(ctx, **params)
def call_func():
return func(ctx, **params)
else:
raise TypeError(f"Expected at least 1 argument (ctx), got {len(args)}")
pkg/hanzo-mcp/hanzo_mcp/tools/common/base.py
+6 -6
View File
@@ -5,22 +5,22 @@ for all tools used in Hanzo AI. These abstractions help ensure consistent tool
behavior and provide a foundation for tool registration and management.
"""
import inspect
import functools
import inspect
from abc import ABC, abstractmethod
from typing import Any, Callable, final
from collections.abc import Awaitable
from typing import Any, Callable, final
from mcp.server import FastMCP
from mcp.server.fastmcp import Context as MCPContext
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.error_logger import log_call_signature_error, log_tool_error
from hanzo_mcp.tools.common.permissions import PermissionManager
from hanzo_mcp.tools.common.validation import (
ValidationResult,
validate_path_parameter,
)
from hanzo_mcp.tools.common.permissions import PermissionManager
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.error_logger import log_tool_error, log_call_signature_error
def with_error_logging(tool_name: str) -> Callable:
@@ -49,7 +49,7 @@ def with_error_logging(tool_name: str) -> Callable:
log_call_signature_error(tool_name, expected, actual, e)
# Log the error
log_tool_error(tool_name, e, params=kwargs, context=f"Call signature mismatch or type error")
log_tool_error(tool_name, e, params=kwargs, context="Call signature mismatch or type error")
# Return user-friendly error message
return (
pkg/hanzo-mcp/hanzo_mcp/tools/common/cli_tool_factory.py
+4 -4
View File
@@ -13,13 +13,13 @@ Usage:
from __future__ import annotations
import os
import json
import asyncio
import json
import logging
from typing import TYPE_CHECKING, Any
import os
from dataclasses import asdict, dataclass, field
from pathlib import Path
from dataclasses import field, asdict, dataclass
from typing import TYPE_CHECKING, Any
if TYPE_CHECKING:
from mcp.server import FastMCP
pkg/hanzo-mcp/hanzo_mcp/tools/common/context.py
+1 -1
View File
@@ -4,8 +4,8 @@ This module provides an enhanced Context class that wraps the MCP Context
and adds additional functionality specific to Hanzo tools.
"""
from typing import ClassVar, final
from collections.abc import Iterable
from typing import ClassVar, final
from mcp.server.fastmcp import Context as MCPContext
from mcp.server.lowlevel.helper_types import ReadResourceContents
pkg/hanzo-mcp/hanzo_mcp/tools/common/context_fix.py
+3 -1
View File
@@ -9,9 +9,11 @@ DEPRECATED: Use hanzo_mcp.tools.common.decorators directly.
# Re-export for backward compatibility
from hanzo_mcp.tools.common.decorators import (
MockContext,
_is_valid_context as is_valid_context,
with_context_normalization,
)
from hanzo_mcp.tools.common.decorators import (
_is_valid_context as is_valid_context,
)
# Backward compatibility function
pkg/hanzo-mcp/hanzo_mcp/tools/common/decorators.py
+2 -2
View File
@@ -4,9 +4,9 @@ This module provides decorators that handle common cross-cutting concerns
for MCP tools, such as context normalization and error handling.
"""
import inspect
import functools
from typing import Any, TypeVar, Callable, cast
import inspect
from typing import Any, Callable, TypeVar, cast
F = TypeVar("F", bound=Callable[..., Any])
pkg/hanzo-mcp/hanzo_mcp/tools/common/enhanced_base.py
+2 -2
View File
@@ -76,7 +76,7 @@ class AutoRegisterTool(BaseTool, ABC):
sig = inspect.signature(call_method)
# Get type hints for proper typing
hints = get_type_hints(call_method)
get_type_hints(call_method)
# Create a dynamic handler function
tool_self = self
@@ -85,7 +85,7 @@ class AutoRegisterTool(BaseTool, ABC):
params = list(sig.parameters.items())
# Skip 'self' and 'ctx' parameters
tool_params = [(name, param) for name, param in params if name not in ("self", "ctx")]
[(name, param) for name, param in params if name not in ("self", "ctx")]
# Create the handler function dynamically
async def handler(ctx: MCPContext, **kwargs: Any) -> Any:
pkg/hanzo-mcp/hanzo_mcp/tools/common/entrypoint_loader.py
+3 -3
View File
@@ -12,14 +12,14 @@ Where TOOLS is a list of BaseTool subclasses.
from __future__ import annotations
import sys
import logging
from typing import TYPE_CHECKING, Any
import sys
from importlib.metadata import entry_points
from typing import TYPE_CHECKING, Any
if TYPE_CHECKING:
from mcp.server import FastMCP
from hanzo_tools.core import BaseTool, PermissionManager
from mcp.server import FastMCP
logger = logging.getLogger(__name__)
pkg/hanzo-mcp/hanzo_mcp/tools/common/error_logger.py
+2 -3
View File
@@ -4,13 +4,12 @@ This module provides comprehensive error logging for all MCP tool operations,
writing errors to ~/.hanzo/mcp/logs/ for debugging and analysis.
"""
import os
import json
import logging
import traceback
from typing import Any, Optional
from pathlib import Path
from datetime import datetime
from pathlib import Path
from typing import Any, Optional
class MCPErrorLogger:
pkg/hanzo-mcp/hanzo_mcp/tools/common/fastmcp_pagination.py
+6 -5
View File
@@ -3,12 +3,12 @@
This module provides pagination utilities optimized for FastMCP with minimal latency.
"""
import json
import time
import base64
import hashlib
from typing import Any, Dict, List, Union, Generic, TypeVar, Optional
from dataclasses import field, dataclass
import json
import time
from dataclasses import dataclass, field
from typing import Any, Dict, Generic, List, Optional, TypeVar, Union
T = TypeVar("T")
@@ -269,7 +269,8 @@ class TokenAwarePaginator:
# Default token estimation
if not estimate_func:
estimate_func = lambda x: estimate_tokens(json.dumps(x) if not isinstance(x, str) else x)
def estimate_func(x):
return estimate_tokens(json.dumps(x) if not isinstance(x, str) else x)
# Parse cursor
cursor_data = CursorData.from_cursor(cursor) if cursor else CursorData()
pkg/hanzo-mcp/hanzo_mcp/tools/common/forgiving_edit.py
+3 -3
View File
@@ -1,8 +1,8 @@
"""Forgiving edit helper for AI-friendly text matching."""
import re
import difflib
from typing import List, Tuple, Optional
import re
from typing import List, Optional, Tuple
class ForgivingEditHelper:
@@ -73,7 +73,7 @@ class ForgivingEditHelper:
# Find corresponding position in original
original_lines = haystack.split("\n")
norm_lines = norm_haystack.split("\n")
norm_haystack.split("\n")
start_pos = sum(len(line) + 1 for line in original_lines[:lines_before])
pkg/hanzo-mcp/hanzo_mcp/tools/common/mode.py
+2 -2
View File
@@ -1,12 +1,12 @@
"""Mode system for organizing development tools based on programmer personalities."""
import os
from typing import Set, Dict, List, Optional
from dataclasses import dataclass
from typing import Dict, List, Optional, Set
from hanzo_mcp.tools.common.personality import (
ToolPersonality,
PersonalityRegistry,
ToolPersonality,
ensure_agent_enabled,
register_default_personalities,
)
pkg/hanzo-mcp/hanzo_mcp/tools/common/mode_loader.py
+1 -1
View File
@@ -1,7 +1,7 @@
"""Tool mode loader for dynamic tool configuration."""
import os
from typing import Set, Dict, Optional
from typing import Dict, Optional, Set
from hanzo_mcp.tools.common.mode import (
ModeRegistry,
pkg/hanzo-mcp/hanzo_mcp/tools/common/paginated_base.py
+3 -3
View File
@@ -9,10 +9,10 @@ from typing import Any, Dict, Union
from mcp.server.fastmcp import Context as MCPContext
from hanzo_mcp.tools.common.base import BaseTool, handle_connection_errors
from hanzo_mcp.tools.common.pagination import CursorManager
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool, handle_connection_errors
from hanzo_mcp.tools.common.paginated_response import paginate_if_needed
from hanzo_mcp.tools.common.pagination import CursorManager
class PaginatedBaseTool(BaseTool):
@@ -160,7 +160,7 @@ def migrate_tool_to_paginated(tool_class):
def description(self):
# Add pagination info to description
desc = self._wrapped_tool.description
if not "pagination" in desc.lower():
if "pagination" not in desc.lower():
desc += "\n\nThis tool supports automatic pagination. If the response is too large, it will be split across multiple requests. Use the returned cursor to continue."
return desc
pkg/hanzo-mcp/hanzo_mcp/tools/common/paginated_response.py
+2 -2
View File
@@ -5,10 +5,10 @@ when they exceed token limits.
"""
import json
from typing import Any, Dict, List, Union, Optional
from typing import Any, Dict, List, Optional, Union
from hanzo_mcp.tools.common.truncate import estimate_tokens
from hanzo_mcp.tools.common.pagination import CursorManager
from hanzo_mcp.tools.common.truncate import estimate_tokens
class AutoPaginatedResponse:
pkg/hanzo-mcp/hanzo_mcp/tools/common/pagination.py
+2 -2
View File
@@ -4,10 +4,10 @@ This module provides utilities for implementing cursor-based pagination
according to the MCP pagination protocol.
"""
import json
import base64
from typing import Any, Dict, List, Generic, TypeVar, Optional
import json
from dataclasses import dataclass
from typing import Any, Dict, Generic, List, Optional, TypeVar
T = TypeVar("T")
pkg/hanzo-mcp/hanzo_mcp/tools/common/permissions.py
+4 -4
View File
@@ -4,14 +4,14 @@ Extends the base PermissionManager from hanzo-tools-core with additional
security features and MCP-specific functionality.
"""
import os
import sys
import json
import logging
import os
import sys
import tempfile
from typing import Any, TypeVar, final
from collections.abc import Awaitable, Callable
from pathlib import Path
from collections.abc import Callable, Awaitable
from typing import Any, TypeVar, final
# Import base from hanzo-tools-core
from hanzo_tools.core.permissions import PermissionManager as BasePermissionManager
pkg/hanzo-mcp/hanzo_mcp/tools/common/persona_adapter.py
+5 -6
View File
@@ -6,20 +6,19 @@ with the hanzo-mcp tool personality system.
import json
import logging
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, List, Optional
from hanzo_mcp.tools.common.personality import (
AI_TOOLS,
UNIX_TOOLS,
BUILD_TOOLS,
SEARCH_TOOLS,
VECTOR_TOOLS,
DATABASE_TOOLS,
ESSENTIAL_TOOLS,
CLIToolDef,
ToolPersonality,
SEARCH_TOOLS,
UNIX_TOOLS,
VECTOR_TOOLS,
PersonalityRegistry,
ToolPersonality,
)
logger = logging.getLogger(__name__)
pkg/hanzo-mcp/hanzo_mcp/tools/common/personality.py
+2 -2
View File
@@ -5,8 +5,8 @@ Actual personality profiles are loaded from the hanzo-persona package.
"""
import os
from typing import Set, Dict, List, Optional
from dataclasses import field, dataclass
from dataclasses import dataclass
from typing import Dict, List, Optional, Set
@dataclass
pkg/hanzo-mcp/hanzo_mcp/tools/common/plugin_loader.py
+6 -6
View File
@@ -1,13 +1,13 @@
"""Plugin loader for custom user tools."""
import importlib.util
import inspect
import json
import os
import sys
import json
import inspect
import importlib.util
from typing import Any, Dict, List, Type, Optional
from pathlib import Path
from dataclasses import dataclass
from pathlib import Path
from typing import Any, Dict, List, Optional, Type
from .base import BaseTool
@@ -269,7 +269,7 @@ Modify `{tool_name}_tool.py` to implement your custom functionality.
f.write(readme_content)
print(f"Created plugin template in {output_dir}")
print(f"Files created:")
print("Files created:")
print(f" - {tool_file}")
print(f" - {metadata_file}")
print(f" - {readme_file}")
pkg/hanzo-mcp/hanzo_mcp/tools/common/stats.py
+7 -7
View File
@@ -1,18 +1,18 @@
"""Comprehensive system and MCP statistics."""
from typing import Unpack, TypedDict, final, override
from pathlib import Path
from datetime import datetime
from pathlib import Path
from typing import TypedDict, Unpack, final, override
import psutil
from mcp.server.fastmcp import Context as MCPContext
from hanzo_tools.shell.run_background import RunBackgroundTool
from hanzo_tools.database.database_manager import DatabaseManager
from hanzo_tools.shell.run_background import RunBackgroundTool
from mcp.server.fastmcp import Context as MCPContext
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.mcp.mcp_add import McpAddTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.mcp.mcp_add import McpAddTool
class StatsParams(TypedDict, total=False):
pkg/hanzo-mcp/hanzo_mcp/tools/common/test_helpers.py
+1 -1
View File
@@ -1,6 +1,6 @@
"""Test helper classes for MCP tools testing."""
from typing import Any, Dict, List, Optional
from typing import Any, Dict
class PaginatedResponseWrapper:
pkg/hanzo-mcp/hanzo_mcp/tools/common/tool.py
+5 -6
View File
@@ -3,18 +3,17 @@
Combines install, uninstall, upgrade, enable, disable, and list into a single tool.
"""
from typing import Any, Unpack, Literal, Optional, Annotated, TypedDict, final, override
from pathlib import Path
from pydantic import Field
from typing import Annotated, Any, Literal, Optional, TypedDict, Unpack, final, override
# Import async I/O utilities
from hanzo_async import mkdir, read_json, write_json, path_exists
from hanzo_async import mkdir, path_exists, read_json, write_json
from mcp.server.fastmcp import Context as MCPContext
from pydantic import Field
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
Action = Annotated[
Literal[
@@ -332,7 +331,7 @@ Description: {config.description}"""
if result["success"]:
upgraded = [r["package"] for r in result.get("results", []) if r.get("success")]
return f"✓ Upgraded: {', '.join(upgraded) if upgraded else 'none'}"
return f"✗ Upgrade failed"
return "✗ Upgrade failed"
async def _handle_reload(self, name: Optional[str], tool_ctx: Any) -> str:
"""Hot-reload a tool package."""
pkg/hanzo-mcp/hanzo_mcp/tools/common/tool_disable.py
+3 -3
View File
@@ -1,14 +1,14 @@
"""Disable tools dynamically."""
from typing import Unpack, Annotated, TypedDict, final, override
from typing import Annotated, TypedDict, Unpack, final, override
from pydantic import Field
from mcp.server.fastmcp import Context as MCPContext
from pydantic import Field
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.common.tool_enable import ToolEnableTool
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
ToolName = Annotated[
str,
pkg/hanzo-mcp/hanzo_mcp/tools/common/tool_enable.py
+4 -5
View File
@@ -1,17 +1,16 @@
"""Enable tools dynamically."""
from typing import Unpack, Annotated, TypedDict, final, override
from pathlib import Path
from pydantic import Field
from typing import Annotated, TypedDict, Unpack, final, override
# Import async I/O utilities
from hanzo_async import mkdir, read_json, write_json, path_exists
from hanzo_async import mkdir, path_exists, read_json, write_json
from mcp.server.fastmcp import Context as MCPContext
from pydantic import Field
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
ToolName = Annotated[
str,
pkg/hanzo-mcp/hanzo_mcp/tools/common/tool_install.py
+4 -4
View File
@@ -7,14 +7,14 @@ Allows the AI to:
- Hot-reload without restart
"""
from typing import Any, Unpack, Literal, Annotated, TypedDict, final, override
from typing import Annotated, Literal, TypedDict, Unpack, final, override
from pydantic import Field
from mcp.server.fastmcp import Context as MCPContext
from pydantic import Field
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
Action = Annotated[
Literal[
@@ -141,7 +141,7 @@ Sources:
errors = [
f"{r['package']}: {r.get('error')}" for r in result.get("results", []) if not r.get("success")
]
return f"✗ Some upgrades failed:\n" + "\n".join(errors)
return "✗ Some upgrades failed:\n" + "\n".join(errors)
elif action == "reload":
if not package:
pkg/hanzo-mcp/hanzo_mcp/tools/common/tool_list.py
+4 -4
View File
@@ -1,14 +1,14 @@
"""List all available tools and their status."""
from typing import Unpack, Optional, Annotated, TypedDict, final, override
from typing import Annotated, Optional, TypedDict, Unpack, final, override
from pydantic import Field
from mcp.server.fastmcp import Context as MCPContext
from pydantic import Field
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.context import create_tool_context
from hanzo_mcp.tools.common.tool_enable import ToolEnableTool
from hanzo_mcp.tools.common.auto_timeout import auto_timeout
ShowDisabled = Annotated[
bool,
@@ -183,7 +183,7 @@ Use 'tool_enable' and 'tool_disable' to change tool status.
category_filter = params.get("category")
# Get all tool states
all_states = ToolEnableTool.get_all_states()
ToolEnableTool.get_all_states()
output = []
pkg/hanzo-mcp/hanzo_mcp/tools/common/tool_registry.py
+6 -8
View File
@@ -11,17 +11,15 @@ Note: This is distinct from:
- config/tool_config.py::DynamicToolRegistry - For discovering tools from entry points
"""
import os
import sys
import json
import asyncio
import logging
import importlib
import subprocess
from typing import Any, Type, Callable, Optional
from pathlib import Path
import json
import logging
import sys
from dataclasses import dataclass, field
from datetime import datetime
from dataclasses import field, dataclass
from pathlib import Path
from typing import Any, Callable, Optional, Type
logger = logging.getLogger(__name__)
pkg/hanzo-mcp/hanzo_mcp/tools/common/version_tool.py
-1
View File
@@ -1,7 +1,6 @@
"""Version tool for hanzo-mcp."""
import sys
from typing import Any
from hanzo_async import using_uvloop
from mcp.server.fastmcp import FastMCP
pkg/hanzo-mcp/hanzo_mcp/tools/dev_tools.py
+2 -5
View File
@@ -9,14 +9,11 @@ and is go.work aware.
"""
import os
import json
import subprocess
from enum import Enum
from typing import Any, Dict, List, Union, Literal, Optional
from pathlib import Path
from dataclasses import field, dataclass
from typing import Any, Dict, List, Literal, Optional, Union
from pydantic import Field, BaseModel
from pydantic import BaseModel, Field
# Common types
LanguageType = Literal["auto", "go", "ts", "py", "rs", "cc", "sol", "schema"]
pkg/hanzo-mcp/hanzo_mcp/tools/dev_tools_mcp.py
+5 -5
View File
@@ -17,18 +17,18 @@ Each tool supports:
- Composable workflows
"""
from typing import Any, Dict, List, Optional
from typing import Any, Dict, List
from mcp.types import Tool, TextContent
from mcp.server import Server
from mcp.types import TextContent, Tool
try:
from .fmt_tool import fmt_tool_handler
from .build_tool import build_tool_handler
from .edit_tool import edit_tool_handler
from .fmt_tool import fmt_tool_handler
from .guard_tool import guard_tool_handler
from .lint_tool import lint_tool_handler
from .test_tool import test_tool_handler
from .build_tool import build_tool_handler
from .guard_tool import guard_tool_handler
except ImportError:
# Fallback for testing
print("Warning: Could not import all dev tool handlers")
pkg/hanzo-mcp/hanzo_mcp/tools/edit_tool.py
+1 -4
View File
@@ -13,10 +13,7 @@ Operations:
Supports gopls, tsserver, pyright, rust-analyzer, clangd
"""
import json
import asyncio
from typing import Any, Dict, List, Union, Optional
from pathlib import Path
from typing import Any, Dict, List, Optional
from .dev_tools import DevResult, DevToolBase, create_dev_result
pkg/hanzo-mcp/hanzo_mcp/tools/fmt_tool.py
-1
View File
@@ -15,7 +15,6 @@ Backends:
"""
from typing import Any, Dict, List, Optional
from pathlib import Path
from .dev_tools import DevResult, DevToolBase, create_dev_result
pkg/hanzo-mcp/hanzo_mcp/tools/guard_tool.py
+3 -4
View File
@@ -16,10 +16,9 @@ Example rules for Hanzo ecosystem:
"""
import re
import glob
from typing import Any, Dict, List, Optional
from pathlib import Path
from dataclasses import dataclass
from pathlib import Path
from typing import Any, Dict, List, Optional
from .dev_tools import DevResult, DevToolBase, create_dev_result
@@ -218,7 +217,7 @@ class GuardTool(DevToolBase):
for file_path in files:
# Make path relative to workspace root for matching
try:
rel_path = Path(file_path).relative_to(root_path)
Path(file_path).relative_to(root_path)
if Path(file_path).match(glob_pattern):
matching_files.append(file_path)
except ValueError:
pkg/hanzo-mcp/hanzo_mcp/tools/lint_tool.py
+1 -1
View File
@@ -17,8 +17,8 @@ Options:
- fix: Apply fixes where supported
"""
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, Optional
from .dev_tools import DevResult, DevToolBase, create_dev_result
pkg/hanzo-mcp/hanzo_mcp/tools/test_tool.py
+1 -2
View File
@@ -17,9 +17,8 @@ Backends:
- rs: cargo test with -p, --features
"""
import re
from typing import Any, Dict, List, Optional
from pathlib import Path
from typing import Any, Dict, Optional
from .dev_tools import DevResult, DevToolBase, create_dev_result
pkg/hanzo-mcp/hanzo_mcp/types.py
+1 -1
View File
@@ -1,7 +1,7 @@
"""Type definitions for Hanzo MCP tools."""
from typing import Any, Dict, Optional
from dataclasses import dataclass
from typing import Any, Dict, Optional
@dataclass
pkg/hanzo-mcp/hanzo_mcp/unified_backend.py
+9 -10
View File
@@ -14,19 +14,19 @@ Features:
- Cross-language tool execution
"""
import os
import json
import uuid
import asyncio
import json
import logging
import os
import sqlite3
import subprocess
from typing import Any, Dict, List, Union, Literal, Optional
from pathlib import Path
from datetime import datetime
import uuid
from dataclasses import asdict, dataclass
from datetime import datetime
from pathlib import Path
from typing import Any, Dict, List, Optional, Union
from pydantic import Field, BaseModel
from pydantic import BaseModel, Field
# Core Types
@@ -219,7 +219,6 @@ class CodebaseIndexer:
(file_path, content_hash, language, file_size, modified_time),
)
file_id = conn.lastrowid
# Symbol extraction handled by LSP integration in the tools layer
# See hanzo_mcp/tools/common/lsp.py for language-specific parsing
@@ -394,7 +393,7 @@ class UnifiedBackend:
# LSP code actions
file_path = kwargs.get("file")
range_spec = kwargs.get("range", {})
only = kwargs.get("only", [])
kwargs.get("only", [])
actions = await self.lsp_bridge.code_actions(
file_path, range_spec.get("start", {}).get("line", 0), range_spec.get("end", {}).get("line", 0)
@@ -531,8 +530,8 @@ backend = UnifiedBackend()
if __name__ == "__main__":
# CLI interface for testing
import sys
import asyncio
import sys
async def main():
if len(sys.argv) < 2:
pkg/hanzo-mcp/hanzo_mcp/utils/event_loop.py
+2 -2
View File
@@ -6,11 +6,11 @@ with optional uvloop support for improved performance on Linux/macOS.
Uses hanzo_async for unified async I/O configuration.
"""
import sys
import asyncio
import sys
from typing import Optional
from hanzo_async import using_uvloop, configure_loop
from hanzo_async import configure_loop, using_uvloop
def configure_event_loop(*, quiet: bool = False) -> Optional[str]:
pkg/hanzo-mcp/pyproject.toml
+32
View File
@@ -190,3 +190,35 @@ exclude_lines = [
[tool.coverage.html]
directory = "htmlcov"
[tool.ruff]
exclude = [
".git",
".venv",
"venv",
"__pycache__",
"build",
"dist",
"*.egg-info",
"node_modules",
"vscode-extension",
]
[tool.ruff.lint]
select = ["E", "F", "I", "B"]
ignore = [
"E501", # Line too long
"B017", # Blind exception test
"B023", # Function definition loop variable
"B904", # Raise without from in except
"E741", # Ambiguous variable name
"E402", # Module level import not at top
"F401", # Unused imports (handled by tests themselves)
"F821", # Undefined name (tests may have conditional imports)
"F841", # Local variable assigned but never used
"B007", # Loop control variable not used
]
[tool.ruff.lint.per-file-ignores]
"tests/*" = ["F401", "F821", "F841", "B007"] # Relax rules for test files
"*/__init__.py" = ["F401"] # Allow unused imports in __init__.py
pkg/hanzo-mcp/tests/benchmark_search.py
+11 -10
View File
@@ -1,21 +1,22 @@
"""Benchmark suite for search and database storage performance."""
import sys
import json
import time
import asyncio
import json
import statistics
from typing import Any, Dict, List
import sys
import time
from pathlib import Path
from typing import Any, Dict, List
# Add project root to path for imports
sys.path.insert(0, str(Path(__file__).parent.parent))
from hanzo_mcp.tools.common.permissions import PermissionManager
from hanzo_tools.filesystem.search_tool import SearchTool
from hanzo_mcp.tools.vector.ast_analyzer import ASTAnalyzer
from hanzo_mcp.tools.vector.vector_index import VectorIndexTool
from hanzo_mcp.tools.vector.project_manager import ProjectVectorManager
from hanzo_mcp.tools.vector.vector_index import VectorIndexTool
from hanzo_tools.filesystem.search_tool import SearchTool
from hanzo_mcp.tools.common.permissions import PermissionManager
class SearchBenchmark:
@@ -187,7 +188,7 @@ def apply_transformations(record: Any, params: Dict[str, Any]) -> Any:
async def benchmark_indexing(self, project_path: Path) -> Dict[str, Any]:
"""Benchmark vector database indexing performance."""
print(f"\\n=== Benchmarking Indexing Performance ===")
print("\\n=== Benchmarking Indexing Performance ===")
print(f"Project: {project_path}")
# Initialize project manager
@@ -265,7 +266,7 @@ def apply_transformations(record: Any, params: Dict[str, Any]) -> Any:
async def benchmark_search_performance(self, project_path: Path, queries: List[str]) -> Dict[str, Any]:
"""Benchmark search performance across different query types."""
print(f"\\n=== Benchmarking Search Performance ===")
print("\\n=== Benchmarking Search Performance ===")
# Initialize tools
project_manager = ProjectVectorManager(
@@ -341,7 +342,7 @@ def apply_transformations(record: Any, params: Dict[str, Any]) -> Any:
async def benchmark_ast_analysis(self, project_path: Path) -> Dict[str, Any]:
"""Benchmark AST analysis performance."""
print(f"\\n=== Benchmarking AST Analysis ===")
print("\\n=== Benchmarking AST Analysis ===")
analyzer = ASTAnalyzer()
python_files = list(project_path.rglob("*.py"))
pkg/hanzo-mcp/tests/benchmark_simple.py
+6 -5
View File
@@ -1,18 +1,19 @@
"""Simplified benchmark script for testing search on this project."""
# Add project root to path
import asyncio
import sys
import time
import asyncio
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent.parent))
from hanzo_tools.filesystem.grep import Grep
from hanzo_tools.filesystem.ast_tool import ASTTool
from hanzo_mcp.tools.common.permissions import PermissionManager
from hanzo_tools.filesystem.search_tool import SearchTool
from hanzo_mcp.tools.vector.ast_analyzer import ASTAnalyzer
from hanzo_tools.filesystem.ast_tool import ASTTool
from hanzo_tools.filesystem.grep import Grep
from hanzo_tools.filesystem.search_tool import SearchTool
from hanzo_mcp.tools.common.permissions import PermissionManager
async def simple_benchmark():
pkg/hanzo-mcp/tests/conftest.py
+5 -5
View File
@@ -16,14 +16,14 @@ import pytest
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from test_utils import (
TestContext,
ToolTestHelper,
TestEnvironment,
TestDataGenerator,
FileSystemTestHelper,
TestContext,
TestDataGenerator,
TestEnvironment,
ToolTestHelper,
create_mock_ctx,
create_test_server,
create_permission_manager,
create_test_server,
)
# Set environment variables for testing
pkg/hanzo-mcp/tests/e2e/test_litellm_integration.py
+1 -1
View File
@@ -6,8 +6,8 @@ Only run in CI with proper secrets configured.
import os
import pytest
import litellm
import pytest
@pytest.mark.skipif(
pkg/hanzo-mcp/tests/manual/test_filters_manual.py
+1 -1
View File
@@ -1,9 +1,9 @@
#!/usr/bin/env python3
"""Test find tool filters."""
import asyncio
import os
import sys
import asyncio
import tempfile
from pathlib import Path
pkg/hanzo-mcp/tests/test_agent/test_agent_tool.py
+2 -2
View File
@@ -1,12 +1,12 @@
"""Tests for the agent tool implementation."""
import os
import json
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_tools.agent.agent_tool import AgentTool
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.permissions import PermissionManager
pkg/hanzo-mcp/tests/test_agent/test_litellm_providers.py
+3 -3
View File
@@ -1,12 +1,12 @@
"""Test LiteLLM with different providers."""
import os
import pytest
import litellm
from hanzo_mcp.tools.common.base import BaseTool
import pytest
from hanzo_tools.agent.tool_adapter import convert_tools_to_openai_functions
from hanzo_mcp.tools.common.base import BaseTool
class EchoTool(BaseTool):
"""A simple tool that echoes back the input."""
pkg/hanzo-mcp/tests/test_agent/test_prompt.py
+4 -3
View File
@@ -5,11 +5,12 @@ from unittest.mock import MagicMock
import pytest
from hanzo_tools.agent.prompt import (
get_default_model,
get_system_prompt,
get_model_parameters,
get_allowed_agent_tools,
get_default_model,
get_model_parameters,
get_system_prompt,
)
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_mcp.tools.common.permissions import PermissionManager
pkg/hanzo-mcp/tests/test_agent/test_tool_adapter.py
+2 -1
View File
@@ -3,12 +3,13 @@
from unittest.mock import MagicMock
import pytest
from hanzo_mcp.tools.common.base import BaseTool
from hanzo_tools.agent.tool_adapter import (
convert_tool_parameters,
convert_tools_to_openai_functions,
)
from hanzo_mcp.tools.common.base import BaseTool
class TestToolAdapter:
"""Test cases for the agent tool adapter module."""
pkg/hanzo-mcp/tests/test_batch_tool_edge_cases.py
+3 -3
View File
@@ -1,11 +1,11 @@
"""Test edge cases for the batch tool to prevent errors."""
import asyncio
from unittest.mock import Mock, AsyncMock, patch
from unittest.mock import AsyncMock, Mock, patch
import pytest
from mcp.server.fastmcp import Context as MCPContext
from hanzo_mcp.tools.common.batch_tool import BatchTool
from mcp.server.fastmcp import Context as MCPContext
class TestBatchToolEdgeCases:
@@ -172,7 +172,7 @@ class TestBatchToolEdgeCases:
invocations = [{"tool_name": "tool1", "input": {"id": i}} for i in range(20)]
start_time = asyncio.get_event_loop().time()
result = await batch_tool.call(ctx=mock_ctx, description="Concurrent test", invocations=invocations)
await batch_tool.call(ctx=mock_ctx, description="Concurrent test", invocations=invocations)
end_time = asyncio.get_event_loop().time()
# Check that execution was concurrent but limited
pkg/hanzo-mcp/tests/test_cli.py
+3 -2
View File
@@ -2,12 +2,13 @@
import os
import sys
from typing import Callable
from pathlib import Path
from typing import Callable
from unittest.mock import MagicMock, patch
import pytest
from hanzo_mcp.cli import main, install_claude_desktop_config
from hanzo_mcp.cli import install_claude_desktop_config, main
class TestCLI:
pkg/hanzo-mcp/tests/test_cli_tools.py
+6 -10
View File
@@ -1,26 +1,23 @@
"""Test suite for CLI tools in batch operations."""
import os
import time
import shutil
import asyncio
from typing import Any, Dict
from unittest.mock import Mock, AsyncMock, MagicMock, patch
import time
from unittest.mock import AsyncMock, MagicMock, Mock, patch
import pytest
from hanzo_mcp.tools.common.batch_tool import BatchTool
from hanzo_tools.agent.cli_tools import (
GrokCLITool,
AiderCLITool,
ClaudeCLITool,
ClaudeCodeCLITool,
ClineCLITool,
CodexCLITool,
ClaudeCLITool,
GeminiCLITool,
GrokCLITool,
HanzoDevCLITool,
OpenHandsCLITool,
ClaudeCodeCLITool,
OpenHandsShortCLITool,
)
from hanzo_mcp.tools.common.batch_tool import BatchTool
class TestCLITools:
@@ -319,7 +316,6 @@ class TestBatchWithCLITools:
@pytest.mark.asyncio
async def test_batch_cli_tools_parallel_execution(self, batch_tool, mock_context):
"""Test that CLI tools execute in parallel in batch."""
import time
# Add delay to mock executions
for tool in batch_tool.tools.values():
pkg/hanzo-mcp/tests/test_common/test_permissions.py
+2 -1
View File
@@ -4,9 +4,10 @@ import os
from pathlib import Path
import pytest
from hanzo_mcp.tools.common.permissions import (
PermissionManager,
PermissibleOperation,
PermissionManager,
)
pkg/hanzo-mcp/tests/test_e2e_demo.py
+2 -2
View File
@@ -11,10 +11,10 @@ try:
sys.path.insert(0, str(Path(__file__).parent.parent.parent / "hanzo-network" / "src"))
from hanzo_network import (
create_tool,
create_local_agent,
check_local_llm_status,
create_local_agent,
create_local_distributed_network,
create_tool,
)
HANZO_NETWORK_AVAILABLE = True
pkg/hanzo-mcp/tests/test_e2e_simple.py
+3 -2
View File
@@ -11,9 +11,10 @@ def test_imports():
"""Test that all imports work correctly."""
# Test hanzo-network imports
# Test hanzo-mcp imports
from hanzo_mcp import __version__ as mcp_version
from hanzo_network import __version__ as network_version
from hanzo_mcp import __version__ as mcp_version
assert network_version is not None
assert mcp_version is not None
print(f"✅ All imports successful! hanzo-network: {network_version}, hanzo-mcp: {mcp_version}")
@@ -31,7 +32,7 @@ def test_hanzo_net_provider():
def test_local_agent_creation():
"""Test creating a local agent."""
from hanzo_network import create_tool, create_local_agent
from hanzo_network import create_local_agent, create_tool
def dummy_tool(text: str) -> str:
return f"Processed: {text}"
pkg/hanzo-mcp/tests/test_error_logging.py
+5 -4
View File
@@ -1,14 +1,15 @@
"""Tests for MCP error logging functionality."""
from pathlib import Path
from datetime import datetime
from pathlib import Path
import pytest
from hanzo_mcp.tools.common.error_logger import (
MCPErrorLogger,
log_tool_error,
get_error_logger,
log_call_signature_error,
log_tool_error,
)
@@ -156,9 +157,9 @@ class TestErrorLoggingIntegration:
@pytest.mark.asyncio
async def test_read_tool_with_error_logging(self, tmp_path):
"""Test that ReadTool logs errors correctly."""
from mcp.server.fastmcp import Context as MCPContext
from hanzo_mcp.tools.common import error_logger as el_module
from hanzo_tools.filesystem.read import ReadTool
from hanzo_mcp.tools.common import error_logger as el_module
from hanzo_mcp.tools.common.permissions import PermissionManager
# Override global logger
pkg/hanzo-mcp/tests/test_exact_tools_contracts.py
+2 -2
View File
@@ -1,8 +1,8 @@
import asyncio
from pathlib import Path
import pytest
from hanzo_mcp.exact_tools import FmtArgs, EditArgs, GuardArgs, GuardRule, HanzoTools, TargetSpec
from hanzo_mcp.exact_tools import EditArgs, GuardArgs, GuardRule, HanzoTools, TargetSpec
@pytest.mark.asyncio
pkg/hanzo-mcp/tests/test_failure_cases.py
-1
View File
@@ -17,7 +17,6 @@ def test_no_api_key():
try:
# Try to create swarm without API key
model = "anthropic/claude-3-5-sonnet-20241022"
api_key = os.environ.get("ANTHROPIC_API_KEY") or os.environ.get("CLAUDE_API_KEY")
if not api_key:
pkg/hanzo-mcp/tests/test_filesystem/test_fs_tools.py
+1 -1
View File
@@ -9,7 +9,7 @@ import pytest
if TYPE_CHECKING:
from hanzo_mcp.tools.common.permissions import PermissionManager
from hanzo_tools.filesystem import Edit, Write, ReadTool, get_filesystem_tools
from hanzo_tools.filesystem import Edit, ReadTool, Write, get_filesystem_tools
class TestRefactoredFileTools:
pkg/hanzo-mcp/tests/test_find_tool_ffind.py
+3 -3
View File
@@ -1,7 +1,7 @@
"""Test FindTool with ffind for performance."""
import time
import tempfile
import time
from pathlib import Path
import pytest
@@ -200,7 +200,7 @@ class TestFindToolFFind:
page1_data = result_page1.data
assert len(page1_data["results"]) == 5
assert page1_data["pagination"]["page"] == 1
assert page1_data["pagination"]["has_next"] == True
assert page1_data["pagination"]["has_next"]
# Get second page
result_page2 = await find_tool.run(
@@ -245,7 +245,7 @@ class TestFindToolFFind:
result_ffind = await find_tool.run(pattern="*.py", path=test_directory, type="file")
ffind_time = time.time() - start_time
print(f"\nPerformance comparison:")
print("\nPerformance comparison:")
print(f"Python implementation: {python_time:.3f}s")
print(f"ffind implementation: {ffind_time:.3f}s")
print(f"Speedup: {python_time / ffind_time:.1f}x")
pkg/hanzo-mcp/tests/test_find_tool_integration.py
+4 -4
View File
@@ -1,15 +1,15 @@
"""Integration test for FindTool registration and functionality."""
import os
import time
import asyncio
import os
import tempfile
import time
from pathlib import Path
import pytest
from hanzo_mcp.server import HanzoMCPServer
from hanzo_mcp.tools.search import create_find_tool
from hanzo_mcp.server import HanzoMCPServer
from tests.test_utils import ToolTestHelper
@@ -87,7 +87,7 @@ async def test_find_tool_direct_usage(tool_helper):
)
fuzzy_results = result.data["results"]
fuzzy_names = [f["name"] for f in fuzzy_results]
[f["name"] for f in fuzzy_results]
# TODO: Fix fuzzy search with ffind or use Python implementation
# assert "utils.py" in fuzzy_names
# For now, just check that the search completes without error
pkg/hanzo-mcp/tests/test_find_tool_registration.py
+2 -1
View File
@@ -3,8 +3,9 @@
import asyncio
from fastmcp import FastMCP
from hanzo_mcp.server import HanzoMCPServer
from hanzo_tools.filesystem import register_filesystem_tools
from hanzo_mcp.server import HanzoMCPServer
from hanzo_mcp.tools.common.permissions import PermissionManager
pkg/hanzo-mcp/tests/test_git_ingestion.py
+2 -2
View File
@@ -1,9 +1,9 @@
"""Tests for Git history ingestion into vector store."""
import tempfile
import subprocess
from pathlib import Path
import tempfile
from datetime import datetime
from pathlib import Path
import pytest
from hanzo_mcp.tools.vector.infinity_store import InfinityVectorStore
pkg/hanzo-mcp/tests/test_hanzo_agents_integration.py
+1 -1
View File
@@ -43,7 +43,7 @@ except Exception as e:
# Test marketplace
try:
from hanzo_agents.core.marketplace import ServiceType, AgentMarketplace
from hanzo_agents.core.marketplace import AgentMarketplace, ServiceType
print("\n✓ Marketplace module imported")
pkg/hanzo-mcp/tests/test_hanzo_mcp_integration.py
+8 -8
View File
@@ -1,16 +1,16 @@
"""Integration tests for hanzo-mcp with Claude CLI and basic operations."""
import os
import json
import tempfile
import os
import subprocess
import tempfile
from pathlib import Path
import pytest
from hanzo_mcp.server import create_server
from mcp.client.session import ClientSession
from mcp.server.fastmcp import FastMCP
from hanzo_mcp.server import create_server
class TestHanzoMCPIntegration:
"""Test hanzo-mcp server functionality and Claude CLI integration."""
@@ -85,7 +85,7 @@ class TestHanzoMCPIntegration:
assert test_content in str(read_result)
# Test edit operation
edit_result = await mcp_server.mcp.call_tool(
await mcp_server.mcp.call_tool(
"edit",
arguments={
"file_path": str(test_file),
@@ -277,7 +277,7 @@ def main():
}
# Write the notebook using the write tool
write_result = await mcp_server.mcp.call_tool(
await mcp_server.mcp.call_tool(
"write",
arguments={
"file_path": str(notebook_path),
@@ -288,7 +288,7 @@ def main():
assert notebook_path.exists()
# Now edit to add a markdown cell using the unified jupyter tool
edit_result = await mcp_server.mcp.call_tool(
await mcp_server.mcp.call_tool(
"jupyter",
arguments={
"action": "edit",
@@ -363,7 +363,7 @@ class TestHanzoMCPStdioServer:
assert "grep" in tool_names
# Test a simple tool call
result = await session.call_tool(
await session.call_tool(
"write",
arguments={
"file_path": str(tmp_path / "test.txt"),
pkg/hanzo-mcp/tests/test_hanzo_mcp_local.py
+6 -6
View File
@@ -1,11 +1,11 @@
#!/usr/bin/env python3
"""Test hanzo-mcp locally to ensure it works."""
import sys
import json
import asyncio
import tempfile
import json
import subprocess
import sys
import tempfile
from pathlib import Path
# Add the package to path for local testing
@@ -67,7 +67,7 @@ async def test_file_operations():
test_content = "Hello from hanzo-mcp test!"
# Test write operation using call_tool
write_result = await server.mcp.call_tool(
await server.mcp.call_tool(
"write", arguments={"file_path": str(test_file), "content": test_content}
)
@@ -89,7 +89,7 @@ async def test_file_operations():
assert test_content in str(read_result)
# Test edit operation
edit_result = await server.mcp.call_tool(
await server.mcp.call_tool(
"edit",
arguments={
"file_path": str(test_file),
@@ -232,7 +232,7 @@ async def test_notebook_operations():
assert notebook_content["cells"][1]["cell_type"] == "markdown"
# Test editing notebook to add a new cell
edit_result = await server.mcp.call_tool(
await server.mcp.call_tool(
"jupyter",
arguments={
"action": "edit",
pkg/hanzo-mcp/tests/test_hanzo_mcp_simple.py
+6 -5
View File
@@ -1,11 +1,11 @@
#!/usr/bin/env python3
"""Simple test to verify hanzo-mcp works locally."""
import os
import sys
import asyncio
import tempfile
import os
import subprocess
import sys
import tempfile
from pathlib import Path
# Add the package to path for local testing
@@ -41,8 +41,8 @@ def test_cli_version():
async def test_stdio_server():
"""Test the stdio server with a simple interaction."""
import json
import asyncio
import json
with tempfile.TemporaryDirectory() as tmpdir:
# Start the server process
@@ -116,8 +116,9 @@ async def test_stdio_server():
def test_import_tools():
"""Test that we can import tools directly."""
try:
from hanzo_mcp.tools import register_all_tools
from mcp.server.fastmcp import FastMCP
from hanzo_mcp.tools import register_all_tools
from hanzo_mcp.tools.common.permissions import PermissionManager
# Create a test server
pkg/hanzo-mcp/tests/test_hanzo_network_integration.py
+9 -5
View File
@@ -1,21 +1,21 @@
"""Integration tests for hanzo-network multi-agent orchestration."""
import tempfile
from typing import Any, Dict
from pathlib import Path
from typing import Any, Dict
import pytest
# Try to import hanzo-network components
try:
from hanzo_network import (
Tool,
Agent,
State,
Network,
NetworkConfig,
LocalComputeNode,
LocalComputeOrchestrator,
Network,
NetworkConfig,
State,
Tool,
create_agent_network,
)
from hanzo_network.tools import MemoryTool, create_memory_tool
@@ -28,7 +28,11 @@ except ImportError:
try:
from hanzo_agents import (
Agent as HanzoAgent,
)
from hanzo_agents import (
Network as HanzoNetwork,
)
from hanzo_agents import (
create_agent,
create_network,
)
pkg/hanzo-mcp/tests/test_integration/test_agent_clarification_critic.py
+3 -2
View File
@@ -5,9 +5,10 @@ from unittest.mock import Mock, patch
import pytest
from hanzo_tools.agent.agent_tool import AgentTool
from hanzo_tools.agent.critic_tool import ReviewType
from hanzo_mcp.tools.common.permissions import PermissionManager
from hanzo_tools.agent.clarification_protocol import ClarificationType
from hanzo_tools.agent.critic_tool import ReviewType
from hanzo_mcp.tools.common.permissions import PermissionManager
@pytest.fixture
pkg/hanzo-mcp/tests/test_integration/test_all_agent_tools.py
+2 -1
View File
@@ -5,6 +5,7 @@ from unittest.mock import Mock, patch
import pytest
from hanzo_tools.agent.agent_tool import AgentTool
from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -333,9 +334,9 @@ async def test_review_vs_critic_difference():
@pytest.mark.asyncio
async def test_tool_limits():
"""Test that tools respect their usage limits."""
from hanzo_tools.agent.clarification_protocol import ClarificationHandler
from hanzo_tools.agent.critic_tool import CriticProtocol
from hanzo_tools.agent.review_tool import ReviewProtocol
from hanzo_tools.agent.clarification_protocol import ClarificationHandler
# Test clarification limit (1)
clarifier = ClarificationHandler()
pkg/hanzo-mcp/tests/test_jupyter/test_unified_jupyter.py
+2 -1
View File
@@ -3,8 +3,9 @@
import json
import pytest
from mcp.server.fastmcp import Context as MCPContext
from hanzo_tools.jupyter.jupyter import JupyterTool
from mcp.server.fastmcp import Context as MCPContext
from hanzo_mcp.tools.common.permissions import PermissionManager
pkg/hanzo-mcp/tests/test_litellm_warnings.py
+1 -1
View File
@@ -1,8 +1,8 @@
#!/usr/bin/env python
"""Test that litellm deprecation warnings are properly suppressed."""
import sys
import subprocess
import sys
def test_no_pydantic_warnings():
pkg/hanzo-mcp/tests/test_lsp_tool.py
+1 -1
View File
@@ -3,7 +3,7 @@
import asyncio
import tempfile
from pathlib import Path
from unittest.mock import AsyncMock, MagicMock, patch
from unittest.mock import AsyncMock, patch
import pytest
from hanzo_tools.lsp import create_lsp_tool
pkg/hanzo-mcp/tests/test_manual.py
+6 -5
View File
@@ -1,17 +1,18 @@
#!/usr/bin/env python3
"""Manual test script for Hanzo AI functionality."""
import os
import asyncio
import os
import tempfile
from pathlib import Path
from hanzo_mcp.server import HanzoMCPServer
from hanzo_tools.filesystem.diff import create_diff_tool
from hanzo_tools.filesystem.read import ReadTool
# from hanzo_mcp.tools.common.palette import PaletteRegistry # Module doesn't exist
from hanzo_tools.shell.bash_tool import bash_tool
from hanzo_mcp.server import HanzoMCPServer
from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -87,14 +88,14 @@ async def test_basic_functionality():
# Test 5: Server Creation
print("5️⃣ Testing Server Creation...")
try:
server = HanzoMCPServer(
HanzoMCPServer(
name="test-server",
allowed_paths=[temp_dir],
use_palette=True,
force_palette="python",
)
print("✅ Server created successfully")
print(f"✅ Python palette applied")
print("✅ Python palette applied")
except Exception as e:
print(f"❌ Server creation failed: {e}")
print()
@@ -141,7 +142,7 @@ def test_dev_mode():
try:
from hanzo_mcp.dev_server import DevServer
dev_server = DevServer(
DevServer(
name="test-dev",
allowed_paths=["/tmp"],
)
pkg/hanzo-mcp/tests/test_memory_base.py
+2 -3
View File
@@ -1,11 +1,11 @@
"""Base test class for memory-related tests to reduce redundancy."""
from datetime import datetime
from unittest.mock import Mock, AsyncMock, patch
from unittest.mock import Mock, patch
import pytest
from fastmcp import FastMCP
from conftest import ToolTestHelper, create_mock_ctx
from fastmcp import FastMCP
# Import guard for optional hanzo_memory dependency
try:
@@ -26,7 +26,6 @@ if HANZO_MEMORY_AVAILABLE:
DeleteMemoriesTool,
RecallMemoriesTool,
UpdateMemoriesTool,
register_memory_tools,
)
pkg/hanzo-mcp/tests/test_memory_basic.py
+1 -1
View File
@@ -2,7 +2,7 @@
import asyncio
from datetime import datetime
from unittest.mock import Mock, AsyncMock, patch
from unittest.mock import AsyncMock, Mock, patch
def test_basic_memory():
pkg/hanzo-mcp/tests/test_memory_consolidated.py
+2 -3
View File
@@ -1,6 +1,5 @@
"""Consolidated memory tests using parametrization to reduce redundancy."""
from unittest.mock import Mock, AsyncMock, patch
import pytest
@@ -17,7 +16,6 @@ pytestmark = pytest.mark.skipif(not HANZO_MEMORY_AVAILABLE, reason="hanzo_memory
# Only import these if hanzo_memory is available
if HANZO_MEMORY_AVAILABLE:
from test_memory_base import MemoryTestBase
from hanzo_tools.memory import (
CreateMemoriesTool,
DeleteMemoriesTool,
@@ -25,6 +23,7 @@ if HANZO_MEMORY_AVAILABLE:
UpdateMemoriesTool,
register_memory_tools,
)
from test_memory_base import MemoryTestBase
else:
# Dummy class to prevent NameError when module is skipped
class MemoryTestBase:
@@ -222,7 +221,7 @@ class TestMemoryToolsConsolidated(MemoryTestBase):
"importance": importance,
}
result = await tool_helper.run_tool(tool, params, mock_ctx)
await tool_helper.run_tool(tool, params, mock_ctx)
# Verify service was called with correct parameters
mock_memory_service.create_memory.assert_called_once()

Some files were not shown because too many files have changed in this diff Show More