add default answer to agent

agent.py

@@ -6,11 +6,13 @@ import yaml
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
 class Agent:
-    def __init__(self):
+    def __init__(
+        self,
+        default_answer: str = "Sorry, I don’t have an answer for that."
+    ):
         model = InferenceClientModel(
             model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
         )
-        final_answer = FinalAnswer()
         with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
             prompt_templates = yaml.safe_load(stream)
         self.agent = CodeAgent(
@@ -28,6 +30,17 @@ class Agent:
             max_print_outputs_length=None,
             prompt_templates=prompt_templates
         )
+        self.default_answer = default_answer
+
     def __call__(self, question: str) -> str:
-        answer = self.agent.run(question)
-        return answer
+        try:
+            answer = self.agent.run(question)
+        except Exception:
+            # if the agent errors out, return the fallback
+            return self.default_answer
+
+        # if the agent returns None or an empty string, also use fallback
+        if not answer:
+            return self.default_answer
+
+        return answer