import asyncio
from fastmcp import Client
async def test_unsloth_server():
"""Test the Unsloth MCP server."""
client = Client("unsloth_mcp_server.py")
async with client:
print("=== Testing Unsloth MCP Server ===\n")
# Test 1: Get quickstart guide
print("1. Getting Unsloth quickstart guide...")
result = await client.call_tool("get_unsloth_quickstart", {})
print(f"Result: {result}\n")
print("-" * 50)
# Test 2: Search for specific topic
print("2. Searching for 'fine-tuning'...")
result = await client.call_tool("search_unsloth_docs", {"query": "fine-tuning"})
print(f"Result: {result}\n")
print("-" * 50)
# Test 3: Get model information
print("3. Getting supported models...")
result = await client.call_tool("get_unsloth_models", {})
print(f"Result: {result}\n")
print("-" * 50)
# Test 4: Get installation info
print("4. Getting installation instructions...")
result = await client.call_tool("get_unsloth_installation", {})
print(f"Result: {result}\n")
if __name__ == "__main__":
asyncio.run(test_unsloth_server())
