This commit is contained in:
guanjihuan 2024-12-17 22:09:34 +08:00
parent da6c153571
commit 73fc56877b
3 changed files with 14 additions and 10 deletions

View File

@ -1,7 +1,7 @@
[metadata] [metadata]
# replace with your username: # replace with your username:
name = guan name = guan
version = 0.1.133 version = 0.1.134
author = guanjihuan author = guanjihuan
author_email = guanjihuan@163.com author_email = guanjihuan@163.com
description = An open source python package description = An open source python package

View File

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

View File

@ -68,6 +68,8 @@ def chat(prompt='你好', stream=1, model=1, top_p=0.8, temperature=0.85):
def run(function_name, *args, **kwargs): def run(function_name, *args, **kwargs):
import socket import socket
import json import json
import pickle
import base64
import time import time
import guan import guan
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket: with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
@ -102,23 +104,25 @@ def run(function_name, *args, **kwargs):
send_message = json.dumps(message) send_message = json.dumps(message)
client_socket.send(send_message.encode()) client_socket.send(send_message.encode())
time.sleep(0.15) time.sleep(0.15)
return_data = None return_data = ''
print_data = ''
while True: while True:
try: try:
data = client_socket.recv(1024) data = client_socket.recv(1024)
return_text = data.decode() return_text = data.decode()
return_dict = json.loads(return_text) return_dict = json.loads(return_text)
return_data = return_dict['return_data'] return_data += return_dict['return_data']
print_data = return_dict['print_data'] print_data += return_dict['print_data']
end_message = return_dict['end_message'] end_message = return_dict['end_message']
if end_message == 1:
break
except:
break
if print_data != '': if print_data != '':
print('--- Start Print ---\n') print('--- Start Print ---\n')
print(print_data) print(print_data)
print('--- End Print ---\n') print('--- End Print ---\n')
if end_message == 1 or return_text == '': return_data = pickle.loads(base64.b64decode(return_data))
break
except:
break
client_socket.close() client_socket.close()
return return_data return return_data