update

rag_pipeline.py

@@ -118,22 +118,26 @@ def generate_response(query: str, components: dict) -> str:
     
     # Tạo cấu trúc tin nhắn theo chuẩn
     messages = [
-        {
-            "role": "system",
-            "content": "Bạn là một trợ lý pháp luật chuyên trả lời các câu hỏi về luật giao thông Việt Nam. Hãy dựa vào các thông tin được cung cấp để trả lời một cách chính xác và dễ hiểu."
-        },
-        {
-            "role": "user",
-            "content": f"""Dựa vào các trích dẫn luật dưới đây:
+    {
+        "role": "system",
+        "content": [
+            {"type": "text", "text": "Bạn là một trợ lý pháp luật chuyên trả lời các câu hỏi về luật giao thông Việt Nam..."}
+        ]
+    },
+    {
+        "role": "user",
+        "content": [
+            {"type": "text", "text": f"""Dựa vào các trích dẫn luật dưới đây:
 
 ### Thông tin luật:
 {context}
 
 ### Câu hỏi:
 {query}
-"""
-        }
-    ]
+"""}
+        ]
+    }
+]
     
     # SỬA LỖI: Dùng apply_chat_template để tokenize trực tiếp
     # Nó sẽ tự động định dạng và chuyển thành tensor, tương thích với mô hình Vision