VietCat commited on
Commit
f8458ce
·
1 Parent(s): eaee65f

fix bug LLM

Browse files
Files changed (1) hide show
  1. 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.replace(keywords, "")
 
 
 
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