add beautifulsoup4 dependency and update output formatting for research mcp

mcp_servers/research/provider.py

@@ -460,31 +460,48 @@ class MessageGeneratorTool(BaseTool):
     def _generate_template_message(
         self, occasion: str, recipient: str, tone: str, details: str
     ) -> str:
-        """Generate message using templates."""
+        """Generate message using templates - always creates INVITATIONS for parties."""
+        occasion_lower = occasion.lower()
+        
+        # For party invitations, always use invitation template
+        if "invitation" in occasion_lower or "party" in occasion_lower:
+            # Extract the party theme from occasion
+            theme = occasion.replace("invitation", "").replace("party", "").replace("for", "").strip()
+            if not theme:
+                theme = "our special celebration"
+            
+            return (
+                f"🎉 You're Invited to {theme}! 🎉\n\n"
+                f"Dear {recipient},\n\n"
+                f"Get ready for an amazing celebration! We're hosting {theme} and it wouldn't be "
+                f"the same without you.\n\n"
+                f"📅 {details if details else 'Date & Time: TBD'}\n\n"
+                f"Please RSVP by [date] so we can plan accordingly.\n\n"
+                f"Can't wait to celebrate with you!\n"
+                f"🥳 See you there!"
+            )
+        
         templates = {
             "birthday": [
-                f"🎂 Happy Birthday, {recipient}! 🎉\n\nWishing you a day filled with joy, laughter, and all your favorite things. May this year bring you endless happiness and amazing adventures!\n\nCheers to you! 🥳",
-                f"🎈 Happy Birthday, {recipient}!\n\nAnother year older, another year wiser, and definitely another year more awesome! Hope your special day is as wonderful as you are.\n\nEnjoy your day! 🎁",
-            ],
-            "party invitation": [
-                f"🎉 You're Invited!\n\nHey {recipient}!\n\nWe're throwing a celebration and it wouldn't be the same without you! Join us for an unforgettable time.\n\n{details if details else 'Details to follow!'}\n\nHope to see you there! 🥳",
+                f"🎂 Happy Birthday Celebration! 🎉\n\nDear {recipient},\n\nYou're invited to a special birthday celebration!\n\n{details if details else 'Details coming soon!'}\n\nJoin us for fun, food, and festivities!\n\nRSVP: [contact]\n🎈 Hope to see you there!",
             ],
             "thank you": [
                 f"Dear {recipient},\n\nThank you so much for being part of our celebration! Your presence made the day even more special.\n\nWith gratitude,\n❤️",
             ],
         }
         
-        occasion_lower = occasion.lower()
         for key, msgs in templates.items():
             if key in occasion_lower:
                 return msgs[0]
         
-        # Default template
+        # Default invitation template
         return (
+            f"🎉 You're Invited!\n\n"
             f"Dear {recipient},\n\n"
-            f"You're invited to celebrate with us! "
-            f"{'Details: ' + details if details else 'More details coming soon.'}\n\n"
-            f"Looking forward to seeing you!\n🎉"
+            f"We're hosting a special celebration and would love for you to join us!\n\n"
+            f"{'📅 ' + details if details else '📅 Details to follow!'}\n\n"
+            f"Please let us know if you can make it.\n\n"
+            f"Looking forward to celebrating with you!\n🥳"
         )
     
     def get_schema(self) -> Dict[str, Any]:
@@ -746,16 +763,9 @@ Maximum 3 tool calls per response."""
                 "params": {"query": f"how to plan {state.idea} party ideas activities guide"}
             })
             
-            # Search for specific supplies based on highlights
-            amazon_terms = [idea_themes]
-            if idea_highlights:
-                # Extract shopping-related terms from highlights
-                for highlight in idea_highlights[:2]:
-                    highlight_lower = highlight.lower()
-                    if any(word in highlight_lower for word in ["station", "kit", "supplies", "decor", "activity"]):
-                        amazon_terms.append(highlight[:30])
-            
-            amazon_query = " ".join(amazon_terms[:3]) + " party supplies"
+            # Search for specific supplies based on idea title and highlights
+            # Use the full idea name for more specific results
+            amazon_query = f"{state.idea} party supplies decorations"
             calls.append({
                 "tool": "amazon_search",
                 "params": {"query": amazon_query}
@@ -763,20 +773,20 @@ Maximum 3 tool calls per response."""
             
             # Generate invitation message with more context
             invite_details = []
-            if state.context.get("location"):
-                invite_details.append(f"Location: {state.context['location']}")
             if state.context.get("event_date"):
                 invite_details.append(f"Date: {state.context['event_date']}")
