fix vector search

app/main.py

@@ -399,51 +399,63 @@ async def format_search_results(question: str, matches: List[Dict[str, Any]]) ->
     top_result_text = ""
     full_result_text = ""
 
+    def arr_to_str(arr, sep=", "):
+        if not arr:
+            return ""
+        if isinstance(arr, list):
+            return sep.join([str(x) for x in arr if x not in (None, "")])
+        return str(arr)
+
     for i, match in enumerate(matches, 1):
         if not top or (match.get('similarity', 0) > top.get('similarity', 0)):
             top = match
 
-        # Chuẩn bị context cho LLM: liệt kê tất cả các item với chú thích rõ ràng
         full_result_text += f"\nĐoạn {i}:\n"
-         # Thực hiện hành vi
-        tieude = (top.get('tieude') or '').strip()
-        noidung = (top.get('noidung') or '').strip()
+        tieude = (match.get('tieude') or '').strip()
+        noidung = (match.get('noidung') or '').strip()
         hanhvi = (tieude + "\n" + noidung).strip().replace('\n', ' ')
         full_result_text += f"Thực hiện hành vi:\n{hanhvi}"
         # Cá nhân bị phạt tiền
-        if top.get('canhantu') or top.get('canhanden'):
-            full_result_text += f"\nCá nhân sẽ bị phạt tiền từ {top.get('canhantu', '')} VNĐ đến {top.get('canhanden', '')} VNĐ"
+        canhanTu = arr_to_str(match.get('canhanTu'))
+        canhanDen = arr_to_str(match.get('canhanDen'))
+        if canhanTu or canhanDen:
+            full_result_text += f"\nCá nhân sẽ bị phạt tiền từ {canhanTu} VNĐ đến {canhanDen} VNĐ"
         # Tổ chức bị phạt tiền
-        if top.get('tochuctu') or top.get('tochucden'):
-            full_result_text += f"\nTổ chức sẽ bị phạt tiền từ {top.get('tochuctu', '')} VNĐ đến {top.get('tochucden', '')} VNĐ"
+        tochucTu = arr_to_str(match.get('tochucTu'))
+        tochucDen = arr_to_str(match.get('tochucDen'))
+        if tochucTu or tochucDen:
+            full_result_text += f"\nTổ chức sẽ bị phạt tiền từ {tochucTu} VNĐ đến {tochucDen} VNĐ"
         # Hình phạt bổ sung
-        if top.get('hpbsnoidung'):
-            full_result_text += f"\nNgoài việc bị phạt tiền, người vi phạm còn bị {top.get('hpbsnoidung')}"
+        hpbsNoidung = arr_to_str(match.get('hpbsNoidung'), sep="; ")
+        if hpbsNoidung:
+            full_result_text += f"\nNgoài việc bị phạt tiền, người vi phạm còn bị: {hpbsNoidung}"
         # Biện pháp khắc phục hậu quả
-        if top.get('bpkpnoidung'):
-            full_result_text += f"\nNgoài ra, người vi phạm còn bị buộc {top.get('bpkpnoidung')}"
-    
+        bpkpNoidung = arr_to_str(match.get('bpkpNoidung'), sep="; ")
+        if bpkpNoidung:
+            full_result_text += f"\nNgoài ra, người vi phạm còn bị buộc: {bpkpNoidung}"
+
     if top and (top.get('tieude') or top.get('noidung')):
-        # Thực hiện hành vi
         tieude = (top.get('tieude') or '').strip()
         noidung = (top.get('noidung') or '').strip()
         hanhvi = (tieude + "\n" + noidung).strip().replace('\n', ' ')
         top_result_text += f"Thực hiện hành vi:\n{hanhvi}"
