Update tools.py

tools.py

@@ -1,10 +1,59 @@
-def tool_router(plan_text):
-    # For demo: simple simulation
-    if "<action_name>book_free_trial</action_name>" in plan_text:
-        return "✅ تم حجز تجربتك المجانية! نراك قريباً 💪🏼"
-    elif "<action_name>pause_subscription</action_name>" in plan_text:
-        return "✅ تم إيقاف اشتراكك مؤقتاً حسب السياسة."
-    elif "<action_name>cancel_subscription</action_name>" in plan_text:
-        return "✅ تم إلغاء الاشتراك، وستصلك تفاصيل الاسترداد قريباً."
-    else:
-        return "🔧 تم تنفيذ الخطة بنجاح."
+from langchain_core.tools import tool
+
+@tool
+def book_free_trial(name: str, contact: str, preferred_date: str) -> str:
+    """Book a free 30-minute introductory class for a new user."""
+    return f"Free trial booked for {name} on {preferred_date}. Contact: {contact}"
+
+@tool
+def subscribe_to_membership(name: str, phone_number: str, package_type: str) -> str:
+    """Subscribe user to a membership package."""
+    return f"Membership subscription created for {name} ({phone_number}) - Package: {package_type}"
+
+@tool
+def cancel_membership(phone_number: str, request_time: str) -> str:
+    """Cancel a membership within 48 hours of payment."""
+    return f"Membership cancellation processed for {phone_number} at {request_time}"
+
+@tool
+def postpone_membership(phone_number: str, reason: str, start_date: str, end_date: str) -> str:
+    """Postpone membership for 2 weeks to 2 months."""
+    return f"Membership postponed for {phone_number} from {start_date} to {end_date}. Reason: {reason}"
+
+@tool
+def transfer_membership(phone_number: str, new_member_name: str, new_member_phone_number: str) -> str:
+    """Transfer a membership to another person with consent and 200 SAR fee."""
+    return f"Membership transferred from {phone_number} to {new_member_name} ({new_member_phone_number}). Fee: 200 SAR"
+
+@tool
+def get_current_date() -> str:
+    """Get the current date in YYYY-MM-DD format."""
+    from datetime import datetime
+    return datetime.now().strftime("%Y-%m-%d")
+
+@tool 
+def calculate_date_range(start_date: str, duration_months: int) -> str:
+    """Calculate end date given start date and duration in months."""
+    from datetime import datetime
+    from dateutil.relativedelta import relativedelta
+    
+    start = datetime.strptime(start_date, "%Y-%m-%d")
+    end = start + relativedelta(months=duration_months)
+    
+    return f"Start: {start.strftime('%Y-%m-%d')}, End: {end.strftime('%Y-%m-%d')}, Duration: {(end - start).days} days"
+
+def format_messages_with_actions(messages):
+    """Format messages for the manager prompt"""
+    formatted = []
+    for msg in messages:
+        if hasattr(msg, 'content'):
+            formatted.append(f"Message: {msg.content}")
+        elif hasattr(msg, 'invoke'):
+            # Handle tool calls properly
+            formatted.append(f"Tool call: {str(msg)}")
+    return "\n".join(formatted)
+
+
+
+tools_list = [ calculate_date_range, get_current_date, book_free_trial, subscribe_to_membership, cancel_membership, postpone_membership, transfer_membership ]
+