Byted Bytehouse Mcp

# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

#!/usr/bin/env python3
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
ByteHouse MCP Server 使用示例
演示如何调用MCP Server的tools
"""

# /// script
# dependencies = [
#   "mcp>=1.0.0",
# ]
# ///

import asyncio
import sys
import os

# 添加当前目录到路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from mcp_client import ByteHouseMCPClient


async def example_list_tools():
    """示例1: 列出所有可用的tools"""
    print("=" * 80)
    print("示例1: 列出所有可用的tools")
    print("=" * 80)
    
    async with ByteHouseMCPClient() as client:
        await client.connect()
        
        tools = await client.list_tools()
        print(f"\n找到 {len(tools)} 个tools:\n")
        for i, tool in enumerate(tools, 1):
            print(f"{i}. {tool['name']}")
            print(f"   描述: {tool['description']}")
            print(f"   参数: {tool['inputSchema']}")
            print()


async def example_query_databases():
    """示例2: 查询所有数据库"""
    print("=" * 80)
    print("示例2: 查询所有数据库")
    print("=" * 80)
    
    async with ByteHouseMCPClient() as client:
        await client.connect()
        
        # 注意：这里需要根据实际的tool名称来调用
        # 先列出tools看看有什么可用的
        tools = await client.list_tools()
        print("\n可用的tools:")
        for tool in tools:
            print(f"  - {tool['name']}")
        
        # 这里需要根据实际的tool名称和参数来调用
        # 由于不同的MCP Server提供的tools可能不同，这里先列出可用的tools


async def main():
    """主函数"""
    print("\n" + "=" * 80)
    print("ByteHouse MCP Server 使用示例")
    print("=" * 80)
    print()
    print("⚠️  请确保已设置以下环境变量:")
    print("  - BYTEHOUSE_HOST")
    print("  - BYTEHOUSE_PORT")
    print("  - BYTEHOUSE_USER")
    print("  - BYTEHOUSE_PASSWORD")
    print()
    
    try:
        # 示例1: 列出所有tools
        await example_list_tools()
        
        # 示例2: 查询数据库
        # await example_query_databases()
        
    except Exception as e:
        print(f"\n❌ 错误: {e}")
        import traceback
        traceback.print_exc()
        return 1
    
    return 0


if __name__ == "__main__":
    sys.exit(asyncio.run(main()))

#!/usr/bin/env python3
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
MCP Client - 用于与MCP Server通信的客户端
"""

# /// script
# dependencies = [
#   "mcp>=1.0.0",
# ]
# ///

import asyncio
import json
import sys
import os
from typing import Any, Dict, List, Optional
from contextlib import AsyncExitStack

from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client


class ByteHouseMCPClient:
    """ByteHouse MCP客户端"""
    
    def __init__(self):
        self.session: Optional[ClientSession] = None
        self._exit_stack = AsyncExitStack()
        self.server_params = None
        
    async def __aenter__(self):
        await self._exit_stack.__aenter__()
        return self
        
    async def __aexit__(self, exc_type, exc_val, exc_tb):
        await self._exit_stack.__aexit__(exc_type, exc_val, exc_tb)
        
    async def connect(self):
        """连接到ByteHouse MCP Server"""
        
        # 从环境变量获取配置
        env = os.environ.copy()
        
        # MCP Server参数
        self.server_params = StdioServerParameters(
            command='/root/.local/bin/uvx',
            args=[
                '--from',
                'git+https://github.com/volcengine/mcp-server@main#subdirectory=server/mcp_server_bytehouse',
                'mcp_bytehouse',
                '-t',
                'stdio'
            ],
            env=env
        )
        
        print("🔌 正在连接到ByteHouse MCP Server...")
        
        # 启动stdio客户端
        stdio_transport = await self._exit_stack.enter_async_context(
            stdio_client(self.server_params)
        )
        
        # 创建会话
        self.session = await self._exit_stack.enter_async_context(
            ClientSession(stdio_transport[0], stdio_transport[1])
        )
        
        # 初始化会话
        await self.session.initialize()
        
        print("✅ 连接成功！")
        return self
        
    async def list_tools(self) -> List[Dict]:
        """列出所有可用的tools"""
        if not self.session:
            raise RuntimeError("Not connected")
            
        result = await self.session.list_tools()
        tools = []
        for tool in result.tools:
            tools.append({
                'name': tool.name,
                'description': tool.description,
                'inputSchema': tool.inputSchema
            })
        return tools
        
    async def call_tool(self, tool_name: str, arguments: Dict[str, Any]) -> Any:
        """调用tool"""
        if not self.session:
            raise RuntimeError("Not connected")
            
        result = await self.session.call_tool(tool_name, arguments)
        
        # 处理结果
        outputs = []
        for content in result.content:
            if content.type == 'text':
                outputs.append(content.text)
            elif content.type == 'image':
                outputs.append(f"[Image: {content.data[:50]}...]")
                
        return outputs