-        # Cá nhân bị phạt tiền
-        if top.get('canhantu') or top.get('canhanden'):
-            top_result_text += f"\nCá nhân sẽ bị phạt tiền từ {top.get('canhantu', '')} VNĐ đến {top.get('canhanden', '')} VNĐ"
-        # Tổ chức bị phạt tiền
-        if top.get('tochuctu') or top.get('tochucden'):
-            top_result_text += f"\nTổ chức sẽ bị phạt tiền từ {top.get('tochuctu', '')} VNĐ đến {top.get('tochucden', '')} VNĐ"
-        # Hình phạt bổ sung
-        if top.get('hpbsnoidung'):
-            top_result_text += f"\nNgoài việc bị phạt tiền, người vi phạm còn bị {top.get('hpbsnoidung')}"
-        # Biện pháp khắc phục hậu quả
-        if top.get('bpkpnoidung'):
-            top_result_text += f"\nNgoài ra, người vi phạm còn bị buộc {top.get('bpkpnoidung')}"
+        canhanTu = arr_to_str(top.get('canhanTu'))
+        canhanDen = arr_to_str(top.get('canhanDen'))
+        if canhanTu or canhanDen:
+            top_result_text += f"\nCá nhân sẽ bị phạt tiền từ {canhanTu} VNĐ đến {canhanDen} VNĐ"
+        tochucTu = arr_to_str(top.get('tochucTu'))
+        tochucDen = arr_to_str(top.get('tochucDen'))
+        if tochucTu or tochucDen:
+            top_result_text += f"\nTổ chức sẽ bị phạt tiền từ {tochucTu} VNĐ đến {tochucDen} VNĐ"
+        hpbsNoidung = arr_to_str(top.get('hpbsNoidung'), sep="; ")
+        if hpbsNoidung:
+            top_result_text += f"\nNgoài việc bị phạt tiền, người vi phạm còn bị: {hpbsNoidung}"
+        bpkpNoidung = arr_to_str(top.get('bpkpNoidung'), sep="; ")
+        if bpkpNoidung:
+            top_result_text += f"\nNgoài ra, người vi phạm còn bị buộc: {bpkpNoidung}"
     else:
         result_text = "Không có kết quả phù hợp!"
-    
+
     # Prompt cho LLM
     prompt = (
         "Bạn là một trợ lý AI có kiến thức pháp luật, hãy trả lời câu hỏi dựa trên các đoạn luật sau. "
@@ -471,14 +483,20 @@ async def format_search_results(question: str, matches: List[Dict[str, Any]]) ->
         noidung = (match.get('noidung') or '').strip()
         if tieude or noidung:
             fallback += f"  - Hành vi: {(tieude + ' ' + noidung).strip()}\n"
-        if match.get('canhantu') or match.get('canhanden'):
-            fallback += f"  - Cá nhân bị phạt tiền từ {match.get('canhantu', '')} VNĐ đến {match.get('canhanden', '')} VNĐ\n"
-        if match.get('tochuctu') or match.get('tochucden'):
-            fallback += f"  - Tổ chức bị phạt tiền từ {match.get('tochuctu', '')} VNĐ đến {match.get('tochucden', '')} VNĐ\n"
-        if match.get('hpbsnoidung'):
-            fallback += f"  - Hình phạt bổ sung: {match.get('hpbsnoidung')}\n"
-        if match.get('bpkpnoidung'):
-            fallback += f"  - Biện pháp khắc phục hậu quả: {match.get('bpkpnoidung')}\n"
+        canhanTu = arr_to_str(match.get('canhanTu'))
+        canhanDen = arr_to_str(match.get('canhanDen'))
+        if canhanTu or canhanDen:
+            fallback += f"  - Cá nhân bị phạt tiền từ {canhanTu} VNĐ đến {canhanDen} VNĐ\n"
+        tochucTu = arr_to_str(match.get('tochucTu'))
+        tochucDen = arr_to_str(match.get('tochucDen'))
+        if tochucTu or tochucDen:
+            fallback += f"  - Tổ chức bị phạt tiền từ {tochucTu} VNĐ đến {tochucDen} VNĐ\n"
+        hpbsNoidung = arr_to_str(match.get('hpbsNoidung'), sep="; ")
+        if hpbsNoidung:
+            fallback += f"  - Hình phạt bổ sung: {hpbsNoidung}\n"
+        bpkpNoidung = arr_to_str(match.get('bpkpNoidung'), sep="; ")
+        if bpkpNoidung:
+            fallback += f"  - Biện pháp khắc phục hậu quả: {bpkpNoidung}\n"
         fallback += "\n"
     return fallback.strip()