diff --git a/PyPI/setup.cfg b/PyPI/setup.cfg index 4f295c8..bd61053 100644 --- a/PyPI/setup.cfg +++ b/PyPI/setup.cfg @@ -1,7 +1,7 @@ [metadata] # replace with your username: name = guan -version = 0.1.131 +version = 0.1.132 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 6d3effc..a5dc3e3 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.131 +Version: 0.1.132 Summary: An open source python package Home-page: https://py.guanjihuan.com Author: guanjihuan diff --git a/PyPI/src/guan/data_processing.py b/PyPI/src/guan/data_processing.py index 3d990c3..4f0b96d 100644 --- a/PyPI/src/guan/data_processing.py +++ b/PyPI/src/guan/data_processing.py @@ -1,21 +1,43 @@ # Module: data_processing -# AI模型对话(输入长度不能太长,且回答速度有一定限制) +# AI模型对话 def chat(prompt='你好', stream=1, model=1, top_p=0.8, temperature=0.85): import socket import json + import time + import guan with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket: client_socket.settimeout(30) client_socket.connect(('socket.guanjihuan.com', 12345)) - message = { - 'server': "chat.guanjihuan.com", - 'prompt': prompt, - 'model': model, - 'top_p': top_p, - 'temperature': temperature, - } - send_message = json.dumps(message) - client_socket.send(send_message.encode('utf-8')) + split_text_list = guan.split_text(prompt, wrap_width=100) + message_times = len(split_text_list) + if message_times == 1 or message_times == 0: + message = { + 'server': "chat.guanjihuan.com", + 'prompt': prompt, + 'model': model, + 'top_p': top_p, + 'temperature': temperature, + } + send_message = json.dumps(message) + client_socket.send(send_message.encode('utf-8')) + else: + end_message = 0 + for i0 in range(message_times): + if i0 == message_times-1: + end_message = 1 + prompt_0 = split_text_list[i0] + message = { + 'server': "chat.guanjihuan.com", + 'prompt': prompt_0, + 'model': model, + 'top_p': top_p, + 'temperature': temperature, + 'end_message': end_message, + } + send_message = json.dumps(message) + client_socket.send(send_message.encode('utf-8')) + time.sleep(0.15) if stream == 1: print('\n--- Begin Stream Message ---\n') response = ''