From d839180d67b144cf00b4b53d2cd27d8625428828 Mon Sep 17 00:00:00 2001
From: csunny <cfqsunny@163.com>
Date: Tue, 13 Jun 2023 21:17:03 +0800
Subject: [PATCH] Todo: add retry for generate

---
 pilot/scene/base_chat.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pilot/scene/base_chat.py b/pilot/scene/base_chat.py
index c1d831d0d..1eff3312c 100644
--- a/pilot/scene/base_chat.py
+++ b/pilot/scene/base_chat.py
@@ -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