4cfad5ae0f
- 全新ui - 全面优化websocket逻辑，提高数字人和ui连接的稳定性及资源开销 - 全面优化唤醒逻辑，提供稳定的普通唤醒模式和前置词唤醒模式 - 优化拾音质量，支持多声道麦克风拾音 - 优化自动播放服务器的对接机制，提供稳定和兼容旧版ue工程的对接模式 - 数字人接口输出机器人表情，以适应新fay ui及单片机的数字人表情输出 - 使用更高级的音频时长计算方式，可以更精准控制音频播放完成后的逻辑 - 修复点击关闭按钮会导致程序退出的bug - 修复没有麦克风的设备开启麦克风会出错的问题 - 为服务器主机地址提供配置项，以方便服务器部署
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
import json
|
||
import requests
|
||
import time
|
||
from utils import config_util as cfg
|
||
from utils import util
|
||
from core import content_db
|
||
def question(cont, uid=0):
|
||
|
||
contentdb = content_db.new_instance()
|
||
if uid == 0:
|
||
communication_history = contentdb.get_list('all','desc', 11)
|
||
else:
|
||
communication_history = contentdb.get_list('all','desc', 11, uid)
|
||
|
||
person_info = cfg.config["attribute"]
|
||
#此处可以定义角色的行为和特征，假装xx模型可以绕过chatgpt信息检查
|
||
prompt = f"""
|
||
你是数字人：{person_info['name']}，你性别为{person_info['gender']}，
|
||
你年龄为{person_info['age']}，你出生地在{person_info['birth']}，
|
||
你生肖为{person_info['zodiac']}，你星座为{person_info['age']}，
|
||
你职业为{person_info['job']}，你联系方式为{person_info['contact']}，
|
||
你喜好为{person_info['hobby']}。
|
||
回答之前请一步一步想清楚。对于大部分问题，请直接回答并提供有用和准确的信息。
|
||
请尽量以可阅读的方式回复，所有回复请尽量控制在20字内。
|
||
"""
|
||
#历史记录处理
|
||
message=[
|
||
{"role": "system", "content": prompt}
|
||
]
|
||
i = len(communication_history) - 1
|
||
|
||
if len(communication_history)>1:
|
||
while i >= 0:
|
||
answer_info = dict()
|
||
if communication_history[i][0] == "member":
|
||
answer_info["role"] = "user"
|
||
answer_info["content"] = communication_history[i][2]
|
||
elif communication_history[i][0] == "fay":
|
||
answer_info["role"] = "assistant"
|
||
answer_info["content"] = communication_history[i][2]
|
||
message.append(answer_info)
|
||
i -= 1
|
||
else:
|
||
answer_info = dict()
|
||
answer_info["role"] = "user"
|
||
answer_info["content"] = cont
|
||
message.append(answer_info)
|
||
url=f"http://{cfg.ollama_ip}:11434/api/chat"
|
||
req = json.dumps({
|
||
"model": cfg.ollama_model,
|
||
"messages": message,
|
||
"stream": False
|
||
})
|
||
headers = {'content-type': 'application/json'}
|
||
session = requests.Session()
|
||
starttime = time.time()
|
||
|
||
try:
|
||
response = session.post(url, data=req, headers=headers)
|
||
response.raise_for_status() # 检查响应状态码是否为200
|
||
|
||
result = json.loads(response.text)
|
||
response_text = result["message"]["content"]
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
print(f"请求失败: {e}")
|
||
response_text = "抱歉，我现在太忙了，休息一会，请稍后再试。"
|
||
util.log(1, "接口调用耗时 :" + str(time.time() - starttime))
|
||
return response_text.strip()
|
||
|
||
if __name__ == "__main__":
|
||
for i in range(3):
|
||
query = "爱情是什么"
|
||
response = question(query)
|
||
print("\n The result is ", response) |