fb5c04ea60
Major improvements to thinking capabilities and API integration: - Remove all output token limits for future-proof responses - Add 5-level thinking mode system: minimal, low, medium, high, max - Migrate from google-generativeai to google-genai library - Implement native thinkingBudget support for Gemini 2.5 Pro - Set medium thinking as default for all tools, max for think_deeper 🧠 Thinking Modes: - minimal (128 tokens) - simple tasks - low (2048 tokens) - basic reasoning - medium (8192 tokens) - default for most tools - high (16384 tokens) - complex analysis - max (32768 tokens) - default for think_deeper 🔧 Technical Changes: - Complete migration to google-genai>=1.19.0 - Remove google-generativeai dependency - Add ThinkingConfig with thinking_budget parameter - Update all tools to support thinking_mode parameter - Comprehensive test suite with 37 passing unit tests - CI-friendly testing (no API key required for unit tests) - Live integration tests for API verification 🧪 Testing & CI: - Add GitHub Actions workflow with multi-Python support - Unit tests use mocks, no API key required - Live integration tests optional with API key - Contributing guide with development setup - All tests pass without external dependencies 🐛 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
"""
|
|
Tests for the main server functionality
|
|
"""
|
|
|
|
import json
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
|
|
from server import handle_call_tool, handle_list_tools
|
|
|
|
|
|
class TestServerTools:
|
|
"""Test server tool handling"""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_list_tools(self):
|
|
"""Test listing all available tools"""
|
|
tools = await handle_list_tools()
|
|
tool_names = [tool.name for tool in tools]
|
|
|
|
# Check all core tools are present
|
|
assert "think_deeper" in tool_names
|
|
assert "review_code" in tool_names
|
|
assert "debug_issue" in tool_names
|
|
assert "analyze" in tool_names
|
|
assert "chat" in tool_names
|
|
assert "list_models" in tool_names
|
|
assert "get_version" in tool_names
|
|
|
|
# Should have exactly 7 tools
|
|
assert len(tools) == 7
|
|
|
|
# Check descriptions are verbose
|
|
for tool in tools:
|
|
assert (
|
|
len(tool.description) > 50
|
|
) # All should have detailed descriptions
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_call_tool_unknown(self):
|
|
"""Test calling an unknown tool"""
|
|
result = await handle_call_tool("unknown_tool", {})
|
|
assert len(result) == 1
|
|
assert "Unknown tool: unknown_tool" in result[0].text
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_chat(self):
|
|
"""Test chat functionality"""
|
|
# Set test environment
|
|
import os
|
|
os.environ["PYTEST_CURRENT_TEST"] = "test"
|
|
|
|
# Create a mock for the model
|
|
with patch("tools.base.BaseTool.create_model") as mock_create:
|
|
mock_model = Mock()
|
|
mock_model.generate_content.return_value = Mock(
|
|
candidates=[Mock(content=Mock(parts=[Mock(text="Chat response")]))]
|
|
)
|
|
mock_create.return_value = mock_model
|
|
|
|
result = await handle_call_tool("chat", {"prompt": "Hello Gemini"})
|
|
|
|
assert len(result) == 1
|
|
assert result[0].text == "Chat response"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_list_models(self):
|
|
"""Test listing models"""
|
|
result = await handle_call_tool("list_models", {})
|
|
assert len(result) == 1
|
|
|
|
# Check if we got models or an error
|
|
text = result[0].text
|
|
if "Error" in text:
|
|
# API key not set in test environment
|
|
assert "GEMINI_API_KEY" in text
|
|
else:
|
|
# Should have models
|
|
models = json.loads(text)
|
|
assert len(models) >= 1
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_get_version(self):
|
|
"""Test getting version info"""
|
|
result = await handle_call_tool("get_version", {})
|
|
assert len(result) == 1
|
|
|
|
response = result[0].text
|
|
assert "Gemini MCP Server v" in response # Version agnostic check
|
|
assert "Available Tools:" in response
|
|
assert "think_deeper" in response
|