github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-22 10:08:34 +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 8876bd2a7a
commit 9b8adc5d6c
1 changed files with 1 additions and 1 deletions
Show Stats 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()