0.1.31

2023-11-16 03:44:48 +08:00
parent 6c2c155159
commit f0d0875101
3 changed files with 5 additions and 3 deletions
--- a/PyPI/src/guan/data_processing.py
+++ b/PyPI/src/guan/data_processing.py
@@ -411,7 +411,9 @@ def chat(prompt='你好', stream_show=1, top_p=0.8, temperature=0.8):
                    if '连接失败！请过段时间再试或者联系管理员。' in stream_response:
                        print('连接失败！请过段时间再试或者联系管理员。')
                        break
-                    if stream_response == '':
+                    elif 'End_response_from_chat.guanjihuan.com.' in stream_response:
+                        break
+                    elif stream_response == '':
                        break
                    else:
                        if stream_show == 1: