Add community tools

community_tools.py

@@ -0,0 +1,17 @@
+from langchain_community.tools import GooglePlacesTool
+from langchain.agents import load_tools
+from smolagents.tools import Tool 
+
+import os
+os.environ["GPLACES_API_KEY"] = "AIzaSyBpYITz-7liGXiXF-bpkjMFsmDoyRp0YXc"
+
+google_map_tool = Tool.from_langchain(GooglePlacesTool())
+
+from langchain_community.tools import WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
+
+wikipedia_tool = Tool.from_langchain(WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper()))
+
+arxiv_tools = [Tool.from_langchain(tool) for tool in load_tools(["arxiv"])]
+
+community_tools = [google_map_tool, wikipedia_tool, *arxiv_tools]

mini_agents.py

@@ -3,6 +3,7 @@ from tools import sort_list, operate_two_numbers, convert_number
 from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
 from vlm_tools import download_image, image_processing, object_detection, ocr_scan
 from audio_tools import audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization
+from community_tools import community_tools
 import os
 
 MODEL_CHOICES = {
@@ -93,7 +94,7 @@ operation_manager = CodeAgent(
 master_agent = CodeAgent(
     model=MODEL_CHOICES["master"][0],
     managed_agents=[multimodal_manager, operation_manager],
-    tools=[sort_list],
+    tools=[sort_list, *community_tools],
     max_steps=16,
     planning_interval=4,
     name="master_agent",