Update chat_handler.py

chat_handler.py

@@ -3,6 +3,7 @@ Flare – Chat Handler  (v1.7 · parameter parsing düzeltmesi)
 ==========================================
 """
 
+import os
 import re, json, sys, httpx
 from datetime import datetime
 from typing import Dict, List, Optional
@@ -52,6 +53,20 @@ SPARK_URL = str(cfg.global_config.spark_endpoint).rstrip("/")
 ALLOWED_INTENTS = {"flight-booking", "flight-info", "booking-cancel"}
 
 # ─────────────────────────  SPARK ───────────────────────── #
+def _get_spark_token() -> Optional[str]:
+    """Get Spark token based on work_mode"""
+    if cfg.global_config.is_cloud_mode():
+        # Cloud mode - use HuggingFace Secrets
+        token = os.getenv("SPARK_TOKEN")
+        if not token:
+            log("❌ SPARK_TOKEN not found in HuggingFace Secrets!")
+        return token
+    else:
+        # On-premise mode - use .env file
+        from dotenv import load_dotenv
+        load_dotenv()
+        return os.getenv("SPARK_TOKEN")
+        
 async def spark_generate(s: Session, prompt: str, user_msg: str) -> str:
     """Call Spark with proper error handling"""
     try:
@@ -63,24 +78,50 @@ async def spark_generate(s: Session, prompt: str, user_msg: str) -> str:
         if not version:
             raise ValueError("No published version found")
         
+        # Get Spark token
+        spark_token = _get_spark_token()
+        if not spark_token:
+            log("❌ SPARK_TOKEN not configured!")
+            raise ValueError("Spark authentication token not configured")
+        
+        # Prepare headers with authorization
+        headers = {
+            "Authorization": f"Bearer {spark_token}",
+            "Content-Type": "application/json"
+        }
+        
+        # Spark'a gönderilecek payload'ı hazırla
         payload = {
             "project_name": s.project_name,
+            "system_prompt": prompt,
             "user_input": user_msg,
-            "context": s.chat_history[-10:],
-            "system_prompt": prompt
+            "context": s.chat_history[-10:]
         }
         
         log(f"🚀 Calling Spark for session {s.session_id[:8]}...")
         log(f"📋 Prompt preview (first 200 chars): {prompt[:200]}...")
         
+        # Doğru endpoint'e istek at
+        spark_url = SPARK_URL + "/generate"
+        log(f"🌐 Spark URL: {spark_url}")
+        
         async with httpx.AsyncClient(timeout=60) as client:
-            response = await client.post(SPARK_URL + "/generate", json=payload)
+            response = await client.post(spark_url, json=payload, headers=headers)
             response.raise_for_status()
             data = response.json()
-            raw = (data.get("assistant") or data.get("model_answer") or data.get("text", "")).strip()
+            
+            # Spark'tan gelen yanıtı parse et
+            raw = data.get("model_answer", "").strip()
+            if not raw:
+                # Fallback to other possible fields
+                raw = (data.get("assistant") or data.get("text", "")).strip()
+            
             log(f"🪄 Spark raw: {raw[:120]!r}")
             return raw
             
+    except httpx.HTTPStatusError as e:
+        log(f"❌ Spark HTTP error: {e.response.status_code} - {e.response.text}")
+        raise
     except httpx.TimeoutException:
         log(f"⏱️ Spark timeout for session {s.session_id[:8]}")
         raise