Update agent.py

agent.py

@@ -23,7 +23,7 @@ configure(api_key=os.getenv("GOOGLE_API_KEY"))
 #logger = logging.getLogger(__name__)
 
 # --- Model Configuration ---
-GEMINI_MODEL_NAME = "gemini/gemini-2.0-flash"
+GEMINI_MODEL_NAME = "gemini/gemini-2.0-flash-lite"
 OPENAI_MODEL_NAME = "openai/gpt-4o"
 GROQ_MODEL_NAME = "groq/llama3-70b-8192"
 DEEPSEEK_MODEL_NAME = "deepseek/deepseek-chat"
@@ -250,6 +250,8 @@ class BasicAgent:
             A: b, e
             Q: How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?,
             A: 519
+
+            OUR 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.
             """
         )