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

chore: Roll back json parsing code

Browse Source
This commit is contained in:
FangYin Cheng 2023-09-28 14:38:00 +08:00
parent a2108e51a7
commit 1c092538ef
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/scene/chat_dashboard/out_parser.py
View File

@ -21,9 +21,9 @@ class ChatDashboardOutputParser(BaseOutputParser):
super().__init__(sep=sep, is_stream_out=is_stream_out)
def parse_prompt_response(self, model_out_text):
# clean_str = super().parse_prompt_response(model_out_text)
print("clean prompt response:", model_out_text)
response = json.loads(model_out_text)
clean_str = super().parse_prompt_response(model_out_text)
print("clean prompt response:", clean_str)
response = json.loads(clean_str)
chart_items: List[ChartItem] = []
if not isinstance(response, list):
response = [response]