Update agent.py

agent.py

@@ -254,20 +254,25 @@ sys_msg = SystemMessage(content=system_prompt)
 # -----------------------------
 # Map tool names to functions
 # -----------------------------
-tool_map = {
-    "math": calculator,
-    "wiki_search": wiki_search,
-    "web_search": web_search,
-    "arvix_search": arvix_search,
-    "get_youtube_transcript": get_youtube_transcript,
-    "extract_video_id": extract_video_id,
-    "analyze_attachment": analyze_attachment,
-    "wikidata_query": wikidata_query
-    
-}
+enabled_tool_names = [
+    "math",
+    "wiki_search",
+    "web_search",
+    "arvix_search",
+    "get_youtube_transcript",
+    "extract_video_id",
+    "analyze_attachment",
+    "wikidata_query"
+]
 
 tools = [tool_map[name] for name in enabled_tool_names]
 
+tools = []
+for name in enabled_tool_names:
+    if name in tool_map:
+        tools.append(tool_map[name])
+    else:
+        print(f"⚠️ Warning: Tool '{name}' not found in tool_map. Skipping.")
 
 # -------------------------------
 # Step 2: Load the JSON file or tasks (Replace this part if you're loading tasks dynamically)