This commit is contained in:
guanjihuan 2024-12-14 02:31:20 +08:00
parent 39665bf0cb
commit 2ee9172fef
3 changed files with 5 additions and 3 deletions

View File

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

View File

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

View File

@ -1,6 +1,6 @@
# Module: data_processing
# AI模型对话
# AI模型对话(输入长度不能太长,且回答速度有一定限制)
def chat(prompt='你好', stream=1, model=1, top_p=0.8, temperature=0.85):
import socket
import json
@ -20,6 +20,8 @@ def chat(prompt='你好', stream=1, model=1, top_p=0.8, temperature=0.85):
print('\n--- Begin Stream Message ---\n')
response = ''
while True:
if prompt == '':
break
try:
data = client_socket.recv(1024)
if data != b'':