github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-16 06:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Todo: add retry for generate

Browse Source
This commit is contained in:
csunny
2023-06-13 21:17:03 +08:00
parent 1f78c21c08
commit d839180d67
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/scene/base_chat.py
View File

@@ -136,6 +136,8 @@ class BaseChat(ABC):
return payload
def stream_call(self):
# TODO Retry when server connection error
payload = self.__call_base()
self.skip_echo_len = len(payload.get("prompt").replace("</s>", " ")) + 11