fix the way to parse langchain tool's tool_arg['type']

modelscope · Jul 23, 2024 · 16c4796 · 16c4796
1 parent 92f6eb0
commit 16c4796
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modelscope_agent/tools/langchain_proxy_tool.py b/modelscope_agent/tools/langchain_proxy_tool.py
@@ -32,7 +32,8 @@ def parse_langchain_schema(self):
             tool_arg = deepcopy(arg)
             tool_arg['name'] = name
             tool_arg['required'] = True
-            tool_arg['type'] = arg['anyOf'][0].get('type', 'string')
+            if 'type' not in arg:
+                tool_arg['type'] = arg['anyOf'][0].get('type', 'string')
             tool_arg.pop('title')
             self.parameters.append(tool_arg)