Update app.py

app.py

@@ -35,6 +35,15 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 # Initialize HF client
 client = InferenceClient(token=HF_TOKEN)
 
+# -------------------------
+# Constants
+# -------------------------
+
+SYSTEM = (
+    "You are a parser-safe assistant.\n"
+    "Output **ONLY** the JSON object requested—no extra words."
+)
+
 # -------------------------
 # Utility helpers
 # -------------------------
@@ -190,9 +199,9 @@ class BasicAgent:
     def _call_llm(self, prompt: str, max_tokens: int = 256) -> str:
         try:
             resp = self.llm.chat.completions.create(
-                model="gpt-4.1-mini",
+                model="gpt-4.1",
                 messages=[
-                    {"role": "system", "content": "You are a careful reasoning assistant."},
+                    {"role": "system", "content": SYSTEM},
                     {"role": "user", "content": prompt},
                 ],
                 temperature=0.3,