entidi2608 commited on
Commit
749ee37
·
1 Parent(s): b6a3050

update: retriever

Browse files
Files changed (1) hide show
  1. utils/AdvancedLawRetriever.py +7 -0
utils/AdvancedLawRetriever.py CHANGED
@@ -253,6 +253,13 @@ class AdvancedLawRetriever(BaseRetriever):
253
 
254
  logger.info("Intent detected: Administrative Procedure -> Preferring 'THÔNG TƯ'.")
255
 
 
 
 
 
 
 
 
256
  # 3. NGUYÊN TẮC CHUNG, QUYỀN NGHĨA VỤ, CẤU TRÚC NHÀ NƯỚC - Ưu tiên LUẬT
257
  elif any(kw in query_lower for kw in [
258
  # Nguyên tắc cơ bản
 
253
 
254
  logger.info("Intent detected: Administrative Procedure -> Preferring 'THÔNG TƯ'.")
255
 
256
+ elif any(kw in query_lower for kw in [
257
+ "diện tích", "dân số", "số lượng tỉnh", "bao nhiêu tỉnh", "danh sách"
258
+ ]) and any(loc_kw in query_lower for loc_kw in [
259
+ "tỉnh", "thành phố", "huyện", "xã", "việt nam"
260
+ ]):
261
+ info["preferred_doc_type"] = "NGHỊ QUYẾT" # Hoặc có thể là một list ['NGHỊ QUYẾT', 'QUYẾT ĐỊNH']
262
+ logger.info("Intent detected: Specific Factual Data (Area, Population) -> Preferring 'NGHỊ QUYẾT'.")
263
  # 3. NGUYÊN TẮC CHUNG, QUYỀN NGHĨA VỤ, CẤU TRÚC NHÀ NƯỚC - Ưu tiên LUẬT
264
  elif any(kw in query_lower for kw in [
265
  # Nguyên tắc cơ bản