#!/usr/bin/env python3
from app import app
from models import db, User
from assistant_routes import search_all_documents
import traceback

with app.app_context():
    # Get admin user
    user = User.query.filter_by(role='admin').first()

    if user:
        print(f"Testing search for user: {user.get_full_name()} (ID: {user.id})")
        try:
            result = search_all_documents(user, "vulnerabilities")
            print("\n" + "="*80)
            print(result)
            print("="*80)
        except Exception as e:
            print(f"\n❌ Error: {e}")
            print("\nFull traceback:")
            traceback.print_exc()
    else:
        print("No admin user found")