From c91c34f76b8ee8cb138f501eff7a27d3c1b3e289 Mon Sep 17 00:00:00 2001 From: guanjihuan Date: Fri, 7 Jun 2024 00:13:13 +0800 Subject: [PATCH] 0.1.106 --- PyPI/setup.cfg | 2 +- PyPI/src/guan.egg-info/PKG-INFO | 2 +- PyPI/src/guan/others.py | 21 +++++++++++---------- 3 files changed, 13 insertions(+), 12 deletions(-) diff --git a/PyPI/setup.cfg b/PyPI/setup.cfg index 68cce96..59f7c13 100644 --- a/PyPI/setup.cfg +++ b/PyPI/setup.cfg @@ -1,7 +1,7 @@ [metadata] # replace with your username: name = guan -version = 0.1.105 +version = 0.1.106 author = guanjihuan author_email = guanjihuan@163.com description = An open source python package diff --git a/PyPI/src/guan.egg-info/PKG-INFO b/PyPI/src/guan.egg-info/PKG-INFO index a36a632..79b2fed 100644 --- a/PyPI/src/guan.egg-info/PKG-INFO +++ b/PyPI/src/guan.egg-info/PKG-INFO @@ -1,6 +1,6 @@ Metadata-Version: 2.1 Name: guan -Version: 0.1.105 +Version: 0.1.106 Summary: An open source python package Home-page: https://py.guanjihuan.com Author: guanjihuan diff --git a/PyPI/src/guan/others.py b/PyPI/src/guan/others.py index 246f76e..c1d1c25 100644 --- a/PyPI/src/guan/others.py +++ b/PyPI/src/guan/others.py @@ -22,16 +22,17 @@ def chat(prompt='你好', model=1, stream=0, top_p=0.8, temperature=0.85): while True: try: data = client_socket.recv(1024) - response_data = data.decode() - response_dict = json.loads(response_data) - stream_response = response_dict['stream_response'] - response = response_dict['response'] - end_message = response_dict['end_message'] - if end_message == 1: - break - else: - if stream == 1: - print(stream_response) + if data != b'': + response_data = data.decode() + response_dict = json.loads(response_data) + stream_response = response_dict['stream_response'] + response += stream_response + end_message = response_dict['end_message'] + if end_message == 1: + break + else: + if stream == 1: + print(stream_response) except: break client_socket.close()