Ryan Malloy
eb524b8c1d
BREAKING CHANGES: - Package renamed from mcp-arduino-server to mcp-arduino - Command changed to 'mcp-arduino' (was 'mcp-arduino-server') - Repository moved to git.supported.systems/MCP/mcp-arduino NEW FEATURES: ✨ Smart client capability detection and dual-mode sampling support ✨ Intelligent WireViz templates with component-specific circuits (LED, motor, sensor, button, display) ✨ Client debug tools for MCP capability inspection ✨ Enhanced error handling with progressive enhancement patterns IMPROVEMENTS: 🧹 Major repository cleanup - removed 14+ experimental files and tests 📝 Consolidated and reorganized documentation 🐛 Fixed import issues and applied comprehensive linting with ruff 📦 Updated author information to Ryan Malloy (ryan@supported.systems) 🔧 Fixed package version references in startup code TECHNICAL DETAILS: - Added dual-mode WireViz: AI generation for sampling clients, smart templates for others - Implemented client capability detection via MCP handshake inspection - Created progressive enhancement pattern for universal MCP client compatibility - Organized test files into proper structure (tests/examples/) - Applied comprehensive code formatting and lint fixes The server now provides excellent functionality for ALL MCP clients regardless of their sampling capabilities, while preserving advanced features for clients that support them. Version: 2025.09.27.1
102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script for WireViz sampling functionality
|
|
|
|
This tests the dual-mode approach:
|
|
1. Clients with sampling support get AI-generated diagrams
|
|
2. Clients without sampling get intelligent templates
|
|
"""
|
|
|
|
import asyncio
|
|
import logging
|
|
|
|
# Setup logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
async def test_wireviz_generation():
|
|
"""Test WireViz generation with different descriptions"""
|
|
|
|
# Import the necessary components
|
|
from src.mcp_arduino_server.components.wireviz import WireViz
|
|
from src.mcp_arduino_server.config import ArduinoServerConfig
|
|
|
|
# Create config and component
|
|
config = ArduinoServerConfig()
|
|
wireviz = WireViz(config)
|
|
|
|
# Test descriptions
|
|
test_cases = [
|
|
{
|
|
"description": "Arduino with LED on pin 9",
|
|
"expected_template": "led"
|
|
},
|
|
{
|
|
"description": "Connect a servo motor to Arduino pin D9",
|
|
"expected_template": "motor"
|
|
},
|
|
{
|
|
"description": "Temperature sensor connected to analog pin A0",
|
|
"expected_template": "sensor"
|
|
},
|
|
{
|
|
"description": "Push button with pull-up resistor on D2",
|
|
"expected_template": "button"
|
|
},
|
|
{
|
|
"description": "I2C OLED display connected to Arduino",
|
|
"expected_template": "display" # Contains 'oled' which triggers display template
|
|
},
|
|
{
|
|
"description": "Generic circuit with resistors and capacitors",
|
|
"expected_template": "generic"
|
|
}
|
|
]
|
|
|
|
print("\n" + "="*60)
|
|
print("TESTING WIREVIZ TEMPLATE GENERATION")
|
|
print("="*60 + "\n")
|
|
|
|
for test in test_cases:
|
|
print(f"Test: {test['description']}")
|
|
print(f"Expected template type: {test['expected_template']}")
|
|
|
|
# Generate template (simulating no sampling available)
|
|
yaml_content = wireviz._generate_template_yaml(test['description'])
|
|
|
|
# Check if the right template was selected
|
|
if test['expected_template'] == 'led' and 'LED' in yaml_content:
|
|
print("✅ LED template correctly selected")
|
|
elif test['expected_template'] == 'motor' and 'Servo' in yaml_content:
|
|
print("✅ Motor/Servo template correctly selected")
|
|
elif test['expected_template'] == 'sensor' and 'Sensor' in yaml_content:
|
|
print("✅ Sensor template correctly selected")
|
|
elif test['expected_template'] == 'button' and 'Button' in yaml_content:
|
|
print("✅ Button template correctly selected")
|
|
elif test['expected_template'] == 'display' and 'I2C Display' in yaml_content:
|
|
print("✅ Display template correctly selected")
|
|
elif test['expected_template'] == 'generic' and 'Component1' in yaml_content:
|
|
print("✅ Generic template correctly selected")
|
|
else:
|
|
print("❌ Template selection might be incorrect")
|
|
|
|
print("-" * 40)
|
|
|
|
print("\n" + "="*60)
|
|
print("TEMPLATE GENERATION TEST COMPLETE")
|
|
print("="*60)
|
|
|
|
print("\nKey Insights:")
|
|
print("• Templates are selected based on keywords in the description")
|
|
print("• Each template provides a good starting point for common circuits")
|
|
print("• Users can customize the YAML for their specific needs")
|
|
print("• When sampling IS available, AI will generate custom YAML")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_wireviz_generation())
|