improved prompt

app.py

@@ -133,9 +133,15 @@ class AgentWrapper:
 
 def assistant(state: MyAgent):
     # Add system message to instruct the agent to use the tool
-    system_message = SystemMessage(content="""You are a general AI assistant. I will ask you a question.
-Report your thoughts, and finish your answer with just the answer — no prefixes like "FINAL ANSWER:".
-Your answer should be a number OR as few words as possible OR a comma-separated list of numbers and/or strings.
+    system_message = SystemMessage(content="""
+You are a general AI assistant.
+
+**Only output the answer. Do not include any explanation, context, or restate the question.**
+If the answer is a number, output only the number.
+If the answer is a string, output only the string.
+If the answer is a list, output only a comma-separated list.
+Do not use any prefixes, suffixes, or extra words.
+
 If you're asked for a number, don't use commas or units like $ or %, unless specified.
 If you're asked for a string, don't use articles or abbreviations (e.g. for cities), and write digits in plain text unless told otherwise.
 
@@ -172,7 +178,8 @@ FILE PROCESSING:
 - File processing is automatically handled before the question reaches you
 - If a file is attached, you'll see the file content/analysis in the question
 - Focus on answering based on the enhanced question that includes file information
-- You don't need to process files manually - just use the information provided""")
+- You don't need to process files manually - just use the information provided
+""")
 
     # Combine system message with user messages
     all_messages = [system_message] + state["messages"]