zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: enhance function parameter handling to support default values

Browse Source
This commit is contained in:
CaptainB 2025-08-13 12:44:39 +08:00
parent 5e99770999
commit 3830ee920d
1 changed files with 26 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
apps/common/utils/tool_code.py
View File

@ -101,29 +101,33 @@ except Exception as e:
imports.append(ast.unparse(node)) imports.append(ast.unparse(node))
elif isinstance(node, ast.FunctionDef): elif isinstance(node, ast.FunctionDef):
# 修改函数参数以包含 params 中的默认值 # 修改函数参数以包含 params 中的默认值
func_name = node.name arg_names = [arg.arg for arg in node.args.args]
if func_name in params:
func_params = params[func_name]
# 为函数参数设置默认值
for i, arg in enumerate(node.args.args):
arg_name = arg.arg
if arg_name in func_params:
# 创建默认值节点
default_value = func_params[arg_name]
if isinstance(default_value, str):
default_node = ast.Constant(value=default_value)
elif isinstance(default_value, (int, float, bool)):
default_node = ast.Constant(value=default_value)
else:
default_node = ast.Constant(value=str(default_value))
# 添加到defaults列表 # 为有默认值的参数添加默认值
if not hasattr(node.args, 'defaults') or node.args.defaults is None: defaults = []
node.args.defaults = [] for arg_name in arg_names:
# 确保defaults列表长度正确 if arg_name in params:
while len(node.args.defaults) < len(node.args.args): # 将参数值转换为 AST 节点
node.args.defaults.insert(0, None) default_value = params[arg_name]
node.args.defaults[i] = default_node if isinstance(default_value, str):
defaults.append(ast.Constant(value=default_value))
elif isinstance(default_value, (int, float, bool)):
defaults.append(ast.Constant(value=default_value))
elif default_value is None:
defaults.append(ast.Constant(value=None))
else:
# 对于复杂类型，使用字符串表示
defaults.append(ast.Constant(value=str(default_value)))
else:
defaults.append(None)
# 只保留非 None 的默认值（从右到左）
while defaults and defaults[-1] is None:
defaults.pop()
# 更新函数的默认参数
if defaults:
node.args.defaults = [d for d in defaults if d is not None]
func_code = ast.unparse(node) func_code = ast.unparse(node)
functions.append(f"@mcp.tool()\n{func_code}\n") functions.append(f"@mcp.tool()\n{func_code}\n")