olivebot/utils/openai_api/openai_api.py
guo zebin 4cfad5ae0f 年翻更新
- 全新ui
- 全面优化websocket逻辑,提高数字人和ui连接的稳定性及资源开销
- 全面优化唤醒逻辑,提供稳定的普通唤醒模式和前置词唤醒模式
- 优化拾音质量,支持多声道麦克风拾音
- 优化自动播放服务器的对接机制,提供稳定和兼容旧版ue工程的对接模式
- 数字人接口输出机器人表情,以适应新fay ui及单片机的数字人表情输出
- 使用更高级的音频时长计算方式,可以更精准控制音频播放完成后的逻辑
- 修复点击关闭按钮会导致程序退出的bug
- 修复没有麦克风的设备开启麦克风会出错的问题
- 为服务器主机地址提供配置项,以方便服务器部署
2024-10-26 11:34:55 +08:00

22 lines
767 B
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 os
import openai
# 设置OpenAI API的密钥
# openai.api_key = os.getenv("OPENAI_API_KEY")
openai.base_url = "http://127.0.0.1:8000/v1/chat/completions"
# 定义API请求的数据
data = {
"model": "chatglm3-6b",
"prompt": "Say this is a test",
"temperature": 0.5, # 控制输出结果的随机性范围从0.0到1.0,越高越随机
"max_tokens": 75, # 控制输出文本的长度
"top_p": 1, # 一个更复杂的参数与temperature类似但更加精细控制
"n": 1, # 要返回的最完整的文本段落数
"stream": False # 是否以流的形式返回输出
}
# 发送API请求
response = openai.Completion.create(**data)
# 打印响应结果
print(response.get("choices")[0]["text"])