Update agent.py

agent.py

@@ -15,7 +15,6 @@ groq_api_key = os.getenv("GROQ_API_KEY")
 serpapi_api_key = os.getenv("SERPAPI_API_KEY")
 
 # --- أدوات ---
-
 def calculator_tool_func(query: str):
     try:
         result = str(eval(query, {"__builtins__": {}}))
@@ -45,8 +44,7 @@ SerpAPI_tool = Tool(
     description="Searches the web for recent information."
 )
 
-# --- تعريف موديل الذكاء الاصطناعي والوكيل ---
-
+# --- تعريف الموديل والوكلاء ---
 tools = [SerpAPI_tool, calculator_tool, weather_tool]
 llm = ChatGroq(model="deepseek-r1-distill-llama-70b", groq_api_key=groq_api_key)
 llm_with_tools = llm.bind_tools(tools)
@@ -64,11 +62,10 @@ builder.add_edge(START, "assistant")
 builder.add_conditional_edges("assistant", tools_condition)
 builder.add_edge("tools", "assistant")
 
-# 👇 وكيل باسم ninu يمكن استيراده من أي ملف آخر
+# 👇 compiled agent object (must be used with .invoke({...}))
 ninu = builder.compile()
 
 # --- دالة تشغيل مستقلة (اختياري) ---
-
 def run_ninu(query: str):
     conversation = []
 
@@ -86,4 +83,7 @@ Only include what the user asked for in the final answer.
     conversation.append(HumanMessage(content=query))
 
     response = ninu.invoke({"messages": conversation})
-    return response["messages"][-1].content
+    for message in reversed(response["messages"]):
+        if isinstance(message.content, str) and "FINAL ANSWER:" in message.content:
+            return message.content.split("FINAL ANSWER:")[-1].strip()
+    return "لم أتمكن من إيجاد إجابة نهائية."