|
|
#!/usr/bin/env python3 |
|
|
""" |
|
|
Quick test script for MCP server tools. |
|
|
Run: python test_tools.py |
|
|
""" |
|
|
|
|
|
import asyncio |
|
|
import json |
|
|
from pprint import pprint |
|
|
|
|
|
from mcp_server import tools |
|
|
|
|
|
|
|
|
async def main(): |
|
|
print("🔍 Testing MCP server tools...\n") |
|
|
|
|
|
# 1. Ping (simulated, since it's just "ok") |
|
|
print("1️⃣ ping → ok") |
|
|
print(" ✅", "ok\n") |
|
|
|
|
|
# 2. Search documents with a trivial AQL (read-only) |
|
|
try: |
|
|
aql = "FOR d IN riksdagen LIMIT 1 RETURN d" |
|
|
print("2️⃣ search_documents...") |
|
|
result = await asyncio.to_thread(tools.search_documents, aql) |
|
|
print(" ✅ rows:", result["row_count"]) |
|
|
except Exception as e: |
|
|
print(" ❌ search_documents failed:", e) |
|
|
print() |
|
|
|
|
|
# 3. Run AQL query directly |
|
|
try: |
|
|
aql = "FOR d IN riksdagen LIMIT 1 RETURN d" |
|
|
print("3️⃣ run_aql_query...") |
|
|
rows = await asyncio.to_thread(tools.run_aql_query, aql) |
|
|
print(" ✅ got", len(rows), "rows") |
|
|
except Exception as e: |
|
|
print(" ❌ run_aql_query failed:", e) |
|
|
print() |
|
|
|
|
|
# 4. Vector search (Chroma) |
|
|
try: |
|
|
print("4️⃣ vector_search_talks...") |
|
|
hits = await asyncio.to_thread(tools.vector_search, "klimatpolitik", 3) |
|
|
print(f" ✅ got {len(hits)} hits") |
|
|
if hits: |
|
|
print(" →", hits[0]) |
|
|
except Exception as e: |
|
|
print(" ❌ vector_search_talks failed:", e) |
|
|
print() |
|
|
|
|
|
# 5. Fetch documents |
|
|
try: |
|
|
print("5️⃣ fetch_documents (demo id)...") |
|
|
# Adjust a known _id if needed; using a dummy for now |
|
|
docs = await asyncio.to_thread(tools.fetch_documents, ["riksdagen/1"]) |
|
|
print(" ✅ got", len(docs), "docs") |
|
|
except Exception as e: |
|
|
print(" ❌ fetch_documents failed:", e) |
|
|
print() |
|
|
|
|
|
# 6. Arango search |
|
|
try: |
|
|
print("6️⃣ arango_search...") |
|
|
result = await asyncio.to_thread(tools.arango_search, "budget", 3) |
|
|
print(" ✅ got", len(result.get("results", [])), "hits") |
|
|
except Exception as e: |
|
|
print(" ❌ arango_search failed:", e) |
|
|
print() |
|
|
|
|
|
print("🏁 Done.\n") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
asyncio.run(main())
|
|
|
|
