Update prompt_builder.py

prompt_builder.py

@@ -7,15 +7,24 @@ from typing import List, Dict
 from datetime import datetime, timedelta
 from utils import log
 from config_provider import ConfigProvider
+from locale_manager import LocaleManager
 
-# Date helper for Turkish date expressions
-def _get_date_context() -> Dict[str, str]:
-    """Generate date context for Turkish date expressions"""
+# Date helper for date expressions
+def _get_date_context(locale_code: str = "tr-TR") -> Dict[str, str]:
+    """Generate date context based on locale"""
     now = datetime.now()
     
-    # Weekday names in Turkish
-    weekdays_tr = ["Pazartesi", "Salı", "Çarşamba", "Perşembe", "Cuma", "Cumartesi", "Pazar"]
-    today_weekday = weekdays_tr[now.weekday()]
+    # Locale verilerini yükle
+    locale_data = LocaleManager.get_locale(locale_code)
+    
+    # Weekday names from locale
+    days = locale_data.get("days", {})
+    weekdays = [days.get(str(i), f"Day{i}") for i in range(7)]
+    
+    # Monday is 0 in Python, Sunday is 6, but in locale Sunday is 0
+    python_weekday = now.weekday()  # 0=Monday, 6=Sunday
+    locale_weekday = (python_weekday + 1) % 7  # Convert to locale format where 0=Sunday
+    today_weekday = days.get(str(locale_weekday), "")
     
     # Calculate various dates
     dates = {
@@ -29,12 +38,12 @@ def _get_date_context() -> Dict[str, str]:
         "today_weekday": today_weekday,
         "today_day": now.day,
         "today_month": now.month,
-        "today_year": now.year
+        "today_year": now.year,
+        "locale_code": locale_code
     }
     
     return dates
 
-
 # ─────────────────────────────────────────────────────────────────────────────
 # INTENT PROMPT
 # ─────────────────────────────────────────────────────────────────────────────
@@ -180,12 +189,18 @@ _FMT = """#PARAMETERS:{"extracted":[{"name":"<param>","value":"<val>"},...],"mis
 def build_parameter_prompt(intent_cfg,
                            missing_params: List[str],
                            user_input: str,
-                           conversation: List[Dict[str, str]]) -> str:
-    
+                           conversation: List[Dict[str, str]],
+                           locale_code: str = None) -> str:
+    # Intent'in locale'ini kullan, yoksa default
+    if not locale_code:
+        locale_code = getattr(intent_cfg, 'locale', 'tr-TR')
+
     date_ctx = _get_date_context()
+
+    locale_data = LocaleManager.get_locale(locale_code)
     
     parts: List[str] = [
-        "You are extracting parameters from user messages in TURKISH.",
+        f"You are extracting parameters from user messages in {locale_data.get('name', 'the target language')}.",
         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.",
@@ -206,22 +221,8 @@ def build_parameter_prompt(intent_cfg,
         if p.name in missing_params:
             # 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"
+                date_prompt = _build_locale_aware_date_prompt(
+                    p, date_ctx, locale_data, locale_code
                 )
                 parts.append(date_prompt)
             else:
@@ -247,6 +248,60 @@ def build_parameter_prompt(intent_cfg,
     log(f"📝 Parameter prompt built for missing: {missing_params}")
     return prompt
 
+def _build_locale_aware_date_prompt(param, date_ctx: Dict, locale_data: Dict, locale_code: str) -> str:
+    """Build date extraction prompt based on locale"""
+    
+    # Locale'e özel date expressions
+    date_expressions = locale_data.get("date_expressions", {})
+    months = locale_data.get("months", {})
+    
+    prompt_parts = [
+        f"• {param.name}: {param.extraction_prompt}",
+        "  IMPORTANT DATE RULES:"
+    ]
+    
+    # Common date expressions
+    if locale_code.startswith("tr"):
+        # Turkish specific
+        prompt_parts.extend([
+            f"  - 'bugün' = {date_ctx['today']}",
+            f"  - 'yarın' = {date_ctx['tomorrow']}",
+            f"  - 'öbür gün' or 'ertesi gün' = {date_ctx['day_after_tomorrow']}",
+            f"  - 'bu hafta sonu' = {date_ctx['this_weekend_saturday']} or {date_ctx['this_weekend_sunday']}",
+            f"  - 'bu cumartesi' = {date_ctx['this_weekend_saturday']}",
+            f"  - 'bu pazar' = {date_ctx['this_weekend_sunday']}",
+            f"  - 'haftaya' or 'gelecek hafta' = add 7 days to current date",
+            f"  - 'haftaya bugün' = {date_ctx['next_week_same_day']}",
+            f"  - 'iki hafta sonra' = {date_ctx['two_weeks_later']}",
+            f"  - 'X gün sonra' = add X days to today"
+        ])
+    elif locale_code.startswith("en"):
+        # English specific
+        prompt_parts.extend([
+            f"  - 'today' = {date_ctx['today']}",
+            f"  - 'tomorrow' = {date_ctx['tomorrow']}",
+            f"  - 'day after tomorrow' = {date_ctx['day_after_tomorrow']}",
+            f"  - 'this weekend' = {date_ctx['this_weekend_saturday']} or {date_ctx['this_weekend_sunday']}",
+            f"  - 'this Saturday' = {date_ctx['this_weekend_saturday']}",
+            f"  - 'this Sunday' = {date_ctx['this_weekend_sunday']}",
+            f"  - 'next week' = add 7 days to current date",
+            f"  - 'in X days' = add X days to today"
+        ])
+    # Diğer diller için de eklenebilir
+    
+    # Month names
+    if months:
+        month_list = ", ".join([
+            f"{name}={num}" for num, name in sorted(months.items())
+        ])
+        prompt_parts.append(f"  - Month names: {month_list}")
+    
+    # Date format hint
+    date_format = locale_data.get("date_format", "YYYY-MM-DD")
+    prompt_parts.append(f"  - Expected date format: YYYY-MM-DD (convert from {date_format})")
+    
+    return "\n".join(prompt_parts)
+    
 def build_smart_parameter_question_prompt(
     collection_config,  # ParameterCollectionConfig
     intent_config,      # IntentConfig