kamkol/AB_AI_RAG_Agent

1#!/usr/bin/env python32 3"""4Script to check installed dependencies for troubleshooting5"""6 7import torch8import sys9import pkg_resources10import importlib11import platform12 13def check_imports():14    """Check if critical imports work"""15    success = True16    critical_imports = [17        "transformers",18        "torch", 19        "streamlit",20        "langchain_core",21        "langchain_openai",22        "langgraph",23        "numpy",24        "scipy"25    ]26 27    print("\n=== Import Checks ===")28    for module_name in critical_imports:29        try:30            module = importlib.import_module(module_name)31            version = getattr(module, "__version__", "Unknown")32            print(f"✓ {module_name}: {version}")33        except ImportError as e:34            success = False35            print(f"✗ {module_name}: FAILED - {str(e)}")36    return success37 38def check_transformers_model():39    """Check if transformers can load your fine-tuned model"""40    print("\n=== Transformers Model Check ===")41    try:42        from transformers import AutoModel, AutoTokenizer43        model_name = "kamkol/ab_testing_finetuned_arctic_ft-36dfff22-0696-40d2-b3bf-268fe2ff2aec"44        print(f"Testing loading model: {model_name}")45        tokenizer = AutoTokenizer.from_pretrained(model_name)46        model = AutoModel.from_pretrained(model_name)47        inputs = tokenizer("Test sentence", return_tensors="pt")48        with torch.no_grad():49            outputs = model(**inputs)50        print("✓ Successfully loaded model and ran forward pass")51        return True52    except Exception as e:53        print(f"✗ Transformers model test failed: {str(e)}")54        return False55 56def check_langchain_components():57    """Check critical LangChain components"""58    print("\n=== LangChain Check ===")59    try:60        from langchain_core.messages import HumanMessage61        from langchain_core.prompts import ChatPromptTemplate62        from langgraph.graph import StateGraph63        print("✓ LangChain core components imported successfully")64        return True65    except Exception as e:66        print(f"✗ LangChain check failed: {str(e)}")67        return False68 69def print_system_info():70    """Print system information"""71    print("\n=== System Information ===")72    print(f"Python version: {sys.version}")73    print(f"Platform: {platform.platform()}")74    print(f"Implementation: {platform.python_implementation()}")75 76def main():77    """Main function to run checks"""78    print("=== Dependency Check ===")79    print_system_info()80 81    # Get all installed packages82    installed_packages = {pkg.key: pkg.version for pkg in pkg_resources.working_set}83 84    # Check for specific dependencies85    key_packages = [86        "transformers",87        "torch",88        "streamlit",89        "langchain-core",90        "langchain-openai",91        "langgraph",92        "numpy",93        "scipy"94    ]95 96    print("\n=== Package Versions ===")97    for pkg in key_packages:98        version = installed_packages.get(pkg, "Not installed")99        print(f"{pkg}: {version}")100 101    # Test imports102    imports_ok = check_imports()103    104    # Test transformers model105    tf_ok = check_transformers_model()106    107    # Test LangChain108    lc_ok = check_langchain_components()109 110    # Final result111    if imports_ok and tf_ok and lc_ok:112        print("\n✓ All critical checks passed")113    else:114        print("\n✗ Some checks failed, see details above")115 116if __name__ == "__main__":117    main()118