Update app.py

app.py

@@ -110,11 +110,19 @@ class BasicAgent:
             prompt.append(f"{role}: {text}")
         return "\n".join(prompt)
 
-    def generate(self, question: str) -> str:
+    def generate(self, question: str, stop_sequences=None, **kwargs) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-
+        # 1. Build the HF kwargs
         allowed = {"max_new_tokens", "temperature", "top_k", "top_p"}
         gen_kwargs = {k: v for k, v in kwargs.items() if k in allowed}
+        
+        # 2. Optionally map SmolAgents’ stop_sequences → HF pipeline’s 'stop'
+        if stop_sequences is not None:
+            # if your pipeline accepts `stop`, you can do:
+            gen_kwargs["stop"] = stop_sequences
+            # otherwise just ignore it 
+        
+        # 3. Serialize the message
         prompt_str = (
             self._serialize_messages(prompt)
             if isinstance(prompt, list)