github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-20 00:54:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ChatExcel): ChatExcel OutParse Bug Fix

Browse Source 
1.ChatExcel OutParse Bug Fix 2
This commit is contained in:
yhjun1026
2023-09-15 18:12:16 +08:00
parent 0755200ca0
commit 95badc5cdf
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/out_parser/base.py
View File

@@ -207,7 +207,7 @@ class BaseOutputParser(ABC):
cleaned_output.strip() cleaned_output.strip()
.replace("\\n", " ") .replace("\\n", " ")
.replace("\n", " ") .replace("\n", " ")
# .replace("\\", " ") .replace("\\", " ")
) )
cleaned_output = self.__illegal_json_ends(cleaned_output) cleaned_output = self.__illegal_json_ends(cleaned_output)
return cleaned_output return cleaned_output

2
pilot/scene/chat_db/auto_execute/out_parser.py
View File

@@ -27,7 +27,7 @@ class DbChatOutputParser(BaseOutputParser):
response = json.loads(clean_str) response = json.loads(clean_str)
for key in sorted(response): for key in sorted(response):
if key.strip() == "sql": if key.strip() == "sql":
sql = response[key].replace("\\", " ") sql = response[key]
if key.strip() == "thoughts": if key.strip() == "thoughts":
thoughts = response[key] thoughts = response[key]
return SqlAction(sql, thoughts) return SqlAction(sql, thoughts)