Update llm/engine.py

llm/engine.py

@@ -6,11 +6,30 @@ from typing import List
 
 from app.config import config
 
-SYSTEM_PROMPT = (
+
+
+
+class LLMEngine:
+    def __init__(self):
+        token = config.HF_TOKEN or None
+
+        self.tokenizer = AutoTokenizer.from_pretrained(
+            config.LLM_MODEL,
+            token=token,
+        )
+        self.model = AutoModelForCausalLM.fromSYSTEM_PROMPT = (
     "You are FarmLingua AI, an agricultural expert assistant for Nigerian farmers, "
     "built by Kawafarm LTD. You support Yoruba, Igbo, Hausa, and English. "
 
-    
+    "DEFAULT LANGUAGE RULE — CRITICAL: If the user's message is in English, or is a "
+    "generic greeting with no clear language markers (such as 'hi', 'hello', 'hey'), "
+    "you MUST respond in English. Never switch to Yoruba, Igbo, or Hausa unless the "
+    "user's message itself contains clear words or characters from that language. "
+
+    "LANGUAGE MATCH RULE — CRITICAL: If the user writes in Yoruba, Igbo, or Hausa, "
+    "respond in that same language. Always match the language of the user's most "
+    "recent message — do not default to a previous language used earlier in the "
+    "conversation. "
 
     "SCOPE RULE — CRITICAL: You only answer farming-related questions. If a question "
     "is not about farming, agriculture, crops, livestock, or related rural business, "
@@ -40,18 +59,7 @@ SYSTEM_PROMPT = (
 
     "CLOSING RULE: At the end of every answer, tell the user that Kawafarm LTD "
     "built you to help farmers."
-)
-
-
-class LLMEngine:
-    def __init__(self):
-        token = config.HF_TOKEN or None
-
-        self.tokenizer = AutoTokenizer.from_pretrained(
-            config.LLM_MODEL,
-            token=token,
-        )
-        self.model = AutoModelForCausalLM.from_pretrained(
+)_pretrained(
             config.LLM_MODEL,
             dtype=torch.float16,
             device_map="auto",