# 便捷函数
async def run_mcp_tool(tool_name: str, arguments: Dict[str, Any] = None):
    """运行单个MCP tool"""
    if arguments is None:
        arguments = {}
        
    async with ByteHouseMCPClient() as client:
        await client.connect()
        
        # 先列出tools
        print("\n📋 可用的Tools:")
        tools = await client.list_tools()
        for tool in tools:
            print(f"  - {tool['name']}: {tool['description']}")
            
        print(f"\n🚀 调用Tool: {tool_name}")
        print(f"   参数: {json.dumps(arguments, ensure_ascii=False)}")
        
        # 调用tool
        result = await client.call_tool(tool_name, arguments)
        
        print("\n📊 结果:")
        for output in result:
            print(output)
            
        return result


if __name__ == "__main__":
    # 测试：列出所有tools
    async def main():
        async with ByteHouseMCPClient() as client:
            await client.connect()
            
            print("\n📋 可用的Tools:")
            tools = await client.list_tools()
            for tool in tools:
                print(f"\n  {tool['name']}")
                print(f"    描述: {tool['description']}")
                print(f"    参数: {json.dumps(tool['inputSchema'], ensure_ascii=False, indent=6)}")
    
    asyncio.run(main())

#!/usr/bin/env python3
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
使用MCP Server查询ByteHouse Top 10大表
"""

# /// script
# dependencies = [
#   "mcp>=1.0.0",
# ]
# ///

import asyncio
import os
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client


async def query_top10_tables():
    """使用MCP查询Top 10大表"""
    print("=" * 80)
    print("使用MCP Server查询ByteHouse Top 10大表")
    print("=" * 80)
    print()
    print("⚠️  请确保已设置以下环境变量:")
    print("  - BYTEHOUSE_HOST")
    print("  - BYTEHOUSE_PORT")
    print("  - BYTEHOUSE_USER")
    print("  - BYTEHOUSE_PASSWORD")
    print()
    
    # 从环境变量获取配置
    env = os.environ.copy()
    
    # MCP Server参数
    server_params = StdioServerParameters(
        command='/root/.local/bin/uvx',
        args=[
            '--from',
            'git+https://github.com/volcengine/mcp-server@main#subdirectory=server/mcp_server_bytehouse',
            'mcp_bytehouse',
            '-t',
            'stdio'
        ],
        env=env
    )
    
    print("\n🔌 正在连接MCP Server...")
    
    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write) as session:
            await session.initialize()
            print("✅ 连接成功！")
            
            # 1. 先列出所有数据库
            print("\n📊 步骤1: 列出所有数据库")
            print("-" * 80)
            result = await session.call_tool("list_databases", {})
            for content in result.content:
                if content.type == 'text':
                    print(content.text)
            
            # 2. 查询Top 10大表
            print("\n🏆 步骤2: 查询Top 10大表")
            print("-" * 80)
            sql = """
                SELECT 
                    database,
                    table,
                    sum(bytes) as total_bytes,
                    sum(rows) as total_rows
                FROM system.parts 
                WHERE active = 1
                GROUP BY database, table
                ORDER BY total_bytes DESC
                LIMIT 10
            """
            
            result = await session.call_tool("run_select_query", {"query": sql})
            
            print("\n📊 查询结果:")
            print("-" * 80)
            for content in result.content:
                if content.type == 'text':
                    print(content.text)
            
            print("\n" + "=" * 80)
            print("✅ 查询完成！")
            print("=" * 80)


if __name__ == "__main__":
    asyncio.run(query_top10_tables())

#!/bin/bash
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# 重启ByteHouse MCP Server

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

echo "=" * 80
echo "重启ByteHouse MCP Server"
echo "=" * 80

"$SCRIPT_DIR/stop_mcp_service.sh"
echo
sleep 2
"$SCRIPT_DIR/start_mcp_service.sh"

#!/bin/bash
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# 启动ByteHouse MCP Server的脚本

# 设置环境变量（请自行配置）
# export BYTEHOUSE_HOST="<ByteHouse-host>"
# export BYTEHOUSE_PORT="<ByteHouse-port>"
# export BYTEHOUSE_USER="<ByteHouse-user>"
# export BYTEHOUSE_PASSWORD="<ByteHouse-password>"
# export BYTEHOUSE_SECURE="true"
# export BYTEHOUSE_VERIFY="true"
# export BYTEHOUSE_CONNECT_TIMEOUT="30"
# export BYTEHOUSE_SEND_RECEIVE_TIMEOUT="30"

echo "=" * 80
echo "启动ByteHouse MCP Server"
echo "=" * 80
echo "⚠️  请确保已在脚本中配置环境变量"
echo "=" * 80
echo

# 检查环境变量是否已配置
if [ -z "$BYTEHOUSE_HOST" ] || [ -z "$BYTEHOUSE_PASSWORD" ]; then
    echo "❌ 错误: 请先配置ByteHouse连接环境变量"
    echo "   编辑此脚本，取消注释并配置环境变量部分"
    exit 1
fi

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
LOG_DIR="$SCRIPT_DIR/logs"
PID_FILE="$SCRIPT_DIR/mcp_server.pid"

# 创建日志目录
mkdir -p "$LOG_DIR"

# 日志文件
LOG_FILE="$LOG_DIR/mcp_server_$(date +%Y%m%d_%H%M%S).log"

echo "=" * 80
echo "ByteHouse MCP Server - 常驻服务"
echo "=" * 80
echo "脚本目录: $SCRIPT_DIR"
echo "日志目录: $LOG_DIR"
echo "日志文件: $LOG_FILE"
echo "PID文件: $PID_FILE"
echo "=" * 80
echo

# 检查是否已经在运行
if [ -f "$PID_FILE" ]; then
    PID=$(cat "$PID_FILE" 2>/dev/null)
    if [ -n "$PID" ] && kill -0 "$PID" 2>/dev/null; then
        echo "⚠️  MCP Server已经在运行 (PID: $PID)"
        echo "   如需重启，请先运行: $SCRIPT_DIR/stop_mcp_service.sh"
        exit 1
    else
        echo "⚠️  PID文件存在但进程不存在，清理旧文件"
        rm -f "$PID_FILE"
    fi
fi

echo "🚀 启动ByteHouse MCP Server..."
echo

# 启动MCP Server（后台运行）
cd "$SCRIPT_DIR"
/root/.local/bin/uvx --from "git+https://github.com/volcengine/mcp-server@main#subdirectory=server/mcp_server_bytehouse" mcp_bytehouse -t stdio > "$LOG_FILE" 2>&1 &
MCP_PID=$!

# 保存PID
echo $MCP_PID > "$PID_FILE"

# 等待一下检查进程
sleep 2

if kill -0 "$MCP_PID" 2>/dev/null; then
    echo "✅ ByteHouse MCP Server启动成功！"
    echo "   PID: $MCP_PID"
    echo "   日志: $LOG_FILE"
    echo ""
    echo "📋 管理命令:"
    echo "   查看状态: $SCRIPT_DIR/status_mcp_service.sh"
    echo "   查看日志: tail -f $LOG_FILE"
    echo "   停止服务: $SCRIPT_DIR/stop_mcp_service.sh"
else
    echo "❌ MCP Server启动失败"
    echo "   查看日志: $LOG_FILE"
    rm -f "$PID_FILE"
    exit 1
fi

#!/bin/bash
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# 查看ByteHouse MCP Server状态

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PID_FILE="$SCRIPT_DIR/mcp_server.pid"
LOG_DIR="$SCRIPT_DIR/logs"

echo "=" * 80
echo "ByteHouse MCP Server 状态"
echo "=" * 80

if [ ! -f "$PID_FILE" ]; then
    echo "❌ MCP Server未运行 (PID文件不存在)"
    echo ""
    echo "启动命令: $SCRIPT_DIR/start_mcp_service.sh"
    exit 1
fi

PID=$(cat "$PID_FILE" 2>/dev/null)
if [ -z "$PID" ]; then
    echo "❌ MCP Server未运行 (PID文件为空)"
    rm -f "$PID_FILE"
    exit 1
fi

if ! kill -0 "$PID" 2>/dev/null; then
    echo "❌ MCP Server未运行 (进程不存在)"
    rm -f "$PID_FILE"
    exit 1
fi

echo "✅ MCP Server正在运行"
echo "   PID: $PID"
echo ""

# 查看进程信息
echo "📊 进程信息:"
ps -p "$PID" -o pid,ppid,cmd,%cpu,%mem,etime
echo ""

# 查看最新日志
if [ -d "$LOG_DIR" ]; then
    LATEST_LOG=$(ls -t "$LOG_DIR"/mcp_server_*.log 2>/dev/null | head -1)
    if [ -n "$LATEST_LOG" ]; then
        echo "📝 最新日志文件: $LATEST_LOG"
        echo ""
        echo "📋 最近20行日志:"
        echo "-" * 80
        tail -20 "$LATEST_LOG"
        echo "-" * 80
        echo ""
        echo "查看完整日志: tail -f $LATEST_LOG"
    fi
fi

echo ""
echo "📋 管理命令:"
echo "   停止服务: $SCRIPT_DIR/stop_mcp_service.sh"
echo "   重启服务: $SCRIPT_DIR/restart_mcp_service.sh"

#!/bin/bash
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# 停止ByteHouse MCP Server

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PID_FILE="$SCRIPT_DIR/mcp_server.pid"

echo "=" * 80
echo "停止ByteHouse MCP Server"
echo "=" * 80

if [ ! -f "$PID_FILE" ]; then
    echo "⚠️  PID文件不存在，MCP Server可能未运行"
    exit 0
fi

PID=$(cat "$PID_FILE" 2>/dev/null)
if [ -z "$PID" ]; then
    echo "⚠️  PID文件为空"
    rm -f "$PID_FILE"
    exit 0
fi

echo "正在停止MCP Server (PID: $PID)..."

# 尝试优雅停止
kill "$PID" 2>/dev/null

# 等待进程结束
for i in {1..10}; do
    if ! kill -0 "$PID" 2>/dev/null; then
        echo "✅ MCP Server已停止"
        rm -f "$PID_FILE"
        exit 0
    fi
    sleep 1
done

# 强制停止
echo "⚠️  进程未响应，强制停止..."
kill -9 "$PID" 2>/dev/null
sleep 1

if ! kill -0 "$PID" 2>/dev/null; then
    echo "✅ MCP Server已强制停止"
else
    echo "❌ 无法停止进程 (PID: $PID)"
    exit 1
fi

rm -f "$PID_FILE"

#!/usr/bin/env python3
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
测试调用 list_tables tool
"""