-            if state.occasion:
-                invite_details.append(f"Occasion: {state.occasion}")
+            if state.context.get("location"):
+                invite_details.append(f"Location: {state.context['location']}")
+            if state.honoree and state.honoree.get("guest_count"):
+                invite_details.append(f"Expected guests: {state.honoree['guest_count']}")
             
             calls.append({
                 "tool": "generate_message",
                 "params": {
-                    "occasion": f"{state.occasion or 'party'} invitation for {state.idea}",
-                    "recipient": "Guest",
+                    "occasion": f"party invitation for {state.idea}",
+                    "recipient": "Friend",
                     "tone": "fun and exciting",
-                    "details": ". ".join(invite_details) if invite_details else "at my place"
+                    "details": "\n".join(invite_details) if invite_details else "Details to be announced"
                 }
             })
         
@@ -1136,12 +1146,8 @@ Be specific, practical, and enthusiastic!"""
             # Extract overview from synthesized text
             overview = final_text[:800] if len(final_text) > 800 else final_text
         
-        # Build steps - prefer idea's next_steps if available
-        steps = []
-        if state.idea_details and state.idea_details.get("next_steps"):
-            steps = state.idea_details["next_steps"]
-        else:
-            steps = self._extract_steps_from_text(final_text)
+        # Build detailed, actionable steps
+        steps = self._build_actionable_steps(state, final_text)
         
         # Build references (combined sources and links)
         references = [
@@ -1218,6 +1224,56 @@ Be specific, practical, and enthusiastic!"""
         
         return "Completed"
     
+    def _build_actionable_steps(self, state: AgentState, synthesis_text: str) -> List[str]:
+        """Build detailed, actionable steps with timing and specific guidance."""
+        idea_name = state.idea
+        
+        # Get idea-specific steps if available
+        idea_steps = []
+        if state.idea_details and state.idea_details.get("next_steps"):
+            idea_steps = state.idea_details["next_steps"]
+        
+        # Extract any steps from synthesis text
+        extracted_steps = self._extract_steps_from_text(synthesis_text)
+        
+        # Build comprehensive action plan - mix generic timeline with specific tasks
+        planning_steps = []
+        
+        # Phase 1: Planning (2 weeks before)
+        planning_steps.append(f"📋 **2 weeks before**: Create your guest list and budget for {idea_name}")
+        if idea_steps and len(idea_steps) > 0:
+            planning_steps.append(f"✉️ **2 weeks before**: {idea_steps[0]}")
+        else:
+            planning_steps.append(f"✉️ **2 weeks before**: Design and send themed invitations")
+        
+        # Phase 2: Preparation (1 week before)
+        planning_steps.append(f"🛒 **1 week before**: Order supplies from the shopping list below")
+        if idea_steps and len(idea_steps) > 1:
+            planning_steps.append(f"🍽️ **1 week before**: {idea_steps[1]}")
+        else:
+            planning_steps.append(f"🍽️ **1 week before**: Plan themed menu and drinks")
+        
+        # Phase 3: Final prep (3-4 days before)
+        if idea_steps and len(idea_steps) > 2:
+            planning_steps.append(f"🎮 **3-4 days before**: {idea_steps[2]}")
+        else:
+            planning_steps.append(f"🎮 **3-4 days before**: Prepare games, activities, and entertainment")
+        
+        # Phase 4: Setup and event
+        planning_steps.append(f"🏠 **1 day before**: Set up decorations and prepare the venue")
+        planning_steps.append(f"✅ **Day of event**: Final walkthrough, welcome guests, and enjoy!")
+        
+        # Add tips from web research
+        if extracted_steps:
+            for step in extracted_steps[:2]:
+                if step and len(step) > 15:
+                    # Only add if it's genuinely new info
+                    step_lower = step.lower()
+                    if not any(kw in step_lower for kw in ["guest list", "invitation", "decoration", "menu"]):
+                        planning_steps.append(f"💡 **Pro tip**: {step[:80]}")
+        
+        return planning_steps[:9]  # Limit to 9 steps
+    
     def _extract_steps_from_text(self, text: str) -> List[str]:
         """Extract actionable steps from synthesized text."""
         steps = []
@@ -1232,17 +1288,7 @@ Be specific, practical, and enthusiastic!"""
                 if clean_line and len(clean_line) > 10:
                     steps.append(clean_line[:100])
         
-        # If no steps found, create default ones
-        if not steps:
-            steps = [
-                "Review the plan and customize for your needs",
-                "Purchase necessary supplies and decorations",
-                "Send invitations to guests",
-                "Prepare the venue and activities",
-                "Enjoy the celebration!"
-            ]
-        
-        return steps[:8]  # Limit to 8 steps
+        return steps[:5]
     
     def _get_web_sources(self, state: AgentState) -> List[Dict[str, str]]:
         """Extract web sources from tool results."""

requirements.txt

@@ -6,3 +6,4 @@ httpx
 ddgs
 langgraph>=1.0.0
 langchain-core>=1.1.0
+beautifulsoup4