github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-09 12:18:12 +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 21:03:21 +08:00
parent 206eb4732b
commit 30ecd9f005
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/scene/chat_execution/out_parser.py
View File

@ -21,6 +21,8 @@ class PluginChatOutputParser(BaseOutputParser):
def parse_prompt_response(self, model_out_text) -> T: def parse_prompt_response(self, model_out_text) -> T:
clean_json_str = super().parse_prompt_response(model_out_text) clean_json_str = super().parse_prompt_response(model_out_text)
print(clean_json_str) print(clean_json_str)
if not clean_json_str:
raise ValueError("model server response not have json!")
try: try:
response = json.loads(clean_json_str) response = json.loads(clean_json_str)
except Exception as e: except Exception as e: