d0dc94e36d
新功能: - 创建AI聊天助手组件（AIChatWidget） - 支持文字+图片多模态对话 - 集成Gemini AI API - 支持图片编辑和内容生成 - 实时对话历史管理 - 优雅的UI设计，支持黑夜模式 技术实现: - 后端API代理Gemini API（/api/ai-chat） - 支持最多5张图片上传 - 流式响应处理 - 对话历史管理 - Base64图片处理 配置: - 在.env中添加Gemini API配置 - GEMINI_API_KEY - GEMINI_API_BASE_URL - GEMINI_MODEL 替换原弹幕功能，提供更强大的AI助手体验！
57 lines
1.3 KiB
Python
Executable File
57 lines
1.3 KiB
Python
Executable File
import os
|
|
from volcengine.maas import MaasService, MaasException, ChatRole
|
|
|
|
|
|
def test_chat(maas, req):
|
|
try:
|
|
resp = maas.chat(req)
|
|
print(resp)
|
|
print(resp.choice.message.content)
|
|
except MaasException as e:
|
|
print(e)
|
|
|
|
|
|
def test_stream_chat(maas, req):
|
|
try:
|
|
resps = maas.stream_chat(req)
|
|
for resp in resps:
|
|
print(resp)
|
|
print(resp.choice.message.content)
|
|
except MaasException as e:
|
|
print(e)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
maas = MaasService('maas-api.ml-platform-cn-beijing.volces.com', 'cn-beijing')
|
|
|
|
maas.set_ak(os.getenv("VOLC_ACCESSKEY"))
|
|
maas.set_sk(os.getenv("VOLC_SECRETKEY"))
|
|
|
|
|
|
# document: "https://www.volcengine.com/docs/82379/1099475"
|
|
# chat
|
|
req = {
|
|
"model": {
|
|
"name": "${YOUR_MODEL_NAME}"
|
|
},
|
|
"parameters": {
|
|
"max_new_tokens": 2000,
|
|
"temperature": 0.8
|
|
},
|
|
"messages": [
|
|
{
|
|
"role": ChatRole.USER,
|
|
"content": "天为什么这么蓝？"
|
|
}, {
|
|
"role": ChatRole.ASSISTANT,
|
|
"content": "因为有你"
|
|
}, {
|
|
"role": ChatRole.USER,
|
|
"content": "花儿为什么这么香？"
|
|
},
|
|
]
|
|
}
|
|
|
|
test_chat(maas, req)
|
|
test_stream_chat(maas, req)
|