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助手体验！
256 lines
9.7 KiB
Python
Executable File
256 lines
9.7 KiB
Python
Executable File
# coding:utf-8
|
|
import json
|
|
import threading
|
|
|
|
from volcengine.ApiInfo import ApiInfo
|
|
from volcengine.Credentials import Credentials
|
|
from volcengine.base.Service import Service
|
|
from volcengine.ServiceInfoHttps import ServiceInfoSms
|
|
from volcengine.const.Const import *
|
|
from retry import retry
|
|
|
|
|
|
class SmsService(Service):
|
|
_instance_lock = threading.Lock()
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
if not hasattr(SmsService, "_instance"):
|
|
with SmsService._instance_lock:
|
|
if not hasattr(SmsService, "_instance"):
|
|
SmsService._instance = object.__new__(cls)
|
|
return SmsService._instance
|
|
|
|
def __init__(self):
|
|
self.service_info = SmsService.get_service_info(self, REGION_CN_NORTH1)
|
|
self.api_info = SmsService.get_api_info()
|
|
super(SmsService, self).__init__(self.service_info, self.api_info)
|
|
|
|
@staticmethod
|
|
def get_service_info(self, region):
|
|
service_info = ServiceInfoSms("sms.volcengineapi.com", {'Accept': 'application/json'},
|
|
Credentials('', '', 'volcSMS', region), 5, 5)
|
|
return service_info
|
|
|
|
@staticmethod
|
|
def get_api_info():
|
|
api_info = {
|
|
"SendSms": ApiInfo("POST", "/", {"Action": "SendSms", "Version": "2020-01-01"}, {}, {}),
|
|
"SendSmsVerifyCode": ApiInfo("POST", "/", {"Action": "SendSmsVerifyCode", "Version": "2020-01-01"}, {}, {}),
|
|
"CheckSmsVerifyCode": ApiInfo("POST", "/", {"Action": "CheckSmsVerifyCode", "Version": "2020-01-01"}, {},
|
|
{}),
|
|
"SendBatchSms": ApiInfo("POST", "/", {"Action": "SendBatchSms", "Version": "2021-01-01"}, {}, {}),
|
|
"Conversion": ApiInfo("POST", "/", {"Action": "Conversion", "Version": "2020-01-01"}, {}, {}),
|
|
"GetSmsTemplateAndOrderList": ApiInfo("GET", "/",
|
|
{"Action": "GetSmsTemplateAndOrderList", "Version": "2021-01-11"}, {},
|
|
{}),
|
|
"ApplySmsTemplate": ApiInfo("POST", "/", {"Action": "ApplySmsTemplate", "Version": "2021-01-11"}, {}, {}),
|
|
"DeleteSmsTemplate": ApiInfo("POST", "/", {"Action": "DeleteSmsTemplate", "Version": "2021-01-11"}, {}, {}),
|
|
"GetSubAccountList": ApiInfo("GET", "/", {"Action": "GetSubAccountList", "Version": "2021-01-11"}, {}, {}),
|
|
"GetSubAccountDetail": ApiInfo("GET", "/", {"Action": "GetSubAccountDetail", "Version": "2021-01-11"}, {},
|
|
{}),
|
|
"GetSignatureAndOrderList": ApiInfo("GET", "/",
|
|
{"Action": "GetSignatureAndOrderList", "Version": "2021-01-11"}, {},
|
|
{}),
|
|
"ApplySmsSignature": ApiInfo("POST", "/", {"Action": "ApplySmsSignature", "Version": "2021-01-11"}, {}, {}),
|
|
"DeleteSignature": ApiInfo("POST", "/", {"Action": "DeleteSignature", "Version": "2021-01-11"}, {},
|
|
{}),
|
|
"ApplyVmsTemplate": ApiInfo("POST", "/", {"Action": "ApplyVmsTemplate", "Version": "2021-01-11"}, {},
|
|
{}),
|
|
"GetVmsTemplateStatus": ApiInfo("POST", "/", {"Action": "GetVmsTemplateStatus", "Version": "2021-01-11"},
|
|
{},
|
|
{}),
|
|
"InsertSubAccount": ApiInfo("POST", "/", {"Action": "InsertSubAccount", "Version": "2021-01-11"},
|
|
{},
|
|
{}),
|
|
"GetSmsSendDetails": ApiInfo("POST","/", {"Action": "GetSmsSendDetails", "Version": "2021-01-11"}, {}, {}),
|
|
"ApplySignatureIdent": ApiInfo("POST","/", {"Action": "ApplySignatureIdent", "Version": "2021-01-11"}, {}, {}),
|
|
"GetSignatureIdentList": ApiInfo("POST","/", {"Action": "GetSignatureIdentList", "Version": "2021-01-11"}, {}, {}),
|
|
"BatchBindSignatureIdent": ApiInfo("POST","/", {"Action": "BatchBindSignatureIdent", "Version": "2021-01-11"}, {}, {})
|
|
}
|
|
return api_info
|
|
|
|
@retry(tries=2, delay=0)
|
|
def send_sms(self, body):
|
|
res = self.json('SendSms', {}, body)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def send_sms_verify_code(self, body):
|
|
res = self.json('SendSmsVerifyCode', {}, body)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def check_sms_verify_code(self, body):
|
|
res = self.json('CheckSmsVerifyCode', {}, body)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def send_batch_sms(self, body):
|
|
res = self.json('SendBatchSms', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def conversion(self, body):
|
|
res = self.json('Conversion', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_sms_template_and_order_list(self, param):
|
|
res = self.get('GetSmsTemplateAndOrderList', param)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def apply_sms_template(self, body):
|
|
res = self.json('ApplySmsTemplate', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def delete_sms_template(self, body):
|
|
res = self.post('DeleteSmsTemplate', {}, body)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_sub_account_list(self, param):
|
|
res = self.get('GetSubAccountList', param)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_sub_account_detail(self, param):
|
|
res = self.get('GetSubAccountDetail', param)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_signature_and_order_list(self, param):
|
|
res = self.get('GetSignatureAndOrderList', param)
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def apply_sms_signature(self, body):
|
|
res = self.json('ApplySmsSignature', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def delete_signature(self, body):
|
|
res = self.json('DeleteSignature', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def apply_vms_template(self, body):
|
|
body["caller"] = "sdk"
|
|
if body["channelType"] is None or body["channelType"] == "":
|
|
body["channelType"] = "CN_VMS"
|
|
res = self.json('ApplyVmsTemplate', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
if 'Result' in res_json.keys() and 'templateId' in res_json['Result'].keys():
|
|
res_json['Result'] = {
|
|
"templateId": res_json['Result']["templateId"]}
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_vms_template_status(self, body):
|
|
res = self.json('GetVmsTemplateStatus', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def send_vms(self, body):
|
|
res = self.json('SendSms', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def insert_sms_sub_account(self, body):
|
|
res = self.json('InsertSubAccount', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
res_json = json.loads(res)
|
|
|
|
return res_json
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_sms_send_details(self, body):
|
|
res = self.json('GetSmsSendDetails', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
return json.loads(res)
|
|
|
|
@retry(tries=2, delay=0)
|
|
def apply_signature_ident(self, body):
|
|
res = self.json('ApplySignatureIdent', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
return json.loads(res)
|
|
|
|
@retry(tries=2, delay=0)
|
|
def get_signature_ident_list(self, body):
|
|
res = self.json('GetSignatureIdentList', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
return json.loads(res)
|
|
|
|
@retry(tries=2, delay=0)
|
|
def batch_bind_signature_ident(self, body):
|
|
res = self.json('BatchBindSignatureIdent', {}, json.dumps(body))
|
|
if res == '':
|
|
raise Exception("empty response")
|
|
return json.loads(res) |