diff --git a/beautyleg/beauty_leg.py b/beautyleg/beauty_leg.py deleted file mode 100644 index fc31e45..0000000 --- a/beautyleg/beauty_leg.py +++ /dev/null @@ -1,77 +0,0 @@ -import logging -import os -import random -import tomllib - -import requests -from wcferry import WxMsg, Wcf - -from robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager -import lz4.block as lb - - -class BeautyLeg: - def __init__(self, wcf: Wcf, gbm: GroupBotManager): - self.LOG = logging.getLogger(__name__) - self.wcf = wcf # 假设 wcf 对象在此类中初始化 - self.gbm = gbm # 权限功能 - with open("beautyleg/config.toml", "rb") as f: - plugin_config = tomllib.load(f) - - config = plugin_config["Beautyleg"] - - self.enable = config["enable"] - self.command = config["command"] - self.LOG.info(f"[美腿] 组件初始化完成，指令： {self.command}") - - def get_random_file_from_dir(self, directory): - image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'} - image_files = [] - - if not os.path.exists(directory): - print(f"Error: Directory '{directory}' does not exist.") - return None - - if not os.access(directory, os.R_OK): - print(f"Error: No read access to directory '{directory}'.") - return None - - print(f"Scanning directory: {directory} (including subdirectories)") - - # 使用 os.walk() 递归遍历所有子目录 - for root, _, files in os.walk(directory): - for file in files: - _, ext = os.path.splitext(file) - if ext.lower() in image_extensions: - full_path = os.path.join(root, file) - image_files.append(full_path) - - if not image_files: - print("No image files found in the directory (including subdirectories).") - return None - - return random.choice(image_files) - - def handle_message(self, message: WxMsg): - if not self.enable: - return - - content = str(message.content).strip() - command = content.split(" ") - - if command[0] not in self.command: - return - - # 如果触发了指令，但是没有权限，则返回权限不足 - if self.gbm.get_group_permission(message.roomid, Feature.BEAUTY_LEG) == PermissionStatus.DISABLED: - return - - # 使用示例 - directory = 'beautyleg/download_dir' # 替换为你的目录路径 - random_file_path = self.get_random_file_from_dir(directory) - random_file_path = os.path.abspath(random_file_path) - self.LOG.info(f"BeautyLeg.random_file_path: {random_file_path}") - if random_file_path: - return self.wcf.send_file(random_file_path, message.roomid) - else: - return None diff --git a/beautyleg/config.toml b/beautyleg/config.toml deleted file mode 100644 index 1cda299..0000000 --- a/beautyleg/config.toml +++ /dev/null @@ -1,3 +0,0 @@ -[Beautyleg] -enable = true -command = ["美腿", "腿来"] diff --git a/message_summary/config.toml b/message_summary/config.toml deleted file mode 100644 index 8898e09..0000000 --- a/message_summary/config.toml +++ /dev/null @@ -1,18 +0,0 @@ -[Summmary] -enable = true - -api-key = "app-McGLzBhBjeBCSEi7n83MtuTo" # Dify的API Key -base-url = "http://192.168.2.240/v1" #Dify API接口base url - -commands = ["深度总结", "群聊总结"] -command-tip = """-----Bot----- -💬AI聊天指令： -深度总结 -""" - -price = 0 # 用一次扣积分，如果0则不扣 - -# Http代理设置 -# 格式: http://用户名:密码@代理地址:代理端口 -# 例如：http://127.0.0.1:7890 -http-proxy = "" \ No newline at end of file diff --git a/message_summary/message_summary_4o.py b/message_summary/message_summary_4o.py deleted file mode 100644 index 85570e0..0000000 --- a/message_summary/message_summary_4o.py +++ /dev/null @@ -1,109 +0,0 @@ -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 = """ - # 🌟「[最新日期] 群聊总结」🌟 - - ## 📊 今日数据快报 - - **总消息数**：📩 [消息数] 条 - - **最活跃时段**：🔥 [时段] （📈 [消息数] 条/小时） - - **聊天时段**：🕒 [开始时间] - [结束时间] - - ## 🌌 话题总结 - - ### 1️⃣ 【[话题1]】 ⭐⭐⭐⭐⭐ - 🕒**聊天时段**：[开始时间] - [结束时间] - 🔍 **精彩回顾**： - > [精彩回顾内容提炼] - 🏅 **热点人物**：[@用户A] [@用户B] - 💬 **精彩语录**："[精选金句]" - - ### 2️⃣ 【[话题2]】 ⭐⭐⭐⭐ - 🕒**聊天时段**：[开始时间] - [结束时间] - 🔍 **高能回顾**： - > [精彩回顾内容提炼] - 📌 **实用内容**：[资源链接/知识点总结] - - ## 🎖️ 今日荣誉榜 - - 🏆 **群聊 MVP**：[@用户D] - 👑 **获奖理由**： - ✅ 发起 [话题数] 个热门话题 - ✅ 贡献 [数量] 个表情包/段子 - ✅ **创新贡献**："薯片袋静音开封法"（已申请专利 🎉） - - ✨ *本总结由 AI 自动生成，快来看看你今天是不是最靓的崽！🔥* - - """ - - 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) diff --git a/message_summary/message_summary_dify.py b/message_summary/message_summary_dify.py deleted file mode 100644 index 66e3ee8..0000000 --- a/message_summary/message_summary_dify.py +++ /dev/null @@ -1,94 +0,0 @@ -import requests -import json - -from message_summary.compress_chat_data import compress_chat_data -from message_summary.markdown_to_image import convert_md_str_to_image - - -def message_summary_dify(content, sender: str = None): - """ - 使用Dify API生成群聊消息总结 - - Args: - content: 需要总结的群聊消息内容 - - Returns: - 生成的总结内容和图片路径 - """ - # Dify API配置 - api_key = "app-McGLzBhBjeBCSEi7n83MtuTo" # 请替换为实际的API密钥 - url = "http://192.168.2.240/v1/chat-messages" - content_compress = content - try: - content_compress = compress_chat_data(content) - print(f"压缩内容成功：{len(content_compress)}--{len(content)}") - except Exception as e: - print(f"压缩内容失败：{e}") - - # 准备请求数据 - data = { - "inputs": {}, - "query": f"请根据以下{sender}群聊记录生成一份精华总结：\n\n{content_compress}", - "response_mode": "blocking", # 使用阻塞模式，直接获取完整响应 - "conversation_id": "", - "user": sender if sender is not None else "message_summary_bot", - "files": [] # 不包含文件 - } - - # 设置请求头 - headers = { - "Authorization": f"Bearer {api_key}", - "Content-Type": "application/json" - } - - try: - # 发送POST请求 - response = requests.post(url, headers=headers, json=data) - response.raise_for_status() # 检查请求是否成功 - - # 解析响应 - response_data = response.json() - print(f"Dify API响应状态码: {response.status_code}") - print(f"响应数据: {json.dumps(response_data, ensure_ascii=False, indent=2)}") - - # 提取回答内容 - answer = response_data.get("answer", "") - spath = "" - # 提取token使用情况 - metadata = response_data.get("metadata", {}) - usage = metadata.get("usage", {}) - - if usage: - prompt_tokens = usage.get("prompt_tokens", 0) - completion_tokens = usage.get("completion_tokens", 0) - total_tokens = usage.get("total_tokens", 0) - - # 添加token信息 - tokens_info = f"\n\n【tokens】输入: {prompt_tokens} 生成: {completion_tokens} 总: {total_tokens}" - answer += tokens_info - try: - spath = convert_md_str_to_image(answer, "output.png") - except Exception as e: - print(f"生成image失败:{e}") - # 返回文本内容和图片路径 - return answer, spath - - except requests.exceptions.RequestException as e: - print(f"请求Dify API时出错: {e}") - return f"生成总结时出错: {str(e)}", None - - except json.JSONDecodeError as e: - print(f"解析Dify API响应时出错: {e}") - return "解析API响应时出错", None - - except Exception as e: - print(f"处理总结时出现未知错误: {e}") - return f"生成总结时出现未知错误: {str(e)}", None - - -if __name__ == '__main__': - content = """ -2025-03-14 14:30:15,Jyunere,别这样啊。 - """ - msg = compress_chat_data(content, 5) - print(f"{msg}\n {len(msg)},{len(content)}") diff --git a/message_summary/compress_chat_data.py b/plugins/message_summary/compress_chat_data.py similarity index 100% rename from message_summary/compress_chat_data.py rename to plugins/message_summary/compress_chat_data.py diff --git a/plugins/message_summary/main.py b/plugins/message_summary/main.py index a49e1fe..9f4e314 100644 --- a/plugins/message_summary/main.py +++ b/plugins/message_summary/main.py @@ -4,10 +4,10 @@ from typing import Dict, Any, Tuple, Optional, List import requests from message_storage.message_to_db import MessageStorage -from message_summary.compress_chat_data import compress_chat_data -from message_summary.markdown_to_image import convert_md_str_to_image +from compress_chat_data import compress_chat_data from plugin_common.message_plugin_interface import MessagePluginInterface from plugin_common.plugin_interface import PluginStatus +from plugins.message_summary.markdown_to_image import convert_md_str_to_image from plugins.stats_collector.decorators import plugin_stats_decorator from robot_cmd.robot_command import GroupBotManager, Feature, PermissionStatus diff --git a/message_summary/markdown_to_image.py b/plugins/message_summary/markdown_to_image.py similarity index 100% rename from message_summary/markdown_to_image.py rename to plugins/message_summary/markdown_to_image.py