Update prompt_builder.py

prompt_builder.py

@@ -1,12 +1,40 @@
 """
-Flare – Prompt Builder  (v5 · improved parameter prompt)
+Flare – Prompt Builder  (v6 · date handling)
 ==============================================================
 """
 
 from typing import List, Dict
+from datetime import datetime, timedelta
 from utils import log
 
 
+# Date helper for Turkish date expressions
+def _get_date_context() -> Dict[str, str]:
+    """Generate date context for Turkish date expressions"""
+    now = datetime.now()
+    
+    # Weekday names in Turkish
+    weekdays_tr = ["Pazartesi", "Salı", "Çarşamba", "Perşembe", "Cuma", "Cumartesi", "Pazar"]
+    today_weekday = weekdays_tr[now.weekday()]
+    
+    # Calculate various dates
+    dates = {
+        "today": now.strftime("%Y-%m-%d"),
+        "tomorrow": (now + timedelta(days=1)).strftime("%Y-%m-%d"),
+        "day_after_tomorrow": (now + timedelta(days=2)).strftime("%Y-%m-%d"),
+        "this_weekend_saturday": (now + timedelta(days=(5-now.weekday())%7)).strftime("%Y-%m-%d"),
+        "this_weekend_sunday": (now + timedelta(days=(6-now.weekday())%7)).strftime("%Y-%m-%d"),
+        "next_week_same_day": (now + timedelta(days=7)).strftime("%Y-%m-%d"),
+        "two_weeks_later": (now + timedelta(days=14)).strftime("%Y-%m-%d"),
+        "today_weekday": today_weekday,
+        "today_day": now.day,
+        "today_month": now.month,
+        "today_year": now.year
+    }
+    
+    return dates
+
+
 # ─────────────────────────────────────────────────────────────────────────────
 # INTENT PROMPT
 # ─────────────────────────────────────────────────────────────────────────────
@@ -53,8 +81,11 @@ def build_parameter_prompt(intent_cfg,
                            user_input: str,
                            conversation: List[Dict[str, str]]) -> str:
     
+    date_ctx = _get_date_context()
+    
     parts: List[str] = [
-        "You are extracting parameters from user messages.",
+        "You are extracting parameters from user messages in TURKISH.",
+        f"Today is {date_ctx['today']} ({date_ctx['today_weekday']}). Tomorrow is {date_ctx['tomorrow']}.",
         "Extract ONLY the parameters listed below from the conversation.",
         "Look at BOTH the current message AND previous messages to find parameter values.",
         "If a parameter cannot be found, is invalid, or wasn't provided, keep it in the \"missing\" list.",
@@ -66,7 +97,28 @@ def build_parameter_prompt(intent_cfg,
     parts.append("Parameters to extract:")
     for p in intent_cfg.parameters:
         if p.name in missing_params:
-            parts.append(f"• {p.name}: {p.extraction_prompt}")
+            # Special handling for date type parameters
+            if p.type == "date":
+                date_prompt = (
+                    f"• {p.name}: {p.extraction_prompt}\n"
+                    f"  IMPORTANT DATE RULES:\n"
+                    f"  - 'bugün' = {date_ctx['today']}\n"
+                    f"  - 'yarın' = {date_ctx['tomorrow']}\n"
+                    f"  - 'öbür gün' = {date_ctx['day_after_tomorrow']}\n"
+                    f"  - 'bu hafta sonu' = {date_ctx['this_weekend_saturday']} or {date_ctx['this_weekend_sunday']}\n"
+                    f"  - 'bu cumartesi' = {date_ctx['this_weekend_saturday']}\n"
+                    f"  - 'bu pazar' = {date_ctx['this_weekend_sunday']}\n"
+                    f"  - 'haftaya' or 'gelecek hafta' = add 7 days to current date\n"
+                    f"  - 'haftaya bugün' = {date_ctx['next_week_same_day']}\n"
+                    f"  - 'iki hafta sonra' = {date_ctx['two_weeks_later']}\n"
+                    f"  - '15 gün sonra' = add 15 days to today\n"
+                    f"  - '10 Temmuz' = {date_ctx['today_year']}-07-10\n"
+                    f"  - Turkish months: Ocak=01, Şubat=02, Mart=03, Nisan=04, Mayıs=05, Haziran=06, "
+                    f"Temmuz=07, Ağustos=08, Eylül=09, Ekim=10, Kasım=11, Aralık=12"
+                )
+                parts.append(date_prompt)
+            else:
+                parts.append(f"• {p.name}: {p.extraction_prompt}")
     
     # Add format instruction
     parts.append("")