100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Test script to demonstrate the AI analysis functionality.
|
||
|
|
This script shows how to use the new async thread analysis function.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
# Add the src directory to the path
|
||
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
|
||
|
|
|
||
|
|
from database import (
|
||
|
|
SessionLocal,
|
||
|
|
Thread,
|
||
|
|
analyze_and_update_threads,
|
||
|
|
create_db_tables,
|
||
|
|
get_threads_needing_analysis,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""Main function to test the AI analysis."""
|
||
|
|
|
||
|
|
# Create database tables if they don't exist
|
||
|
|
create_db_tables()
|
||
|
|
|
||
|
|
# Example account email - replace with your actual account
|
||
|
|
account_email = "your-email@example.com"
|
||
|
|
|
||
|
|
# Get a database session
|
||
|
|
db = SessionLocal()
|
||
|
|
try:
|
||
|
|
# Check how many threads need analysis
|
||
|
|
threads_needing_analysis = get_threads_needing_analysis(db, account_email)
|
||
|
|
print(
|
||
|
|
f"Found {len(threads_needing_analysis)} threads needing analysis for {account_email}"
|
||
|
|
)
|
||
|
|
|
||
|
|
if not threads_needing_analysis:
|
||
|
|
print(
|
||
|
|
"No threads need analysis. Make sure you have ingested some emails first."
|
||
|
|
)
|
||
|
|
|
||
|
|
# Show all threads for this account
|
||
|
|
all_threads = (
|
||
|
|
db.query(Thread)
|
||
|
|
.filter(Thread.account_email == account_email.lower())
|
||
|
|
.all()
|
||
|
|
)
|
||
|
|
print(f"Total threads for {account_email}: {len(all_threads)}")
|
||
|
|
|
||
|
|
if all_threads:
|
||
|
|
print("Sample threads:")
|
||
|
|
for thread in all_threads[:3]:
|
||
|
|
print(
|
||
|
|
f" Thread {thread.id}: {thread.subject} (analyzed: {thread.last_analyzed_at is not None})"
|
||
|
|
)
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
print(f"Starting AI analysis for {len(threads_needing_analysis)} threads...")
|
||
|
|
|
||
|
|
# Run the analysis with max 3 concurrent tasks
|
||
|
|
analyze_and_update_threads(
|
||
|
|
account_email=account_email, max_concurrent=3, only_unanalyzed=True
|
||
|
|
)
|
||
|
|
|
||
|
|
print("Analysis complete!")
|
||
|
|
|
||
|
|
# Show results
|
||
|
|
analyzed_threads = (
|
||
|
|
db.query(Thread)
|
||
|
|
.filter(
|
||
|
|
Thread.account_email == account_email.lower(),
|
||
|
|
Thread.last_analyzed_at.isnot(None),
|
||
|
|
)
|
||
|
|
.all()
|
||
|
|
)
|
||
|
|
|
||
|
|
print(f"\nAnalyzed {len(analyzed_threads)} threads:")
|
||
|
|
for thread in analyzed_threads[:5]: # Show first 5
|
||
|
|
print(f" Thread {thread.id}: {thread.subject[:50]}...")
|
||
|
|
print(f" Actionable: {thread.actionable}")
|
||
|
|
print(f" Confidence: {thread.ai_confidence:.2f}")
|
||
|
|
print(f" Summary: {thread.ai_summary[:100]}...")
|
||
|
|
print()
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error: {e}")
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
traceback.print_exc()
|
||
|
|
finally:
|
||
|
|
db.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|