Add system prompt

app.py

@@ -32,11 +32,15 @@ class BasicAgent:
         
         # Create the model
         self.model = InferenceClientModel()
+
+        # Define your system prompt
+        self.system_prompt = """You are a general AI assistant. I will ask you a question. Finish your answer with only: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
         
         # Create the agent with tools
         self.agent = CodeAgent(
             tools=[self.search_tool],
             model=self.model,
+            system_prompt=self.system_prompt,
             max_steps=5  # Limit reasoning steps
         )