init

negot8/backend/features/roommate.py

@@ -0,0 +1,123 @@
+from features.base_feature import BaseFeature
+from tools.tavily_search import TavilySearchTool
+
+_tavily = TavilySearchTool()
+
+
+class RoommateFeature(BaseFeature):
+
+    async def get_context(self, preferences_a: dict, preferences_b: dict, user_a_id: int = None, user_b_id: int = None) -> str:
+        """
+        Fetch real product/plan options via Tavily (e.g., actual WiFi plans,
+        furniture prices) so agents propose real options.
+        """
+        raw_a = preferences_a.get("raw_details", {})
+        raw_b = preferences_b.get("raw_details", {})
+
+        decision_type = (
+            raw_a.get("decision_type") or raw_b.get("decision_type")
+            or "shared living decision"
+        )
+        city = (
+            raw_a.get("city") or raw_b.get("city")
+            or raw_a.get("location") or raw_b.get("location")
+            or "India"
+        )
+        budget_a = raw_a.get("budget") or raw_a.get("max_budget") or ""
+        budget_b = raw_b.get("budget") or raw_b.get("max_budget") or ""
+
+        # Build a Tavily query based on decision type
+        if "wifi" in str(decision_type).lower() or "internet" in str(decision_type).lower():
+            query = f"best WiFi broadband plans {city} 2026 price speed"
+        elif "furniture" in str(decision_type).lower():
+            query = f"furniture prices India 2026 online shopping"
+        elif "chore" in str(decision_type).lower() or "cleaning" in str(decision_type).lower():
+            query = f"chore schedule roommates fair division strategies"
+        else:
+            query = f"{decision_type} options India 2026"
+
+        search_text = ""
+        try:
+            result = await _tavily.execute(query)
+            answer = result.get("answer", "")
+            results = result.get("results", [])[:4]
+            parts = []
+            if answer:
+                parts.append(f"Summary: {answer[:300]}")
+            for r in results:
+                title = r.get("title", "")
+                content = r.get("content", "")[:150]
+                if title:
+                    parts.append(f"  • {title}: {content}")
+            search_text = "\n".join(parts)
+        except Exception as e:
+            search_text = f"Search unavailable. Use your knowledge of {decision_type} options in {city}."
+
+        lines = [
+            "ROOMMATE DECISION DOMAIN RULES:",
+            "• Only propose options (plans, products) that appear in the real data below.",
+            "• Budget ceiling = lower of both parties' stated budgets.",
+            "• Unequal cost splits need usage-based justification.",
+            "• Both parties must stay within their stated budget constraints.",
+            "• If no option satisfies both budgets, propose cheapest viable option + fair split.",
+        ]
+        if decision_type:
+            lines.append(f"\nDecision type: {decision_type}")
+        if city:
+            lines.append(f"Location: {city}")
+        if budget_a:
+            lines.append(f"Person A max budget: ₹{budget_a}/month")
+        if budget_b:
+            lines.append(f"Person B max budget: ₹{budget_b}/month")
+
+        if search_text:
+            lines.append(f"\nREAL OPTIONS from web search (only propose from this list):\n{search_text}")
+
+        return "\n".join(lines)
+
+    def format_resolution(
+        self, resolution: dict, preferences_a: dict, preferences_b: dict
+    ) -> str:
+        status = resolution.get("status", "resolved")
+        final = resolution.get("final_proposal", {})
+        details = final.get("details", {})
+        rounds = resolution.get("rounds_taken", "?")
+        summary = resolution.get("summary", "")
+
+        raw_a = preferences_a.get("raw_details", {})
+        raw_b = preferences_b.get("raw_details", {})
+        decision_type = raw_a.get("decision_type") or raw_b.get("decision_type") or "Decision"
+
+        if status == "escalated":
+            return (
+                f"⚠️ *{decision_type.title()} — Human Decision Needed*\n\n"
+                f"_{summary}_\n\n"
+                f"Agents proposed options but couldn't finalize in {rounds} round(s)."
+            )
+
+        chosen = (
+            details.get("chosen_option") or details.get("plan") or details.get("option")
+            or details.get("decision") or final.get("summary", "")
+        )
+        cost = details.get("monthly_cost") or details.get("cost") or details.get("price") or ""
+        split = details.get("split") or details.get("each_pays") or ""
+        rules = details.get("rules") or details.get("terms") or []
+
+        lines = [f"🏠 *{decision_type.title()} — Decision Made!*\n"]
+        if chosen:
+            lines.append(f"✅ *Choice:* {chosen}")
+        if cost:
+            lines.append(f"💰 *Cost:* ₹{cost}/month")
+        if split:
+            lines.append(f"💳 *Each pays:* ₹{split}")
+        if rules and isinstance(rules, list):
+            lines.append("📋 *Agreed rules:*")
+            for rule in rules[:4]:
+                lines.append(f"  • {rule}")
+        elif rules:
+            lines.append(f"📋 *Terms:* {rules}")
+        lines.append(f"\n⏱ Decided in {rounds} round(s)")
+        if summary and summary != "Agreement reached":
+            lines.append(f"_{summary}_")
+
+        return "\n".join(lines)