# /// script
# dependencies = [
#   "mcp>=1.0.0",
# ]
# ///

import asyncio
import os
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client


async def test_list_tables():
    """测试list_tables tool"""
    print("=" * 80)
    print("测试调用 list_tables tool")
    print("=" * 80)
    
    # 从环境变量获取配置
    env = os.environ.copy()
    
    # MCP Server参数
    server_params = StdioServerParameters(
        command='/root/.local/bin/uvx',
        args=[
            '--from',
            'git+https://github.com/volcengine/mcp-server@main#subdirectory=server/mcp_server_bytehouse',
            'mcp_bytehouse',
            '-t',
            'stdio'
        ],
        env=env
    )
    
    print("\n🔌 正在连接MCP Server...")
    
    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write) as session:
            await session.initialize()
            print("✅ 连接成功！")
            
            # 1. 先列出所有数据库
            print("\n📊 步骤1: 列出所有数据库")
            print("-" * 80)
            result = await session.call_tool("list_databases", {})
            databases = []
            for content in result.content:
                if content.type == 'text':
                    print(content.text)
                    databases = [db.strip() for db in content.text.split('\n') if db.strip()]
            
            # 2. 列出每个数据库的表
            print(f"\n📊 步骤2: 列出每个数据库的表")
            print("-" * 80)
            
            for db in databases:
                print(f"\n📁 数据库: {db}")
                print("-" * 40)
                
                try:
                    result = await session.call_tool("list_tables", {"database": db})
                    for content in result.content:
                        if content.type == 'text':
                            tables = [t.strip() for t in content.text.split('\n') if t.strip()]
                            print(f"   找到 {len(tables)} 张表:")
                            for table in tables:
                                print(f"     - {table}")
                except Exception as e:
                    print(f"   ❌ 查询失败: {e}")
            
            print("\n" + "=" * 80)
            print("✅ 测试完成！")
            print("=" * 80)


