test_app.py

#!/usr/bin/env python3
"""
Test script to debug app.py initialization issues
"""

import os

os.environ["TOKENIZERS_PARALLELISM"] = "false"

try:
    from warbler_cda import (
        RetrievalAPI,
        SemanticAnchorGraph,
        EmbeddingProviderFactory,
        STAT7RAGBridge,
        RetrievalQuery,
        RetrievalMode,
    )
    from warbler_cda.pack_loader import PackLoader

    WARBLER_AVAILABLE = True
    print("[OK] Imports successful")

except ImportError:
    WARBLER_AVAILABLE = False
    print("Warning: Warbler CDA not installed. Using mock mode.")

api = None

if WARBLER_AVAILABLE:
    try:
        print("[INFO] Initializing embedding provider...")
        embedding_provider = EmbeddingProviderFactory.get_default_provider()
        print(f"[OK] Using embedding provider: {type(embedding_provider).__name__}")

        print("[INFO] Initializing semantic anchors...")
        semantic_anchors = SemanticAnchorGraph(embedding_provider=embedding_provider)
        print("[OK] Semantic anchors initialized")

        print("[INFO] Initializing STAT7 bridge...")
        stat7_bridge = STAT7RAGBridge()
        print("[OK] STAT7 bridge initialized")

        print("[INFO] Initializing retrieval API...")
        api = RetrievalAPI(
            semantic_anchors=semantic_anchors,
            embedding_provider=embedding_provider,
            stat7_bridge=stat7_bridge,
            config={"enable_stat7_hybrid": True},
        )
        print("[OK] API initialized")

        print(f"[OK] Total documents in context store: {api.get_context_store_size()}")

    except Exception as e:
        print(f"[ERROR] Failed to initialize Warbler CDA: {e}")
        import traceback

        traceback.print_exc()
        api = None

print("[DONE] Test completed")