MCP Agent Tracker

#!/usr/bin/env python3 """ Test script to verify interaction filtering logic """ import sqlite3 import os def test_filtering_logic(): """Test the filtering logic used in the UI""" print("🔍 Testing Interaction Filtering Logic") print("=" * 60) db_path = "./data/agent_tracker.db" if not os.path.exists(db_path): print(f"❌ Database not found: {db_path}") return try: conn = sqlite3.connect(db_path) cursor = conn.cursor() # Test 1: Total count cursor.execute("SELECT COUNT(*) FROM agent_interactions") total_count = cursor.fetchone()[0] print(f"📊 Total interactions: {total_count}") # Test 2: Filter by conversation types conversation_types = ['conversation_turn', 'client_request', 'agent_response', 'user_prompt'] placeholders = ','.join(['?'] * len(conversation_types)) query = f"SELECT COUNT(*) FROM agent_interactions WHERE interaction_type IN ({placeholders})" cursor.execute(query, conversation_types) conversation_count = cursor.fetchone()[0] print(f"💬 Conversation types ({', '.join(conversation_types)}): {conversation_count}") # Test 3: Filter by system types system_types = ['health_check', 'monitoring_started', 'module_import', 'system_startup', 'system_shutdown'] placeholders = ','.join(['?'] * len(system_types)) query = f"SELECT COUNT(*) FROM agent_interactions WHERE interaction_type IN ({placeholders})" cursor.execute(query, system_types) system_count = cursor.fetchone()[0] print(f"⚙️ System types ({', '.join(system_types)}): {system_count}") # Test 4: Filter by tool types tool_types = ['tool_call', 'tool_response', 'tool_error'] placeholders = ','.join(['?'] * len(tool_types)) query = f"SELECT COUNT(*) FROM agent_interactions WHERE interaction_type IN ({placeholders})" cursor.execute(query, tool_types) tool_count = cursor.fetchone()[0] print(f"🔧 Tool types ({', '.join(tool_types)}): {tool_count}") # Test 5: Verify totals calculated_total = conversation_count + system_count + tool_count print(f"🧮 Calculated total: {calculated_total}") print(f"✅ Match: {'Yes' if calculated_total == total_count else 'No'}") # Test 6: Show all interaction types print(f"\n📋 All interaction types:") cursor.execute("SELECT interaction_type, COUNT(*) FROM agent_interactions GROUP BY interaction_type ORDER BY COUNT(*) DESC") for interaction_type, count in cursor.fetchall(): print(f" • {interaction_type}: {count}") conn.close() except Exception as e: print(f"❌ Error: {e}") def main(): """Main test function""" print("🚀 Interaction Filtering Test") print("=" * 60) test_filtering_logic() print(f"\n🎯 Expected Results:") print(f"• Total: 617 interactions") print(f"• Conversations: 107 interactions") print(f"• System: 501 interactions") print(f"• Tools: 16 interactions") print(f"• Total: 107 + 501 + 16 = 624 (should match 617)") if __name__ == "__main__": main()