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

fix: Code standards (#1393)

Browse Source
This commit is contained in:
yyhhyy
2024-04-09 21:01:32 +08:00
committed by GitHub
parent 4fe10c0906
commit 0f2b46da62
4 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/core/interface/output_parser.py
View File

@@ -127,8 +127,8 @@ class BaseOutputParser(MapOperator[ModelOutput, Any], ABC):
ai_response = ai_response.replace("assistant:", "")
ai_response = ai_response.replace("Assistant:", "")
ai_response = ai_response.replace("ASSISTANT:", "")
ai_response = ai_response.replace("\_", "_")
ai_response = ai_response.replace("\*", "*")
ai_response = ai_response.replace("\\_", "_")
ai_response = ai_response.replace("\\*", "*")
ai_response = ai_response.replace("\t", "")
# ai_response = ai_response.strip().replace("\\n", " ").replace("\n", " ")
@@ -234,7 +234,7 @@ class BaseOutputParser(MapOperator[ModelOutput, Any], ABC):
.replace("\\n", " ")
.replace("\n", " ")
.replace("\\", " ")
.replace("\_", "_")
.replace("\\_", "_")
)
cleaned_output = self._illegal_json_ends(cleaned_output)
return cleaned_output