Open Census MCP Server

#!/usr/bin/env python3 """ Quick import test to verify components are loadable """ import sys import os from pathlib import Path # Add paths current_dir = Path(__file__).parent kb_path = current_dir / "knowledge-base" src_path = current_dir / "src" sys.path.insert(0, str(kb_path)) sys.path.insert(0, str(src_path)) print("🔧 Testing component imports...") try: from llm_statistical_advisor import create_llm_statistical_advisor, LLMStatisticalAdvisor print("✅ LLM Statistical Advisor imports") except Exception as e: print(f"❌ LLM Statistical Advisor import failed: {e}") try: from kb_search import create_search_engine, CensusSearchEngine print("✅ Search engine imports") except Exception as e: print(f"❌ Search engine import failed: {e}") try: from geographic_parsing import GeographicContext, create_geographic_parser print("✅ Geographic parsing imports") except Exception as e: print(f"❌ Geographic parsing import failed: {e}") try: from variable_search import create_variables_search, VariablesSearch print("✅ Variable search imports") except Exception as e: print(f"❌ Variable search import failed: {e}") # Check environment openai_key = os.getenv('OPENAI_API_KEY') print(f"🔑 OpenAI API Key: {'Present' if openai_key else 'Missing'}") # Check critical directories kb_dirs = [ kb_path / "variables-db", kb_path / "table-catalog", kb_path / "methodology-db", kb_path / "geo-db" / "geography.db" ] for dir_path in kb_dirs: exists = dir_path.exists() print(f"📁 {dir_path.name}: {'✅' if exists else '❌'}") print("\n🎯 Component availability summary:") print("- All critical Python modules should import successfully") print("- OpenAI API key should be present for embeddings") print("- Knowledge base directories should exist") print("\nIf all show ✅, the wiring should work!")