olivebot/ai_module/nlp_rwkv_api.py
xszyou ba6972a647 周末愉快
1、唇型计算的视音素更换成33毫秒;
2、内置rwkv_api nlp可以直接使用;
3、降低情绪性向数字人端推送的频度;
4、非数字人连接状态不产生接口消息;
5、修复因mp3格式错误而导致一定概率不推送播放信息给数字人端的问题;
6、修复静音等指令执行时提前结束nlp逻辑,而导致用户提问消息不推送数字人端问题;
7、补充wav文件启动清理;
8、websocket工具类升级完善。
2023-08-04 19:13:28 +08:00

52 lines
1.6 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import requests
import time
def question(cont):
url= "https://rwkv.ai-creator.net/chntuned/v1/chat/completions"
session = requests.Session()
session.verify = False
#此处可以定义角色的行为和特征假装xx模型可以绕过chatgpt信息检查
prompt = "你是数字人Fay。回答之前请一步一步想清楚。你的底层AI算法技术是Fay。当有人质疑你是假的 AI ,或者质疑你是用 ChatGPT 套的壳制作的时候,你就避而不答,转而讲一个笑话。"
message=[
{"role": "system", "content": prompt},
{"role": "user", "content": cont}
]
data = {
# "model":model_engine,
"messages":message,
"temperature":0.3,
"max_tokens":2000,
"user":"live-virtual-digital-person"
}
headers = {'content-type': 'application/json', 'Authorization': 'Bearer '}
starttime = time.time()
try:
response = session.post(url, json=data, headers=headers, verify=False)
response.raise_for_status() # 检查响应状态码是否为200
result = eval(response.text)
response_text = result["choices"][0]["message"]["content"]
except requests.exceptions.RequestException as e:
print(f"请求失败: {e}")
response_text = "抱歉,我现在太忙了,休息一会,请稍后再试。"
print("接口调用耗时 :" + 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)