fix prompt
Browse files- rag_core/business.py +11 -1
rag_core/business.py
CHANGED
|
@@ -55,7 +55,17 @@ def answer_query(query: str) -> str:
|
|
| 55 |
try:
|
| 56 |
docs = retriever.query(query, get_embedding)
|
| 57 |
logging.info(f"📚 Truy xuất được {len(docs)} đoạn liên quan")
|
| 58 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
answer = generate_answer(prompt)
|
| 60 |
return {"answer": answer}
|
| 61 |
except Exception as e:
|
|
|
|
| 55 |
try:
|
| 56 |
docs = retriever.query(query, get_embedding)
|
| 57 |
logging.info(f"📚 Truy xuất được {len(docs)} đoạn liên quan")
|
| 58 |
+
for i, doc in enumerate(docs):
|
| 59 |
+
logging.info(f"🔍 Đoạn {i+1}: {doc[:300]}{'...' if len(doc) > 300 else ''}")
|
| 60 |
+
|
| 61 |
+
formatted_context = "\n\n".join([f"[Văn bản {i+1}]\n{doc}" for i, doc in enumerate(docs)])
|
| 62 |
+
prompt = (
|
| 63 |
+
"Bạn là một trợ lý pháp lý. Dưới đây là các trích đoạn liên quan từ văn bản luật:\n\n"
|
| 64 |
+
f"{formatted_context}\n\n"
|
| 65 |
+
"Hãy trả lời câu hỏi sau dựa trên các trích đoạn trên. "
|
| 66 |
+
"Nếu không đủ thông tin, hãy nói rõ là không tìm thấy.\n\n"
|
| 67 |
+
f"Câu hỏi: {query}\nTrả lời:"
|
| 68 |
+
)
|
| 69 |
answer = generate_answer(prompt)
|
| 70 |
return {"answer": answer}
|
| 71 |
except Exception as e:
|