Update agent.py

agent.py

@@ -43,6 +43,33 @@ def extract_text(query: str, retriever):
         return "لم يتم العثور على معلومات مطابقة في الملفات."
 
 
+# --- Additional Tools --- 
+
+def calculator_tool_func(query: str):
+    try:
+        # أداة حساب بسيطة باستخدام eval (يمكن تطويرها لاحقًا)
+        result = str(eval(query, {"__builtins__": {}}))
+        return result
+    except Exception:
+        return "تعذر حساب التعبير المدخل."
+
+calculator_tool = Tool(
+    name="Calculator",
+    func=calculator_tool_func,
+    description="Performs simple arithmetic calculations."
+)
+
+def weather_tool_func(location: str):
+    # يمكن ربطها ب API حقيقي للطقس لاحقًا
+    return f"حالة الطقس في {location}: مشمس، درجة الحرارة 25 درجة مئوية."
+
+weather_tool = Tool(
+    name="Weather",
+    func=weather_tool_func,
+    description="Provides weather information for a given location."
+)
+
+
 # --- Create NINU Agent ---
 def create_ninu_agent(user_docs=None):
     bm25_retriever = build_retriever(user_docs) if user_docs else None
@@ -66,7 +93,8 @@ def create_ninu_agent(user_docs=None):
         description="Searches the web for recent information."
     )
 
-    tools = [NINU_tool, SerpAPI_tool]
+    # دمج جميع التولز
+    tools = [NINU_tool, 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)
@@ -93,20 +121,16 @@ def run_ninu(query, user_docs=None):
     conversation = []
 
     intro_prompt = """
-You are a general AI assistant with access to two tools:
-
+You are a general AI assistant with access to several tools:
 1. NINU_Lec_retriever: retrieves content from uploaded PDFs based on a query.
 2. WebSearch: performs web searches to answer questions about current events or general knowledge.
-
-Based on the user's query, decide whether to use NINU_Lec_retriever, WebSearch, or both.
-
+3. Calculator: performs arithmetic calculations.
+4. Weather: provides weather information for given locations.
+Based on the user's query, decide whether to use one or more of these tools.
 When answering, 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 commas or units (like $, %, etc.) unless specified.
-
 If you are asked for a string, avoid articles, abbreviations, and write digits in plain text unless specified.
 """
     conversation.append(HumanMessage(content=intro_prompt))
@@ -114,3 +138,7 @@ If you are asked for a string, avoid articles, abbreviations, and write digits i
 
     response = agent.invoke({"messages": conversation})
     return response["messages"][-1].content
+
+
+# تسجيل الوكيل باسم `ninu` ليتم استيراده من ملفات أخرى
+ninu = create_ninu_agent