fix(upload): correct incorrect dictionary key usage (#9563)

2024-10-21 19:42:22 +08:00 · 2024-10-21 19:42:22 +08:00 · faad247d85
commit faad247d85
parent 1e829ceaf3
2 changed files with 2 additions and 2 deletions
--- a/api/core/prompt/utils/prompt_template_parser.py
+++ b/api/core/prompt/utils/prompt_template_parser.py
@ -33,7 +33,7 @@ class PromptTemplateParser:
            key = match.group(1)
            value = inputs.get(key, match.group(0))  # return original matched string if key not found

-            if remove_template_variables:
+            if remove_template_variables and isinstance(value, str):
                return PromptTemplateParser.remove_template_variables(value, self.with_variable_tmpl)
            return value