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

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
Show Stats 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