Upload backend/hue_portal/chatbot/slow_path_handler.py with huggingface_hub

backend/hue_portal/chatbot/slow_path_handler.py

@@ -62,7 +62,7 @@ class SlowPathHandler:
         selected_document_code_normalized = (
             selected_document_code.strip().upper() if selected_document_code else None
         )
-        
+
         # Handle greetings
         if intent == "greeting":
             query_lower = query.lower().strip()
@@ -80,7 +80,81 @@ class SlowPathHandler:
                     "count": 0,
                     "_source": "slow_path"
                 }
-        
+
+        # Wizard / option-first cho mọi câu hỏi pháp lý chung:
+        # Nếu:
+        #   - intent là search_legal
+        #   - chưa có selected_document_code trong session
+        #   - trong câu hỏi không ghi rõ mã văn bản
+        # Thì: luôn trả về payload options để người dùng chọn văn bản trước,
+        # chưa generate câu trả lời chi tiết.
+        if (
+            intent == "search_legal"
+            and not selected_document_code_normalized
+            and not self._has_explicit_document_code_in_query(query)
+        ):
+            canonical_candidates: List[Dict[str, Any]] = []
+            try:
+                canonical_docs = list(
+                    LegalDocument.objects.filter(
+                        code__in=["264-QD-TW", "QD-69-TW", "TT-02-CAND"]
+                    )
+                )
+                for doc in canonical_docs:
+                    summary = getattr(doc, "summary", "") or ""
+                    metadata = getattr(doc, "metadata", {}) or {}
+                    if not summary and isinstance(metadata, dict):
+                        summary = metadata.get("summary", "")
+                    canonical_candidates.append(
+                        {
+                            "code": doc.code,
+                            "title": getattr(doc, "title", "") or doc.code,
+                            "summary": summary,
+                            "doc_type": getattr(doc, "doc_type", "") or "",
+                            "section_title": "",
+                        }
+                    )
+            except Exception as exc:
+                logger.warning(
+                    "[CLARIFICATION] Canonical documents lookup failed, using static list: %s",
+                    exc,
+                )
+
+            if not canonical_candidates:
+                canonical_candidates = [
+                    {
+                        "code": "264-QD-TW",
+                        "title": "Quyết định 264-QĐ/TW về kỷ luật đảng viên",
+                        "summary": "",
+                        "doc_type": "",
+                        "section_title": "",
+                    },
+                    {
+                        "code": "QD-69-TW",
+                        "title": "Quy định 69-QĐ/TW về kỷ luật tổ chức đảng, đảng viên",
+                        "summary": "",
+                        "doc_type": "",
+                        "section_title": "",
+                    },
+                    {
+                        "code": "TT-02-CAND",
+                        "title": "Thông tư 02/2021/TT-BCA về điều lệnh CAND",
+                        "summary": "",
+                        "doc_type": "",
+                        "section_title": "",
+                    },
+                ]
+
+            clarification_payload = self._build_clarification_payload(
+                query, canonical_candidates
+            )
+            if clarification_payload:
+                clarification_payload.setdefault("intent", intent)
+                clarification_payload.setdefault("_source", "clarification")
+                clarification_payload.setdefault("routing", "clarification")
+                clarification_payload.setdefault("confidence", 0.3)
+                return clarification_payload
+
         # Search based on intent - retrieve top-15 for reranking (balance speed and RAM)
         search_result = self._search_by_intent(
             intent,
@@ -89,19 +163,6 @@ class SlowPathHandler:
             preferred_document_code=selected_document_code_normalized,
         )  # Balance: 15 for good recall, not too slow
         
-        if intent == "search_legal":
-            clarification = self._maybe_request_clarification(
-                query=query,
-                search_result=search_result,
-                selected_document_code=selected_document_code_normalized,
-            )
-            if clarification:
-                clarification.setdefault("intent", intent)
-                clarification.setdefault("_source", "clarification")
-                clarification.setdefault("routing", "clarification")
-                clarification.setdefault("confidence", 0.3)
-                return clarification
-        
         # Fast path for high-confidence legal queries (skip for complex queries)
         fast_path_response = None
         if intent == "search_legal" and not self._is_complex_query(query):