Add default values for optional parameters in API tool and parser (#2491)

2024-02-20 16:07:43 +08:00 · 2024-02-20 16:07:43 +08:00 · 25957d917a
commit 25957d917a
parent 20b932da97
2 changed files with 9 additions and 1 deletions
--- a/api/core/tools/utils/parser.py
+++ b/api/core/tools/utils/parser.py
@ -60,7 +60,7 @@ class ApiBasedToolSchemaParser:
                        required=parameter.get('required', False),
                        form=ToolParameter.ToolParameterForm.LLM,
                        llm_description=parameter.get('description'),
-                        default=parameter['default'] if 'default' in parameter else None,
+                        default=parameter['schema']['default'] if 'schema' in parameter and 'default' in parameter['schema'] else None,
                    ))
            # create tool bundle
            # check if there is a request body