if __name__ == "__main__":
    asyncio.run(test_list_tables())

#!/usr/bin/env python3
# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
测试ByteHouse MCP Server
直接通过stdio与MCP Server通信
"""

# /// script
# dependencies = [
#   "mcp>=1.0.0",
# ]
# ///

import asyncio
import os
import sys
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client


async def test_mcp_server():
    """测试MCP Server"""
    print("=" * 80)
    print("测试ByteHouse MCP Server")
    print("=" * 80)
    print()
    print("⚠️  请确保已设置以下环境变量:")
    print("  - BYTEHOUSE_HOST")
    print("  - BYTEHOUSE_PORT")
    print("  - BYTEHOUSE_USER")
    print("  - BYTEHOUSE_PASSWORD")
    print()
    
    # 从环境变量获取配置
    env = os.environ.copy()
    
    # MCP Server参数
    server_params = StdioServerParameters(
        command='/root/.local/bin/uvx',
        args=[
            '--from',
            'git+https://github.com/volcengine/mcp-server@main#subdirectory=server/mcp_server_bytehouse',
            'mcp_bytehouse',
            '-t',
            'stdio'
        ],
        env=env
    )
    
    print("\n📦 正在安装和启动MCP Server...")
    print("   这可能需要几分钟时间...")
    print()
    
    try:
        # 启动stdio客户端
        async with stdio_client(server_params) as (read, write):
            # 创建会话
            async with ClientSession(read, write) as session:
                # 初始化会话
                print("🔌 正在初始化MCP会话...")
                await session.initialize()
                print("✅ 会话初始化成功！")
                print()
                
                # 列出可用的tools
                print("📋 正在获取可用的tools...")
                result = await session.list_tools()
                
                print(f"\n✅ 找到 {len(result.tools)} 个tools:\n")
                for i, tool in enumerate(result.tools, 1):
                    print(f"{i}. {tool.name}")
                    print(f"   描述: {tool.description}")
                    print(f"   参数: {tool.inputSchema}")
                    print()
                
                # 如果有tools，尝试调用第一个
                if result.tools:
                    first_tool = result.tools[0]
                    print(f"🚀 尝试调用第一个tool: {first_tool.name}")
                    print()
                    
                    try:
                        # 尝试调用tool（使用空参数）
                        call_result = await session.call_tool(first_tool.name, {})
                        
                        print("📊 调用结果:")
                        print("-" * 80)
                        for content in call_result.content:
                            if content.type == 'text':
                                print(content.text)
                            elif content.type == 'image':
                                print(f"[Image data: {len(content.data)} bytes]")
                        print("-" * 80)
                        
                    except Exception as e:
                        print(f"⚠️  调用tool失败 (可能需要特定参数): {e}")
                        print("   但这不影响MCP Server的正常工作")
                
                print("\n" + "=" * 80)
                print("🎉 MCP Server测试成功！")
                print("=" * 80)
                print()
                print("💡 下一步:")
                print("  1. 启动常驻服务: ./start_mcp_service.sh")
                print("  2. 查看状态: ./status_mcp_service.sh")
                print("  3. 停止服务: ./stop_mcp_service.sh")
                print()
                
    except Exception as e:
        print(f"\n❌ 测试失败: {e}")
        import traceback
        traceback.print_exc()
        return 1
    
    return 0


if __name__ == "__main__":
    sys.exit(asyncio.run(test_mcp_server()))