fix tool import bugs (#438)

modelscope · May 15, 2024 · 9d6f9ac · 9d6f9ac
1 parent 3effcd8
commit 9d6f9ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/modelscope_agent/agent.py b/modelscope_agent/agent.py
@@ -124,7 +124,7 @@ def _register_tool(self,
 
             # check if the tenant_id of tool instance or tool service are exists
             # TODO: change from use_tool_api=True to False, to get the tenant_id of the tool changes to
-            if tenant_id in tool_class_with_tenant:
+            if tenant_id in tool_class_with_tenant and self.use_tool_api:
                 return
 
             try:

diff --git a/modelscope_agent/tools/base.py b/modelscope_agent/tools/base.py
@@ -118,9 +118,7 @@ def __contains__(self, key):
 def register_tool(name):
 
     def decorator(cls):
-        if name not in TOOL_REGISTRY:
-            TOOL_REGISTRY[name] = {}
-        TOOL_REGISTRY[name]['class'] = cls
+        TOOL_REGISTRY[name] = {'class': cls}
         return cls
 
     return decorator