updated agent

app.py

@@ -12,26 +12,32 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-        
-        HF_TOKEN = os.environ.get("HF_TOKEN")
-
-        
+        print("Srj's Agent initialized.")
+        self.client = InferenceClient(model= "moonshotai/Kimi-K2.5", token = os.environ.get("HF_TOKEN"))
+               
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
-        promt = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [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.
-        This is the question that you need to answer : {question}"""
-        self.client = InferenceClient(model= "moonshotai/Kimi-K2.5")
+        promt = f"""Question: {question}
+
+Solve step-by-step internally but respond with ONLY the final answer.
+NO "Final Answer:", NO explanation, NO reasoning.
+Just the exact answer string (number, city name, or short list).
+
+Examples:
+Capital of France? → Paris
+2024 Olympics city? → Paris
+15 × 3? → 45"""
+        
         output = self.client.chat.completions.create(
             messages= [
             {"role":"User","content":question}
             ],
             stream= False,
-            max_tokens= 1024,
+            max_tokens= 50,
             extra_body={"thinking":{"type":'disabled'}}
         )
-        return output.choices[0].message.content
+        return output.choices[0].message.content.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """