test_imports.py

"""
Simple test file to verify backend imports.
"""
import streamlit as st
import os
from backend import Chatbot, load_llm_model, load_embedding_model, load_or_create_index
import sys

st.title("Import Test")
st.write("Backend module path:", sys.modules['backend'].__file__)
st.write("Available in backend module:")
st.write("- load_llm_model:", 'load_llm_model' in dir(sys.modules['backend']))
st.write("- load_embedding_model:", 'load_embedding_model' in dir(sys.modules['backend']))
st.write("- load_or_create_index:", 'load_or_create_index' in dir(sys.modules['backend']))
st.write("- Chatbot:", 'Chatbot' in dir(sys.modules['backend']))

st.write("Backend module directory listing:")
st.code(str(dir(sys.modules['backend'])))

if st.button("Try to create chatbot"):
    try:
        # Get API key
        api_key = os.getenv("ANTHROPIC_API_KEY")
        if not api_key:
            st.error("No API key found in environment variables")
            
        # Try to initialize objects
        llm = load_llm_model(api_key)
        st.success("Successfully created LLM model")
        
        embed_model = load_embedding_model()
        st.success("Successfully created embedding model")
        
        st.info("Tests completed successfully!")
    except Exception as e:
        st.error(f"Error: {str(e)}")
        st.exception(e)