Update agent.py

agent.py

@@ -13,13 +13,9 @@ qa_pipeline = pipeline("text2text-generation", model="Qwen/Qwen1.5-1.8B-Chat")
 
 def run_agent_on_question(question_obj):
     question = question_obj["question"]
-
-    prompt = f"Answer the following question as concisely as possible:\n{question}"
+    prompt = f"Answer this question:\n{question}"
     result = qa_pipeline(prompt, max_new_tokens=50, do_sample=False)[0]["generated_text"]
-    
-    # Strip any prefix like "Answer:" or whitespace
-    return result.strip().split("\n")[0].replace("Answer: ", "").strip()
-
+    return result.strip().replace("Answer: ", "").split("\n")[0]
     
 def simple_llm_call(prompt: str) -> str:
     """