fix bug LLM
Browse files- app/main.py +4 -1
app/main.py
CHANGED
|
@@ -207,7 +207,10 @@ async def process_message(message_data: Dict[str, Any]):
|
|
| 207 |
hanh_vi_vi_pham = llm_analysis[0].get('hanh_vi_vi_pham')
|
| 208 |
else:
|
| 209 |
keywords = extract_keywords(message_text, VEHICLE_KEYWORDS)
|
| 210 |
-
hanh_vi_vi_pham = message_text
|
|
|
|
|
|
|
|
|
|
| 211 |
|
| 212 |
logger.info(f"[DEBUG] Phương tiện: {keywords} - Hành vi: {hanh_vi_vi_pham} - Mục đích: {muc_dich}")
|
| 213 |
|
|
|
|
| 207 |
hanh_vi_vi_pham = llm_analysis[0].get('hanh_vi_vi_pham')
|
| 208 |
else:
|
| 209 |
keywords = extract_keywords(message_text, VEHICLE_KEYWORDS)
|
| 210 |
+
hanh_vi_vi_pham = message_text
|
| 211 |
+
for kw in keywords:
|
| 212 |
+
hanh_vi_vi_pham = hanh_vi_vi_pham.replace(kw, "")
|
| 213 |
+
hanh_vi_vi_pham = hanh_vi_vi_pham.strip()
|
| 214 |
|
| 215 |
logger.info(f"[DEBUG] Phương tiện: {keywords} - Hành vi: {hanh_vi_vi_pham} - Mục đích: {muc_dich}")
|
| 216 |
|