github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-31 08:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix plugin mode bug；Optimize the parsing logic for model response

Browse Source
This commit is contained in:
yhjun1026
2023-06-13 22:38:10 +08:00
parent 30ecd9f005
commit 449515fe13
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/out_parser/base.py
View File

@@ -155,7 +155,7 @@ class BaseOutputParser(ABC):
cleaned_output = cleaned_output[: -len("```")]
cleaned_output = cleaned_output.strip()
if not cleaned_output.startswith("{") or not cleaned_output.endswith("}"):
logger.info("illegal json processing")
logger.info("illegal json processing:\n" + cleaned_output)
cleaned_output = self.__extract_json(cleaned_output)
cleaned_output = (
cleaned_output.strip()