feat: support custom tool upload file (#10796)

2024-11-20 13:26:42 +08:00 · 2024-11-20 13:26:42 +08:00 · bf4b6e5f80
commit bf4b6e5f80
parent 25fda7adc5
2 changed files with 13 additions and 3 deletions
--- a/api/core/tools/utils/parser.py
+++ b/api/core/tools/utils/parser.py
@ -161,6 +161,9 @@ class ApiBasedToolSchemaParser:
    def _get_tool_parameter_type(parameter: dict) -> ToolParameter.ToolParameterType:
        parameter = parameter or {}
        typ = None
+        if parameter.get("format") == "binary":
+            return ToolParameter.ToolParameterType.FILE
+
        if "type" in parameter:
            typ = parameter["type"]
        elif "schema" in parameter and "type" in parameter["schema"]: