Update agent.py

agent.py

@@ -254,6 +254,18 @@ 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
+}
+
+# Then define which tools you want enabled
 enabled_tool_names = [
     "math",
     "wiki_search",
@@ -265,6 +277,7 @@ enabled_tool_names = [
     "wikidata_query"
 ]
 
+
 tools = [tool_map[name] for name in enabled_tool_names]
 
 tools = []