mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-14 10:38:29 +00:00
f7cca439cc
- Add EmailState model for Flow state management - Create EmailProcessingFlow class with event-based automation - Update tools and crews for Flow integration - Add comprehensive Flow tests - Implement error handling and state tracking - Add mock implementations for testing This implementation uses CrewAI Flow features to create an event-based email processing system that can analyze emails, research senders, and generate appropriate responses using specialized AI crews. Co-Authored-By: Joe Moura <joao@crewai.com>
104 lines
4.7 KiB
Python
104 lines
4.7 KiB
Python
"""Test script for complete email processing workflow"""
|
|
from email_analysis_crew import EmailAnalysisCrew
|
|
from response_crew import ResponseCrew
|
|
from email_tool import EmailTool
|
|
from mock_email_data import MockEmailThread
|
|
import json
|
|
|
|
def test_email_analysis(email_tool, analysis_crew, thread_id):
|
|
"""Test comprehensive email analysis including similar threads and research"""
|
|
print("\nAnalyzing email thread...")
|
|
|
|
# Get thread context
|
|
thread = email_tool.get_email_thread(thread_id)
|
|
print(f"\nThread subject: {thread.subject}")
|
|
|
|
# Find similar threads
|
|
similar = email_tool.find_similar_threads(thread.subject)
|
|
print(f"\nFound {len(similar)} similar threads")
|
|
|
|
# Get sender history
|
|
sender = thread.messages[0].from_email
|
|
sender_info = email_tool.get_sender_history(sender)
|
|
print(f"\nSender: {sender_info['name']} from {sender_info['company']}")
|
|
print(f"Previous interactions: {sender_info['interaction_frequency']}")
|
|
|
|
# Analyze with crew
|
|
analysis_result = analysis_crew.analyze_email(thread_id)
|
|
print(f"\nAnalysis Results:")
|
|
print(f"Response needed: {analysis_result.get('response_needed', False)}")
|
|
print(f"Priority: {analysis_result.get('priority', 'error')}")
|
|
print(f"Decision factors:")
|
|
context = analysis_result.get('analysis', {}).get('context', {})
|
|
print(f"- Thread type: {context.get('thread_type', 'unknown')}")
|
|
print(f"- Similar threads found: {analysis_result.get('similar_threads_found', 0)}")
|
|
print(f"- Interaction frequency: {sender_info.get('interaction_frequency', 'unknown')}")
|
|
print(f"- Urgency indicators: {context.get('urgency_indicators', False)}")
|
|
print(f"- Conversation stage: {context.get('conversation_stage', 'unknown')}")
|
|
|
|
return analysis_result
|
|
|
|
def test_complete_workflow():
|
|
"""Test the complete email processing workflow"""
|
|
try:
|
|
print("\nTesting Complete Email Processing Workflow")
|
|
print("=========================================")
|
|
|
|
# Initialize tools and crews
|
|
email_tool = EmailTool()
|
|
analysis_crew = EmailAnalysisCrew()
|
|
response_crew = ResponseCrew()
|
|
|
|
# Test 1: Process email requiring response (weekly interaction)
|
|
print("\nTest 1: Processing email requiring response")
|
|
print("------------------------------------------")
|
|
thread_id = "thread_1" # Meeting follow-up thread from frequent contact
|
|
|
|
analysis_result = test_email_analysis(email_tool, analysis_crew, thread_id)
|
|
|
|
if analysis_result.get('response_needed', False):
|
|
print("\nGenerating response...")
|
|
response_result = response_crew.draft_response(thread_id, analysis_result)
|
|
print("\nGenerated Response:")
|
|
print(json.dumps(response_result.get('response', {}), indent=2))
|
|
|
|
# Verify response matches context
|
|
print("\nResponse Analysis:")
|
|
print(f"Tone matches relationship: {response_result['response']['review_notes']['context_awareness']['relationship_acknowledged']}")
|
|
print(f"Priority reflected: {response_result['response']['review_notes']['context_awareness']['priority_reflected']}")
|
|
print(f"Background used: {response_result['response']['review_notes']['context_awareness']['background_used']}")
|
|
else:
|
|
print("\nNo response required.")
|
|
|
|
# Test 2: Process email not requiring response (first-time sender)
|
|
print("\nTest 2: Processing email not requiring response")
|
|
print("----------------------------------------------")
|
|
thread_id = "thread_3" # First-time contact
|
|
|
|
analysis_result = test_email_analysis(email_tool, analysis_crew, thread_id)
|
|
|
|
if analysis_result.get('response_needed', False):
|
|
print("\nGenerating response...")
|
|
response_result = response_crew.draft_response(thread_id, analysis_result)
|
|
print("\nGenerated Response:")
|
|
print(json.dumps(response_result.get('response', {}), indent=2))
|
|
|
|
# Verify response matches context
|
|
print("\nResponse Analysis:")
|
|
context = analysis_result.get('analysis', {}).get('context', {})
|
|
print(f"Thread type: {context.get('thread_type', 'unknown')}")
|
|
print(f"Conversation stage: {context.get('conversation_stage', 'unknown')}")
|
|
print(f"Response priority: {analysis_result.get('priority', 'unknown')}")
|
|
else:
|
|
print("\nNo response required - First time sender with no urgent context")
|
|
|
|
print("\nWorkflow test completed successfully!")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"Analysis error: {str(e)}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
test_complete_workflow()
|