Update llm_client.py

llm_client.py

@@ -96,12 +96,12 @@ class HybridLLM(LLM):
                 "prompt": prompt,
                 "n_predict": 512,
                 "temperature": 0.3,
-                "stop": stop or []
+                "stop": (stop or []) + ["<|im_end|>", "Input:", "Context:"]
             }
             response = requests.post(
                 f"{self.local_server_url}/completion",
                 json=payload,
-                timeout=120
+                timeout=300
             )
             if response.status_code == 200:
                 return response.json()["content"]
@@ -215,20 +215,22 @@ class LLMClient:
             return "❌ Vector Store not initialized."
 
         # Custom Prompt Template
+        # Custom Prompt Template - Stricter Format
         template = """<|im_start|>system
-You are a cybersecurity expert. Task: Determine whether the input is 'PHISHING' or 'BENIGN' (Safe). 
-Respond in the following format:
+You are a strict cybersecurity assistant.
+Your ONLY task is to analyze the input and provide a verdict.
+Output MUST follow this EXACT format:
 LABEL: [PHISHING or BENIGN]
-EXPLANATION: [A brief Vietnamese explanation]
+EXPLANATION: [3-4 sentences in Vietnamese explaining the verdict]
 
-Context:
-{context}
+Do NOT provide any other text, headers, or markdown.
 <|im_end|>
 <|im_start|>user
-Input:
-{question}
+Context from knowledge base:
+{context}
 
-Short Analysis:
+Input to analyze:
+{question}
 <|im_end|>
 <|im_start|>assistant
 """