feat: Implement async AI analysis for email threads

- Added `get_latest_email_date()` function in `database.py` to retrieve the most recent email date for a given account and folder. - Enhanced `fetch_folder_emails()` in `zoho_client.py` to intelligently determine the start date for fetching emails based on the latest email date in the database. - Introduced `analyze_and_update_threads_async()` for asynchronous analysis of email threads, allowing concurrent processing. - Created a synchronous wrapper `analyze_and_update_threads()` for easier integration. - Updated `fetch_emails()` to support database session and account email parameters. - Added comprehensive documentation in `AI_ANALYSIS_GUIDE.md` detailing the new AI analysis functionality. - Implemented tests for the new features, including `test_fetch_with_db.py`, `test_ai_analysis.py`, and `test_single_analysis.py`. - Added error handling and logging improvements throughout the codebase.
2025-08-11 23:20:20 +01:00
parent d553d6f31e
commit 75a0a3fde7
14 changed files with 1358 additions and 476 deletions
@@ -0,0 +1,94 @@
+#!/usr/bin/env python3
+"""
+Simple test to debug the AI 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 ai import analyze_thread
+from database import SessionLocal, Thread, get_thread_messages
+
+
+def test_single_analysis():
+    """Test analyzing a single thread."""
+
+    db = SessionLocal()
+    try:
+        # Get a thread to test with
+        thread = (
+            db.query(Thread)
+            .filter(Thread.account_email == "test-user@company.com")
+            .first()
+        )
+
+        if not thread:
+            print("No threads found for test-user@company.com")
+            return
+
+        print(f"Testing thread {thread.id}: {thread.subject}")
+        print(
+            f"Before analysis: actionable={thread.actionable}, confidence={thread.ai_confidence}"
+        )
+
+        # Get messages
+        messages = get_thread_messages(db, thread.id)
+        print(f"Found {len(messages)} messages")
+
+        # Convert to dict format
+        message_dicts = []
+        for msg in messages:
+            message_dicts.append(
+                {
+                    "date_sent": msg.date_sent,
+                    "is_incoming": msg.is_incoming,
+                    "subject": msg.subject,
+                    "from_email": msg.from_email,
+                    "to_email": msg.to_email,
+                    "body": msg.body,
+                }
+            )
+
+        # Run analysis
+        print("Running AI analysis...")
+        analysis_result = analyze_thread(thread.subject, message_dicts)
+        print(f"Analysis result: {analysis_result}")
+
+        # Update thread
+        thread.actionable = analysis_result.get("actionable", False)
+        thread.ai_summary = analysis_result.get("summary", "")
+        thread.ai_confidence = float(analysis_result.get("confidence", 0.0))
+        from datetime import datetime
+
+        thread.last_analyzed_at = datetime.now()
+
+        print(
+            f"Updated thread: actionable={thread.actionable}, confidence={thread.ai_confidence}"
+        )
+        print(f"Summary: {thread.ai_summary}")
+
+        # Commit changes
+        db.commit()
+        print("Changes committed")
+
+        # Verify the changes
+        db.refresh(thread)
+        print(
+            f"After refresh: actionable={thread.actionable}, confidence={thread.ai_confidence}"
+        )
+
+    except Exception as e:
+        print(f"Error: {e}")
+        import traceback
+
+        traceback.print_exc()
+        db.rollback()
+    finally:
+        db.close()
+
+
+if __name__ == "__main__":
+    test_single_analysis()