新增功能：群总结，总结时间8小时

2025-02-12 14:21:58 +08:00
parent 871dcf5138
commit 690b158965
4 changed files with 164 additions and 8 deletions
@@ -0,0 +1,39 @@
+import json
+
+# 模拟返回的API数据
+data_string = '''{
+    "id":"chatcmpl-1739339885",
+    "object":"chat.completion",
+    "created":1739339885,
+    "model":"windsurf",
+    "choices":[
+        {
+            "index":0,
+            "message":{
+                "role":"assistant",
+                "content":"[聊天主题] 婚姻与创业话题讨论  \n☆☆参与者：wxid_atv7kzgxv9lg21, wxid_j1f8su26egm222, wxid_nn630aasb1nb22, wxid_q1ugj6gbjj3o12, wxid_ne15uczjd2mi21, wxid_ayf2rispnhwi21  \n时间点：2025-02-12 12:59:14 - 2025-02-12 13:56:02  \n内容摘要： 本次聊天涵盖了多个话题，包括使用道具、杭州新兴科技公司、医疗手术、杜蕾斯品牌、创业公司、以及婚姻中性行为的问题。wxid_j1f8su26egm222提到杭州出现了新的科技公司，而wxid_ayf2rispnhwi21则提出了关于婚姻和性行为的疑问，引发了群内的讨论。  \n☑ 点评： 讨论内容较为分散，涉及多个话题。关于杭州科技公司的讨论显示出对当地创业环境的关注，而关于婚姻和性行为的问题则反映了个人生活中的实际困惑。这些话题虽然不直接相关，但都反映了群成员对生活不同方面的关注。  \n· 趣味互动 wxid_ayf2rispnhwi21关于婚姻和性行为的提问引发了群内的笑声和调侃，体现了群成员之间轻松的互动氛围。  \n· 待跟进事项 无具体行动项。未来可以考虑深入讨论某一特定话题，如创业或婚姻问题。  \n· 结语 今日的讨论涉及多个生活领域，展现了群成员对不同话题的兴趣和幽默感。期待未来能有更多深入的讨论，鼓励大家继续积极参与。  \n注： 本次讨论话题较为多样，未深入展开某一单一主题。"
+            },
+            "finish_reason":"stop"
+        }
+    ],
+    "usage":{
+        "completion_tokens":1024,
+        "prompt_tokens":0,
+        "total_tokens":1024
+    }
+}'''
+
+# 解析JSON
+def extract_content(data_string):
+    try:
+        data = json.loads(data_string)
+        # 提取content字段
+        content = data["choices"][0]["message"].get("content", "")
+        return content
+    except json.JSONDecodeError:
+        print("Invalid JSON")
+        return None
+
+# 获取并打印content内容
+content = extract_content(data_string)
+print(f"Extracted Content: {content}")
@@ -0,0 +1,64 @@
+import requests
+import json
+
+
+# 解析JSON
+def extract_content(data_string):
+    try:
+        data = json.loads(data_string)
+        # 提取content字段
+        content = data["choices"][0]["message"].get("content", "")
+        return content
+    except json.JSONDecodeError:
+        print("Invalid JSON")
+        return None
+
+
+def message_summary(content):
+    # 设置Authorization和URL
+    authorization = "46a5674a-e978-491b-a810-5d54605f2c36"  # 请替换为真实的Authorization token
+    url = 'http://127.0.0.1:8080/v1/chat/completions'
+
+    prompt = ('你是一个聊天记录总结助手，请使用如下格式进行聊天记录整理：'
+              '[聊天主题] ☆☆☆'
+              '参与者：[列出参与者名字]'
+              '时间点：[开始时间] - [结束时间]'
+              '内容摘要： [简要总结讨论的内容，包括讨论的主题、观点、信息等。]'
+              '☑ 点评： [对讨论内容的简要评价，结合实际情况分析其意义或影响。]'
+              '· 趣味互动'
+              '[列出群成员之间互动的有趣部分，例如个人经验、笑点或特别的观点分享等。]'
+              '· 待跟进事项'
+              '[列出具体的行动项或下一步的讨论计划，包括需要落实的具体任务。]'
+              '· 结语'
+              '[总结今日讨论的主题和氛围，简要表达对未来讨论的期待。鼓励群成员继续参与。]'
+              '注： [对本次讨论的特殊情况进行说明，例如讨论话题较简单或较复杂等。]')
+    # 设置请求的payload
+    data = {
+        # "stream": True,
+        "model": "windsurf/gpt4o",
+        "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)
+    return extract_content(response.text)