From 95badc5cdf574499e2a61eb163c4092b7f6e0d12 Mon Sep 17 00:00:00 2001 From: yhjun1026 <460342015@qq.com> Date: Fri, 15 Sep 2023 18:12:16 +0800 Subject: [PATCH] fix(ChatExcel): ChatExcel OutParse Bug Fix 1.ChatExcel OutParse Bug Fix 2 --- pilot/out_parser/base.py | 2 +- pilot/scene/chat_db/auto_execute/out_parser.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pilot/out_parser/base.py b/pilot/out_parser/base.py index 1d6ab03be..abd4d304b 100644 --- a/pilot/out_parser/base.py +++ b/pilot/out_parser/base.py @@ -207,7 +207,7 @@ class BaseOutputParser(ABC): cleaned_output.strip() .replace("\\n", " ") .replace("\n", " ") - # .replace("\\", " ") + .replace("\\", " ") ) cleaned_output = self.__illegal_json_ends(cleaned_output) return cleaned_output diff --git a/pilot/scene/chat_db/auto_execute/out_parser.py b/pilot/scene/chat_db/auto_execute/out_parser.py index 1f36fe2ce..8a79bca33 100644 --- a/pilot/scene/chat_db/auto_execute/out_parser.py +++ b/pilot/scene/chat_db/auto_execute/out_parser.py @@ -27,7 +27,7 @@ class DbChatOutputParser(BaseOutputParser): response = json.loads(clean_str) for key in sorted(response): if key.strip() == "sql": - sql = response[key].replace("\\", " ") + sql = response[key] if key.strip() == "thoughts": thoughts = response[key] return SqlAction(sql, thoughts)