128 lines
4.3 KiB
Python
128 lines
4.3 KiB
Python
import requests
|
||
import json
|
||
|
||
|
||
def extract_content(data_string):
|
||
try:
|
||
data = json.loads(data_string)
|
||
|
||
# 提取content字段
|
||
content = data["choices"][0]["message"].get("content", "")
|
||
|
||
# 提取tokens相关内容，加入容错处理
|
||
tokens_usage = data.get("usage", {})
|
||
|
||
# 确保tokens_usage是字典类型
|
||
if isinstance(tokens_usage, dict):
|
||
prompt_tokens = tokens_usage.get("prompt_tokens", 0)
|
||
completion_tokens = tokens_usage.get("completion_tokens", 0)
|
||
total_tokens = tokens_usage.get("total_tokens", 0)
|
||
else:
|
||
prompt_tokens = completion_tokens = total_tokens = 0
|
||
|
||
# 如果tokens信息为空，提供默认值或提示
|
||
if prompt_tokens == 0 and completion_tokens == 0 and total_tokens == 0:
|
||
tokens_info = "\n\n【tokens】暂无数据"
|
||
else:
|
||
tokens_info = (f"\n\n【tokens】输入: {prompt_tokens} 生成: {completion_tokens} 总: {total_tokens}")
|
||
|
||
# 将tokens信息添加到content后面，返回为字符串
|
||
content_with_tokens = content + tokens_info
|
||
|
||
return content_with_tokens
|
||
|
||
except json.JSONDecodeError:
|
||
print("Invalid JSON")
|
||
return None
|
||
|
||
|
||
def message_summary(content):
|
||
# 设置Authorization和URL
|
||
|
||
authorization = "Bearer b8586595-eb81-483d-8e91-a35cc789729e" # 请替换为真实的Authorization token
|
||
url = 'https://ark.cn-beijing.volces.com/api/v3/chat/completions'
|
||
# 群聊精华总结生成指令
|
||
prompt = """
|
||
【风格要求】
|
||
📢 轻松幽默，融入网络流行语+Emoji（如👍/🎉等）。
|
||
|
||
【结构框架】
|
||
📌 标题（10字内，带Emoji）
|
||
📊 数据概览（消息总数/最活跃时段）
|
||
🕒 聊天时段（自动识别最新日期，显示完整时间范围）
|
||
🌌 话题概览（3-5个核心话题，自动聚类，过滤广告/无效内容）
|
||
🏆 今日MVP（贡献最高成员+获奖理由）
|
||
|
||
【话题评分维度】
|
||
🎯 参与度（参与人数/消息密度）
|
||
💡 信息量（实用知识/资源分享）
|
||
🎭 娱乐值（段子/梗图/神回复）
|
||
⚡ 热议指数（讨论时长）
|
||
📌 评分方式：基于话题热度动态生成⭐评分（1-5星）。
|
||
|
||
【贡献者识别】
|
||
🏅 标记话题发起者 + 气氛担当。
|
||
📊 统计优质内容贡献者（表情包/段子/资源等）。
|
||
🎖️ 自动生成趣味头衔（如"段子王"、"知识达人"）。
|
||
|
||
【输出模板】
|
||
|
||
「[群名]-[最新日期]群聊总结」
|
||
|
||
📊 今日数据快报：
|
||
- 总计 [消息数] 条
|
||
- 最活跃时段：[时段]（[消息数]条/小时）
|
||
- 聊天时段：[开始时间] - [结束时间]
|
||
|
||
🌌 话题总结：
|
||
1️⃣ 【[话题1]】 ⭐️⭐️⭐️⭐️⭐️
|
||
🔍 [3句话简要回顾]
|
||
🏅 热点人物：[@用户A] [@用户B]
|
||
💬 精彩语录："[精选金句]"...
|
||
|
||
2️⃣ 【[话题2]】 ⭐️⭐️⭐️⭐️
|
||
🔍 [3句话高能回顾]
|
||
📌 实用内容：[资源链接/知识点总结]...
|
||
|
||
🎖️ 今日荣誉榜：
|
||
🏆 群聊MVP：[@用户D]
|
||
👑 获奖理由：发起 [话题数] 个热门话题，贡献 [数量] 个表情包/段子，发明了“薯片袋静音开封法”已申请专利。
|
||
|
||
|
||
【参数调节】
|
||
⚙ 严谨度：调整评分权重，决定专业度 vs. 趣味性。
|
||
🔍 敏感词过滤：自动屏蔽政治/广告/违规内容。
|
||
🎭 个性化适配：根据群成员特点生成个性化头衔。
|
||
"""
|
||
|
||
data = {
|
||
# "stream": True,
|
||
"model": "doubao-1-5-lite-32k-250115",
|
||
"messages": [
|
||
{
|
||
"role": "system",
|
||
"content": f"{prompt}"
|
||
},
|
||
{
|
||
"role": "user",
|
||
"content": f"{content}"
|
||
}
|
||
|
||
]
|
||
}
|
||
|
||
# 设置请求头
|
||
headers = {
|
||
"Content-Type": "application/json; charset=utf-8",
|
||
"Authorization": authorization
|
||
}
|
||
|
||
# 发送POST请求
|
||
response = requests.post(url, headers=headers, data=json.dumps(data), )
|
||
response.encoding = 'utf-8'
|
||
|
||
# 输出响应内容
|
||
print(response.status_code)
|
||
print(response.text)
|
||
return extract_content(response.text)
|