update pipeline test

modelscope · Jan 13, 2024 · 766051a · 766051a
1 parent 0b81656
commit 766051a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/modelscope_agent/tools/base.py b/modelscope_agent/tools/base.py
@@ -23,7 +23,10 @@ class BaseTool(ABC):
     parameters: List[Dict]
 
     def __init__(self, cfg: Optional[Dict] = {}):
-        self.cfg = cfg or {}
+        """
+        :param schema: Format of tools, default to oai format, in case there is a need for other formats
+        """
+        self.cfg = cfg.get(self.name, {})
 
         self.schema = self.cfg.get('schema', 'oai')
         self.function = self._build_function()