fixes for tools

cerebrum/example/local_agent_load_demo.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 from cerebrum.manager.agent import AgentManager
 
-manager = AgentManager(base_url='https://my.aios.foundation')
+manager = AgentManager(base_url='https://app.aios.foundation')
 
 agent = manager.load_agent(local=True, path=f"/Users/rama2r/Cerebrum/example/agents/academic_agent")
 

cerebrum/example/local_tool_load_demo.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 from cerebrum.manager.tool import ToolManager
 
-manager = ToolManager(base_url='https://my.aios.foundation')
+manager = ToolManager(base_url='https://app.aios.foundation')
 
 tool = manager.load_tool(local=True, name='bing_search')
 

cerebrum/interface/__init__.py

@@ -5,7 +5,7 @@
 from .. import config
 
 class AutoAgent:
-    AGENT_MANAGER = AgentManager('https://my.aios.foundation')
+    AGENT_MANAGER = AgentManager('https://app.aios.foundation')
 
     @classmethod
     def from_preloaded(cls, agent_name: str):
@@ -28,11 +28,15 @@ def from_preloaded(cls, tool_name: str):
         n_slash = tool_name.count('/')
         if n_slash == 1: # load from author/name
         # if tool_name.split('/')[0] != 'core':
-            author, name, version = cls.TOOL_MANAGER.download_tool(
-                author=tool_name.split('/')[0],
-                name=tool_name.split('/')[1]
-            )
-            tool, _ = cls.TOOL_MANAGER.load_tool(author, name, version)
+            try:
+                author, name, version = cls.TOOL_MANAGER.download_tool(
+                    author=tool_name.split('/')[0],
+                    name=tool_name.split('/')[1]
+                )
+                tool, _ = cls.TOOL_MANAGER.load_tool(author, name, version)
+            except:
+                print('reload',tool_name.split('/')[1])
+                tool, _ = cls.TOOL_MANAGER.load_tool(local=True, name=tool_name.split('/')[1])
         else:
             tool, _ = cls.TOOL_MANAGER.load_tool(local=True, name=tool_name)
 
@@ -47,6 +51,7 @@ def from_batch_preload(cls, tool_names: list[str]):
         }
 
         for tool_name in tool_names:
+             print('tool name', tool_name)
              tool = AutoTool.from_preloaded(tool_name)
 
              response['tools'].append(tool.get_tool_call_format())

cerebrum/manager/agent.py

@@ -294,6 +294,6 @@ def load_agent(self,
 
 
 if __name__ == '__main__':
-    manager = AgentManager('https://my.aios.foundation/')
+    manager = AgentManager('https://app.aios.foundation/')
     agent = manager.download_agent('example', 'academic_agent')
     print(agent)

cerebrum/manager/tool.py

@@ -97,6 +97,8 @@ def load_tool(self,
                 local: bool = False):
         """Load a tool dynamically and return its class and configuration."""
 
+        print('local', local)
+
         if not local:
             if version is None:
                 cached_versions = self._get_cached_versions(author, name)

cerebrum/tool/core/registry.py

@@ -1,4 +1,35 @@
 
 PATHS = {
-
-} 
+    'currency_converter': {
+        'module_name': 'currency_converter.currency_converter',
+        'class_name': 'CurrencyConverter'
+    },
+    'file_downloader': {
+        'module_name': 'file.downloader',
+        'class_name': 'Downloader'
+    },
+    'google_search': {
+        'module_name': 'google.google_search',
+        'class_name': 'GoogleSearch'
+    },
+    'google_place': {
+        'module_name': 'google.google_places',
+        'class_name': 'GooglePlaces'
+    },
+    'imdb_top_movies': {
+        'module_name': 'imdb.top_movies',
+        'class_name': 'TopMovies'
+    },
+    'imdb_top_series': {
+        'module_name': 'imdb.top_series',
+        'class_name': 'TopSeries'
+    },
+    'wolfram_alpha': {
+        'module_name': 'wolfram.wolfram_alpha',
+        'class_name': 'WolframAlpha'
+    },
+    'wikipedia': {
+        'module_name': 'wikipedia.wikipedia',
+        'class_name': 'Wikipedia'
+    }
+}