0.1.34

PyPI/src/guan.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: guan
-Version: 0.1.33
+Version: 0.1.34
 Summary: An open source python package
 Home-page: https://py.guanjihuan.com
 Author: guanjihuan

PyPI/src/guan/data_processing.py

@@ -401,20 +401,23 @@ def chat(prompt='你好', stream_show=1, top_p=0.8, temperature=0.8):
         send_message = json.dumps(message)
         client_socket.send(send_message.encode())
         while True:
-            data = client_socket.recv(1024)
-            stream_response = data.decode()
-            response_dict = json.loads(stream_response)
-            stream_response = response_dict['response']
-            end_message = response_dict['end_message']
-            if end_message == 1:
-                break
-            elif stream_response == '':
-                break
-            else:
-                if stream_show == 1:
-                    print(stream_response)
-                    print('\n---\n')
-                response = stream_response
+            try:
+                data = client_socket.recv(1024)
+                stream_response = data.decode()
+                response_dict = json.loads(stream_response)
+                stream_response = response_dict['response']
+                end_message = response_dict['end_message']
+                if end_message == 1:
+                    break
+                elif stream_response == '':
+                    break
+                else:
+                    if stream_show == 1:
+                        print(stream_response)
+                        print('\n---\n')
+                    response = stream_response
+            except:
+                pass
         client_socket.close()
     import guan
     guan.statistics_of_guan_package()