diff options
| author | sigoden <sigoden@gmail.com> | 2024-11-04 07:27:47 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-04 07:27:47 +0800 |
| commit | afbd03f963b08419a3835c9473d757bf936c8b92 (patch) | |
| tree | dfe8e09eaa15ab2a58b6b46ccaf7defa7027f1d2 /scripts/build-declarations.py | |
| parent | 2cac7d90c87c27534d69a9ac76baf7c528246713 (diff) | |
| download | llm-functions-docker-afbd03f963b08419a3835c9473d757bf936c8b92.tar.gz | |
fix(scripts): ignore declarations without a description (#122)
Diffstat (limited to 'scripts/build-declarations.py')
| -rwxr-xr-x | scripts/build-declarations.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/scripts/build-declarations.py b/scripts/build-declarations.py index 72b6aff..3b70e8d 100755 --- a/scripts/build-declarations.py +++ b/scripts/build-declarations.py @@ -22,6 +22,8 @@ def main(is_tool=True): for function in functions: func_name, docstring, func_args = function description, params = parse_docstring(docstring) + if not description: + continue declarations.append( build_declaration(func_name, description, params, func_args) ) @@ -42,10 +44,10 @@ def extract_functions(contents: str, is_tool: bool): if not isinstance(node, ast.FunctionDef): continue func_name = node.name - if func_name.startswith("_"): - continue if is_tool and func_name != TOOL_ENTRY_FUNC: continue + if func_name.startswith("_"): + continue docstring = ast.get_docstring(node) or "" func_args = OrderedDict() for arg in node.args.args: |
