Upload 64 files

cogs/gemini.py

@@ -1,996 +1,976 @@
-from io import BytesIO
-from discord.ext import commands
-import discord
-from google import genai
-from google.genai import types
-import random
-from aiohttp import ClientSession
-from utils.decorator import auto_delete
-from utils.func import async_iter, async_do_thread
-from utils.color_printer import cpr
-from utils.config import config
-from utils.context_prompter import ContextPrompter
-from utils.logger import logger
-from datetime import datetime
-import PIL.Image
-from typing import TYPE_CHECKING
-if TYPE_CHECKING:
-    from cogs.agent_manager import AgentManager
-
-
-class Gemini(commands.Cog):
-    def __init__(
-        self,
-        bot: commands.Bot,
-        webhook: discord.Webhook,
-    ):
-        self.bot = bot
-        self.conversations = {}
-        self.apikeys = config.get("gemini_keys")
-        self.current_key = config.get("current_key")
-        self.num = len(self.apikeys)
-        
-        # 确保chat_channels中的键全部为字符串
-        self.update_chat_channels()
-        
-        self.config = config
-        self.context_length = 20
-        self.target_language = config.get("target_language")
-        
-        # 获取Gemini模型配置
-        self.gemini_models = config.get("gemini_models", {
-            "chat": "gemini-2.0-pro-exp-02-05",  # 默认聊天模型
-            "translate": "gemini-2.0-pro-exp-02-05"  # 默认翻译模型
-        })
-        
-        # 如果配置中没有gemini_models，写入默认配置
-        if not config.get("gemini_models"):
-            config.write("gemini_models", self.gemini_models)
-        
-        self.default_gemini_config = types.GenerateContentConfig(
-            system_instruction="",
-            top_k=55,
-            top_p=0.95,
-            temperature=1.3,
-            safety_settings=[
-                types.SafetySetting(
-                    category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
-                    threshold=types.HarmBlockThreshold.OFF,
-                ),
-                types.SafetySetting(
-                    category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-                    threshold=types.HarmBlockThreshold.OFF,
-                ),
-                types.SafetySetting(
-                    category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
-                    threshold=types.HarmBlockThreshold.OFF,
-                ),
-                types.SafetySetting(
-                    category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-                    threshold=types.HarmBlockThreshold.OFF,
-                ),
-                types.SafetySetting(
-                    category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-                    threshold=types.HarmBlockThreshold.OFF,
-                ),
-            ],
-        )
-        self.webhook = webhook
-        self.context_prompter = ContextPrompter()
-        self.agent_manager: 'AgentManager' = None # 类型提示
-        self.non_gemini_model = None  # for openai model
-        self.openai_api_key = config.get("openai_api_key")
-        self.openai_endpoint = config.get("openai_endpoint")
-
-        if self.openai_api_key is not None and self.openai_endpoint is not None:
-            print(cpr.info("OpenAI API available."))
-
-    def set_agent_manager(self, agent_manager: 'AgentManager'):
-        """注入 AgentManager 实例"""
-        self.agent_manager = agent_manager
-        self.context_prompter.set_agent_manager(agent_manager)
-        print(cpr.info("AgentManager injected into Gemini cog."))
-        # 在这里可以根据 agent_manager 更新频道相关的配置，如果需要的话
-        # 例如，更新允许聊天的频道列表等
-        self.update_chat_channels() # 假设有这个方法来更新频道列表
-        
-    def update_chat_channels(self):
-        """更新聊天频道配置"""
-        # 获取所有服务器配置
-        self.servers = config.get("servers", {})
-        
-        # 更新Gemini模型配置
-        self.gemini_models = config.get("gemini_models", {
-            "chat": "gemini-2.0-pro-exp-02-05",
-            "translate": "gemini-2.0-pro-exp-02-05"
-        })
-        
-        # 获取可用的Gemini模型列表
-        self.available_models = config.get("gemini_available_models", [])
-        if not self.available_models:
-            # 设置默认模型列表
-            self.available_models = [
-                {"name": "gemini-2.0-pro-exp-02-05", "description": "默认聊天和翻译模型"},
-                {"name": "gemini-pro", "description": "旧版Gemini Pro模型"}
-            ]
-            config.write("gemini_available_models", self.available_models)
-        
-        print(f"Gemini cog 已更新服务器配置和模型列表")
-    
-    def get_channel_config(self, guild_id: str, channel_id: str):
-        """获取频道配置"""
-        server_name, server_config = config.get_server_config(guild_id)
-        if not server_config:
-            return None
-        return server_config.get("chat_channels", {}).get(channel_id)
-
-    def get_next_key(self):
-        self.current_key = (self.current_key + 1) % self.num
-        config.write("current_key", self.current_key)
-        return self.apikeys[self.current_key]
-
-    def get_random_key(self):
-        return self.apikeys[random.randint(0, self.num - 1)]
-
-    async def request_gemini(
-        self,
-        ctx: commands.Context,
-        prompt: str,
-        model_config: types.GenerateContentConfig = None,
-        model="gemini-2.0-pro-exp-02-05",
-        username=None,
-        extra_attachment: discord.Attachment = None,
-    ):
-        if model_config is None:
-            model_config = self.default_gemini_config
-        
-        # 获取服务器和频道配置
-        guild_id = str(ctx.guild.id)
-        channel_id = str(ctx.channel.id)
-        channel_config = self.get_channel_config(guild_id, channel_id)
-        
-        if not channel_config:
-            await ctx.send("此频道未配置为聊天频道", ephemeral=True)
-            return
-        
-        print(f"当前频道配置: {channel_config}")
-        
-        # 尝试获取预设
-        agent_manager = self.bot.get_cog("AgentManager")
-        preset_data = None
-        preset_name = "chat_preset.json"  # 默认使用chat_preset.json
-        
-        # 根据情况选择不同的预设
-        if extra_attachment:
-            preset_name = "attachment_preset.json"
-            print(f"使用附件预设: {preset_name}")
-        elif ctx.message.reference:
-            preset_name = "reference_preset.json"
-        
-        # 获取预设内容，传递频道ID和服务器ID以获取该频道对应的预设
-        if agent_manager:
-            preset_data = agent_manager.get_preset_json(preset_name, channel_id, guild_id)
-        
-        if model != "gemini-2.0-pro-exp-02-05":
-            key = self.get_random_key()
-        else:
-            key = self.get_next_key()
-        client = genai.Client(api_key=key)
-        
-        # 处理附件 - 下载附件内容
-        attachment_bytes = None
-        attachment_mime_type = None
-        if extra_attachment:
-            msg = await ctx.send("Downloading the attachment...")
-            bytes_data = await extra_attachment.read()
-            attachment_bytes = bytes_data
-            attachment_mime_type = extra_attachment.content_type.split(";")[0]
-            await msg.edit(content="Processing the attachment...")
-            print(f"附件已下载: {extra_attachment.filename} ({attachment_mime_type})")
-        else:
-            msg = await ctx.send("Typing...") if username is None else await self.webhook.send("Typing...", wait=True, username=username)
-        
-        # 检查预设数据是否存在
-        if not preset_data:
-            await msg.edit(content="无法加载预设数据，请联系管理员")
-            return
-            
-        # 首先获取变量替换所需的数据
-        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        bot_name = ctx.me.name
-        bot_display_name = ctx.me.display_name
-        user_name = ctx.author.name
-        user_display_name = ctx.author.display_name
-        
-        # 处理上下文
-        context = ""
-        if hasattr(ctx, 'context') and ctx.context:
-            context = ctx.context
-        elif hasattr(ctx, 'history') and ctx.history:
-            context = ctx.history
-        else:
-            # 获取历史消息作为上下文
-            context = await self.context_prompter.get_context_for_prompt(ctx, self.context_length)
-        
-        # 确保context是字符串
-        if not isinstance(context, str):
-            context = str(context) if context is not None else ""
-        
-        # 替换预设中的变量
-        first_user_message = preset_data.get("first_user_message", "")
-        first_user_message = first_user_message.replace("{context}", context)
-        first_user_message = first_user_message.replace("{question}", prompt)
-        first_user_message = first_user_message.replace("{name}", bot_display_name)
-        first_user_message = first_user_message.replace("{bot_name}", bot_name)
-        first_user_message = first_user_message.replace("{current_time}", current_time)
-        first_user_message = first_user_message.replace("{user_display_name}", user_display_name)
-        first_user_message = first_user_message.replace("{user_name}", user_name)
-        
-        main_content = preset_data.get("main_content", "")
-        main_content = main_content.replace("{context}", context)
-        main_content = main_content.replace("{question}", prompt)
-        main_content = main_content.replace("{name}", bot_display_name)
-        main_content = main_content.replace("{bot_name}", bot_name)
-        main_content = main_content.replace("{current_time}", current_time)
-        main_content = main_content.replace("{user_display_name}", user_display_name)
-        main_content = main_content.replace("{user_name}", user_name)
-        
-        # 如果是引用回复
-        if ctx.message.reference and 'reference' in preset_name:
-            reference = ctx.message.reference.resolved
-            reference_time = reference.created_at.strftime("%Y-%m-%d %H:%M:%S")
-            reference_user_name = reference.author.name
-            reference_user_display_name = reference.author.display_name
-            reference_content = reference.content
-            
-            main_content = main_content.replace("{reference_time}", reference_time)
-            main_content = main_content.replace("{reference_user_name}", reference_user_name)
-            main_content = main_content.replace("{reference_user_display_name}", reference_user_display_name)
-            main_content = main_content.replace("{reference_content}", reference_content)
-            first_user_message = first_user_message.replace("{reference_time}", reference_time)
-            first_user_message = first_user_message.replace("{reference_user_name}", reference_user_name)
-            first_user_message = first_user_message.replace("{reference_user_display_name}", reference_user_display_name)
-            first_user_message = first_user_message.replace("{reference_content}", reference_content)
-        
-        last_user_message = preset_data.get("last_user_message", "")
-        last_user_message = last_user_message.replace("{context}", context)
-        last_user_message = last_user_message.replace("{question}", prompt)
-        last_user_message = last_user_message.replace("{name}", bot_display_name)
-        last_user_message = last_user_message.replace("{bot_name}", bot_name)
-        last_user_message = last_user_message.replace("{current_time}", current_time)
-        last_user_message = last_user_message.replace("{user_display_name}", user_display_name)
-        last_user_message = last_user_message.replace("{user_name}", user_name)
-        
-        # 构建user-model-user的三个上下文
-        user_parts = [types.Part.from_text(text=first_user_message)]
-        model_parts = [types.Part.from_text(text=main_content)]
-        last_user_parts = [types.Part.from_text(text=last_user_message)]
-        
-        # 如果有附件，添加到最后一个用户消息中
-        if attachment_bytes:
-            # 使用Pillow和inline_data方式添加图片
-            image_bytes = BytesIO(attachment_bytes)
-            image = PIL.Image.open(image_bytes)
-            
-            # 转换为字节数据
-            mime_type = attachment_mime_type or "image/jpeg"
-            img_byte_arr = BytesIO()
-            image.save(img_byte_arr, format=image.format or "JPEG")
-            img_byte_data = img_byte_arr.getvalue()
-            
-            # 添加到消息中
-            last_user_parts.append(
-                types.Part(
-                    inline_data=types.Blob(
-                        mime_type=mime_type,
-                        data=img_byte_data
-                    )
-                )
-            )
-            print("附件已添加到用户消息中")
-        
-        contents = [
-            types.Content(
-                role="user",
-                parts=user_parts,
-            ),
-            types.Content(
-                role="model",
-                parts=model_parts,
-            ),
-            types.Content(
-                role="user",
-                parts=last_user_parts,
-            ),
-        ]
-        
-        # 设置安全设置
-        safety_settings = [
-            types.SafetySetting(
-                category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
-                threshold=types.HarmBlockThreshold.OFF,
-            ),
-            types.SafetySetting(
-                category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-                threshold=types.HarmBlockThreshold.OFF,
-            ),
-            types.SafetySetting(
-                category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
-                threshold=types.HarmBlockThreshold.OFF,
-            ),
-            types.SafetySetting(
-                category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-                threshold=types.HarmBlockThreshold.OFF,
-            ),
-            types.SafetySetting(
-                category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-                threshold=types.HarmBlockThreshold.OFF,
-            ),
-        ]
-        
-        # 获取Gemini配置
-        gemini_config_data = None
-        if agent_manager:
-            gemini_config_data = agent_manager.get_preset_json("gemini_config.json", channel_id, guild_id)
-        
-        # 构建配置
-        generate_content_config = types.GenerateContentConfig(
-            temperature=gemini_config_data.get("temperature", 1.0) if gemini_config_data else 1.0,
-            top_p=gemini_config_data.get("top_p", 0.95) if gemini_config_data else 0.95,
-            top_k=gemini_config_data.get("top_k", 64) if gemini_config_data else 64,
-            max_output_tokens=gemini_config_data.get("max_output_tokens", 8192) if gemini_config_data else 8192,
-            safety_settings=safety_settings,
-            response_mime_type="text/plain",
-            system_instruction=[
-                types.Part.from_text(text=preset_data.get("system_prompt", "")),
-            ],
-        )
-        
-        full = ""
-        n = config.get("gemini_chunk_per_edit")
-        every_n_chunk = 1
-        try:
-            # 记录请求内容
-            log_contents = []
-            for content in contents:
-                parts_text = []
-                for part in content.parts:
-                    if hasattr(part, "text") and part.text:
-                        parts_text.append(f"Text: {part.text}")
-                    elif hasattr(part, "file_data"):
-                        parts_text.append("File: [attachment]")
-                    else:
-                        parts_text.append(f"Unknown part type: {type(part)}")
-                log_contents.append(f"Role: {content.role}, Parts: {parts_text}")
-            
-            system_instruction = "None"
-            if hasattr(generate_content_config, "system_instruction"):
-                if generate_content_config.system_instruction:
-                    system_instruction = generate_content_config.system_instruction[0].text if generate_content_config.system_instruction else "None"
-            
-            # 只记录到日志文件，不再重复打印到控制台
-            logger.info(
-                "Gemini请求发送: 模型=%s, 内容=%s, 系统提示=%s, 配置=%s",
-                model,
-                log_contents,
-                system_instruction,
-                {
-                    "temperature": generate_content_config.temperature,
-                    "top_p": generate_content_config.top_p,
-                    "top_k": generate_content_config.top_k,
-                    "max_tokens": generate_content_config.max_output_tokens,
-                }
-            )
-            
-            # 使用流式响应
-            response = client.models.generate_content_stream(
-                model=model,
-                contents=contents,
-                config=self.default_gemini_config,  # 使用默认配置对象
-            )
-
-            async for chunk in async_iter(response):
-                if chunk.text:
-                    full += chunk.text
-                    if every_n_chunk == n:
-                        await msg.edit(content=full)
-                        every_n_chunk = 1
-                    else:
-                        every_n_chunk += 1
-            await msg.edit(content=full)
-        except Exception as e:
-            logger.error(
-                "Error when requesting gemini with key: %s, error: %s",
-                key,
-                e,
-                exc_info=True,
-            )
-            if full == "":
-                await msg.edit(content="Uh oh, something went wrong...")
-            else:
-                full += "\nUh oh, something went wrong..."
-                await msg.edit(content=full)
-
-    @commands.hybrid_command(name="hey", description="Ask a question to gemini.")
-    async def hey(
-        self,
-        ctx: commands.Context,
-        *,
-        question: str,
-        context_length: int = None,
-    ):
-        # 获取服务器和频道配置
-        guild_id = str(ctx.guild.id)
-        channel_id = str(ctx.channel.id)
-        channel_config = self.get_channel_config(guild_id, channel_id)
-        
-        if not channel_config:
-            await ctx.send("此频道未配置为聊天频道", ephemeral=True)
-            return
-            
-        if context_length is None:
-            context_length = self.context_length
-        extra_attachment = None
-        
-        # 获取历史消息作为上下文
-        history = await self.context_prompter.get_context_for_prompt(ctx, context_length)
-        ctx.history = history  # 将历史消息保存到ctx对象中，供预设处理使用
-        
-        # 检查附件
-        if ctx.message.reference:
-            reference = ctx.message.reference.resolved
-            # 优先查找引用消息中的附件
-            if reference and reference.attachments:
-                extra_attachment = reference.attachments[-1]
-        
-        # 选择合适的预设
-        agent_manager = self.bot.get_cog("AgentManager")
-        preset_name = "chat_preset.json"  # 默认使用chat_preset.json
-        
-        if agent_manager:
-            if ctx.message.reference:
-                reference = ctx.message.reference.resolved
-                if reference and reference.attachments:
-                    preset_name = "attachment_preset.json"
-                else:
-                    preset_name = "reference_preset.json"
-        
-        # 检查附件是否存在，确保传递正确
-        if extra_attachment:
-            print(f"处理附件: {extra_attachment.filename} ({extra_attachment.content_type})")
-        
-        # 使用聊天模型
-        chat_model = self.gemini_models.get("chat", "gemini-2.0-pro-exp-02-05")
-        
-        # 发送请求
-        await self.request_gemini(
-            ctx,
-            question,  # 直接传递原始问题，预设处理在request_gemini中完成
-            model=chat_model,
-            extra_attachment=extra_attachment,
-        )
-
-    @commands.hybrid_command(name="tr", description="Translate a text.")
-    async def translate(
-        self,
-        ctx: commands.Context,
-        target_language: str = None,
-        context_length: int = None,
-    ):
-        # 获取服务器和频道配置
-        guild_id = str(ctx.guild.id)
-        channel_id = str(ctx.channel.id)
-        channel_config = self.get_channel_config(guild_id, channel_id)
-        
-        if not channel_config:
-            await ctx.send("此频道未配置为聊天频道", ephemeral=True)
-            return
-            
-        if ctx.message.reference is None:
-            await ctx.send(
-                "请回复要翻译的消息", ephemeral=True
-            )
-            return
-        if context_length is None:
-            context_length = self.context_length
-        if target_language is None:
-            target_language = self.target_language
-
-        # 使用翻译模型
-        translate_model = self.gemini_models.get("translate", "gemini-2.0-pro-exp-02-05")
-
-        # 尝试获取翻译预设
-        agent_manager = self.bot.get_cog("AgentManager")
-        preset_data = None
-        
-        # 获取被回复的消息
-        reference_message = await ctx.channel.fetch_message(
-            ctx.message.reference.message_id
-        )
-        
-        # 检查是否有附件
-        extra_attachment = None
-        if reference_message and reference_message.attachments:
-            extra_attachment = reference_message.attachments[-1]
-            print(f"翻译附件: {extra_attachment.filename} ({extra_attachment.content_type})")
-        
-        # 下载附件内容
-        attachment_bytes = None
-        attachment_mime_type = None
-        if extra_attachment:
-            msg = await ctx.send("Downloading the attachment...")
-            bytes_data = await extra_attachment.read()
-            attachment_bytes = bytes_data
-            attachment_mime_type = extra_attachment.content_type.split(";")[0]
-            await msg.edit(content="Processing the attachment...")
-            print(f"附件已下载: {extra_attachment.filename} ({attachment_mime_type})")
-        else:
-            msg = await ctx.send("Translating...")
-        
-        # 获取预设内容
-        if agent_manager:
-            preset_data = agent_manager.get_preset_json("translate_preset.json")
-        
-        if preset_data:
-            # 使用预设JSON结构和原始文本
-            key = self.get_next_key()
-            client = genai.Client(api_key=key)
-            
-            # 获取变量替换所需的数据
-            current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-            bot_name = ctx.me.name
-            bot_display_name = ctx.me.display_name
-            user_name = ctx.author.name
-            user_display_name = ctx.author.display_name
-            
-            # 处理上下文和引用内容
-            context = ""
-            if hasattr(ctx, 'context') and ctx.context:
-                context = ctx.context
-            elif hasattr(ctx, 'history') and ctx.history:
-                context = ctx.history
-            else:
-                # 获取历史消息作为上下文
-                context = await self.context_prompter.get_context_for_prompt(ctx, context_length)
-            
-            # 确保context是字符串
-            if not isinstance(context, str):
-                context = str(context) if context is not None else ""
-            
-            # 获取被引用消息的相关信息
-            reference_time = reference_message.created_at.strftime("%Y-%m-%d %H:%M:%S")
-            reference_user_name = reference_message.author.name
-            reference_user_display_name = reference_message.author.display_name
-            reference_content = reference_message.content
-            
-            # 使用预设JSON结构
-            system_prompt = preset_data.get("system_prompt", "")
-            first_user_message = preset_data.get("first_user_message", "")
-            
-            # 确保模型已设置且处于传递中
-            translate_model = self.gemini_models.get("translate", "gemini-2.0-pro-exp-02-05")
-            
-            # 处理不同内容格式情况
-            if attachment_bytes and attachment_mime_type:
-                # 如果有图片附件
-                try:
-                    # 将字节转换为PIL图像以供Gemini处理
-                    if attachment_mime_type.startswith("image/"):
-                        image = PIL.Image.open(BytesIO(attachment_bytes))
-                        
-                        # 按照Gemini推荐尺寸调整图片，适应大多数模型版本
-                        if max(image.size) > 3072:
-                            ratio = 3072 / max(image.size)
-                            new_size = (int(image.size[0] * ratio), int(image.size[1] * ratio))
-                            image = image.resize(new_size, PIL.Image.LANCZOS)
-                        
-                        # 构建内容
-                        content = [
-                            {
-                                "role": "user",
-                                "parts": [
-                                    types.Part.from_text(first_user_message),
-                                    types.Part.from_image(image),
-                                ]
-                            }
-                        ]
-                        
-                        # 获取安全设置
-                        safety_settings = []
-                        try:
-                            gemini_config_data = None
-                            if agent_manager:
-                                gemini_config_data = agent_manager.get_preset_json("gemini_config.json")
-                                
-                            if gemini_config_data and "safety_settings" in gemini_config_data:
-                                safety_settings = self.build_safety_settings(gemini_config_data["safety_settings"])
-                        except Exception as e:
-                            logger.error(f"Error loading safety settings: {e}")
-                            safety_settings = []
-                        
-                        # 构建和应用生成内容配置
-                        generate_content_config = types.GenerateContentConfig(
-                            temperature=0.2,
-                            top_p=0.95,
-                            top_k=55,
-                            max_output_tokens=8192,
-                            safety_settings=safety_settings,
-                            response_mime_type="text/plain",
-                            system_instruction=[
-                                types.Part.from_text(text=system_prompt),
-                            ],
-                        )
-                        
-                        # 使用流式响应
-                        full = ""
-                        n = config.get("gemini_chunk_per_edit")
-                        every_n_chunk = 1
-                        
-                        try:
-                            response = client.models.generate_content_stream(
-                                model=translate_model,  # 使用翻译模型
-                                contents=content,
-                                config=generate_content_config,
-                            )
-                            
-                            async for chunk in async_iter(response):
-                                if chunk.text:
-                                    full += chunk.text
-                                    if every_n_chunk == n:
-                                        await msg.edit(content=full)
-                                        every_n_chunk = 1
-                                    else:
-                                        every_n_chunk += 1
-                            await msg.edit(content=full)
-                        except Exception as e:
-                            logger.error(
-                                "Error when translating with gemini, error: %s",
-                                e,
-                                exc_info=True,
-                            )
-                            if full == "":
-                                await msg.edit(content="Uh oh, something went wrong...")
-                            else:
-                                full += "\nUh oh, something went wrong..."
-                                await msg.edit(content=full)
-                except Exception as e:
-                    logger.error(f"Error processing image: {e}")
-                    await ctx.send("无法处理图片，请检查附件格式", ephemeral=True)
-            else:
-                # 替换预设中的变量
-                first_user_message = preset_data.get("first_user_message", "")
-                first_user_message = first_user_message.replace("{context}", context)
-                first_user_message = first_user_message.replace("{target_language}", target_language)
-                first_user_message = first_user_message.replace("{reference_content}", reference_content)
-                first_user_message = first_user_message.replace("{reference_time}", reference_time)
-                first_user_message = first_user_message.replace("{reference_user_name}", reference_user_name)
-                first_user_message = first_user_message.replace("{reference_user_display_name}", reference_user_display_name)
-                first_user_message = first_user_message.replace("{name}", bot_display_name)
-                first_user_message = first_user_message.replace("{bot_name}", bot_name)
-                first_user_message = first_user_message.replace("{current_time}", current_time)
-                first_user_message = first_user_message.replace("{user_display_name}", user_display_name)
-                first_user_message = first_user_message.replace("{user_name}", user_name)
-                
-                main_content = preset_data.get("main_content", "")
-                main_content = main_content.replace("{context}", context)
-                main_content = main_content.replace("{target_language}", target_language)
-                main_content = main_content.replace("{reference_content}", reference_content)
-                main_content = main_content.replace("{reference_time}", reference_time)
-                main_content = main_content.replace("{reference_user_name}", reference_user_name)
-                main_content = main_content.replace("{reference_user_display_name}", reference_user_display_name)
-                main_content = main_content.replace("{name}", bot_display_name)
-                main_content = main_content.replace("{bot_name}", bot_name)
-                main_content = main_content.replace("{current_time}", current_time)
-                main_content = main_content.replace("{user_display_name}", user_display_name)
-                main_content = main_content.replace("{user_name}", user_name)
-                
-                last_user_message = preset_data.get("last_user_message", "")
-                last_user_message = last_user_message.replace("{context}", context)
-                last_user_message = last_user_message.replace("{target_language}", target_language)
-                last_user_message = last_user_message.replace("{reference_content}", reference_content)
-                last_user_message = last_user_message.replace("{reference_time}", reference_time)
-                last_user_message = last_user_message.replace("{reference_user_name}", reference_user_name)
-                last_user_message = last_user_message.replace("{reference_user_display_name}", reference_user_display_name)
-                last_user_message = last_user_message.replace("{name}", bot_display_name)
-                last_user_message = last_user_message.replace("{bot_name}", bot_name)
-                last_user_message = last_user_message.replace("{current_time}", current_time)
-                last_user_message = last_user_message.replace("{user_display_name}", user_display_name)
-                last_user_message = last_user_message.replace("{user_name}", user_name)
-                
-                # 构建user-model-user的三个上下文
-                contents = [
-                    types.Content(
-                        role="user",
-                        parts=[
-                            types.Part.from_text(text=first_user_message),
-                        ],
-                    ),
-                    types.Content(
-                        role="model",
-                        parts=[
-                            types.Part.from_text(text=main_content),
-                        ],
-                    ),
-                    types.Content(
-                        role="user",
-                        parts=[
-                            types.Part.from_text(text=last_user_message),
-                        ],
-                    ),
-                ]
-                
-                # 如果有附件，添加到最后一个用户消息中
-                if attachment_bytes:
-                    # 使用Pillow和inline_data方式添加图片
-                    image_bytes = BytesIO(attachment_bytes)
-                    image = PIL.Image.open(image_bytes)
-                    
-                    # 转换为字节数据
-                    mime_type = attachment_mime_type or "image/jpeg"
-                    img_byte_arr = BytesIO()
-                    image.save(img_byte_arr, format=image.format or "JPEG")
-                    img_byte_data = img_byte_arr.getvalue()
-                    
-                    # 添加到消息中
-                    contents[2].parts.append(
-                        types.Part(
-                            inline_data=types.Blob(
-                                mime_type=mime_type,
-                                data=img_byte_data
-                            )
-                        )
-                    )
-                    print("附件已添加到翻译请求中")
-                
-                # 设置安全设置
-                safety_settings = [
-                    types.SafetySetting(
-                        category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
-                        threshold=types.HarmBlockThreshold.OFF,
-                    ),
-                    types.SafetySetting(
-                        category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-                        threshold=types.HarmBlockThreshold.OFF,
-                    ),
-                    types.SafetySetting(
-                        category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
-                        threshold=types.HarmBlockThreshold.OFF,
-                    ),
-                    types.SafetySetting(
-                        category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-                        threshold=types.HarmBlockThreshold.OFF,
-                    ),
-                    types.SafetySetting(
-                        category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-                        threshold=types.HarmBlockThreshold.OFF,
-                    ),
-                ]
-                
-                # 获取Gemini配置
-                gemini_config_data = None
-                if agent_manager:
-                    gemini_config_data = agent_manager.get_preset_json("gemini_config.json", channel_id)
-                
-                # 构建配置
-                generate_content_config = types.GenerateContentConfig(
-                    temperature=gemini_config_data.get("temperature", 1.0) if gemini_config_data else 1.0,
-                    top_p=gemini_config_data.get("top_p", 0.95) if gemini_config_data else 0.95,
-                    top_k=gemini_config_data.get("top_k", 64) if gemini_config_data else 64,
-                    max_output_tokens=gemini_config_data.get("max_output_tokens", 8192) if gemini_config_data else 8192,
-                    safety_settings=safety_settings,
-                    response_mime_type="text/plain",
-                    system_instruction=[
-                        types.Part.from_text(text=preset_data.get("system_prompt", "")),
-                    ],
-                )
-                
-                # 使用已经存在的msg变量，而不是创建新消息
-                # 如果之前没有创建消息（例如没有附件），现在才创建
-                if 'msg' not in locals() or msg is None:
-                    msg = await ctx.send("Translating...")
-                
-                full = ""
-                n = config.get("gemini_chunk_per_edit")
-                every_n_chunk = 1
-                
-                try:
-                    # 记录翻译请求内容
-                    log_contents = []
-                    for content in contents:
-                        parts_text = []
-                        for part in content.parts:
-                            if hasattr(part, "text") and part.text:
-                                parts_text.append(f"Text: {part.text}")
-                            else:
-                                parts_text.append(f"Unknown part type: {type(part)}")
-                        log_contents.append(f"Role: {content.role}, Parts: {parts_text}")
-                    
-                    system_instruction = "None"
-                    if hasattr(generate_content_config, "system_instruction"):
-                        if generate_content_config.system_instruction:
-                            system_instruction = generate_content_config.system_instruction[0].text if generate_content_config.system_instruction else "None"
-                    
-                    # 只记录到日志文件，不再重复打印到控制台
-                    logger.info(
-                        "Gemini翻译请求发送: 模型=gemini-2.0-pro-exp-02-05, 内容=%s, 系统提示=%s, 配置=%s",
-                        log_contents,
-                        system_instruction,
-                        {
-                            "temperature": generate_content_config.temperature,
-                            "top_p": generate_content_config.top_p,
-                            "top_k": generate_content_config.top_k,
-                            "max_tokens": generate_content_config.max_output_tokens,
-                        }
-                    )
-                    
-                    response = client.models.generate_content_stream(
-                        model=translate_model,  # 使用翻译模型
-                        contents=contents,
-                        config=generate_content_config,
-                    )
-                    
-                    async for chunk in async_iter(response):
-                        if chunk.text:
-                            full += chunk.text
-                            if every_n_chunk == n:
-                                await msg.edit(content=full)
-                                every_n_chunk = 1
-                            else:
-                                every_n_chunk += 1
-                    await msg.edit(content=full)
-                except Exception as e:
-                    logger.error(
-                        "Error when translating with gemini, error: %s",
-                        e,
-                        exc_info=True,
-                    )
-                    if full == "":
-                        await msg.edit(content="Uh oh, something went wrong...")
-                    else:
-                        full += "\nUh oh, something went wrong..."
-                        await msg.edit(content=full)
-        else:
-            # 预设数据不存在，显示错误信息
-            await ctx.send("无法加载翻译预设，请联系管理员", delete_after=5, ephemeral=True)
-
-    @commands.hybrid_command(
-        name="set_context_length", description="Set the context length."
-    )
-    @commands.is_owner()
-    @auto_delete(delay=0)
-    async def set_context_length(self, ctx: commands.Context, context_length: int):
-        self.context_length = context_length
-        await ctx.send("Context length set.", ephemeral=True, delete_after=5)
-
-    @commands.hybrid_command(
-        name="set_target_language", description="Set the target language."
-    )
-    @commands.is_owner()
-    @auto_delete(delay=0)
-    async def set_target_language(self, ctx: commands.Context, target_language: str):
-        self.target_language = target_language
-        await ctx.send("Target language set.", ephemeral=True, delete_after=5)
-
-    @commands.hybrid_command(name="set_timezone", description="Set the timezone.")
-    @commands.is_owner()
-    @auto_delete(delay=0)
-    async def set_timezone(self, ctx: commands.Context, timezone: str):
-        try:
-            self.context_prompter.set_tz(timezone)
-            await ctx.send(
-                f"Timezone set to {timezone}.", ephemeral=True, delete_after=5
-            )
-        except Exception as e:
-            await ctx.send(f"Invalid timezone.", ephemeral=True, delete_after=5)
-
-    @commands.hybrid_command(name="models", description="列出或切换Gemini模型")
-    @commands.has_permissions(administrator=True)
-    async def models(self, ctx: commands.Context, model_type: str = None, model_name: str = None):
-        """列出或切换Gemini模型
-        
-        参数:
-            model_type: 模型类型（chat或translate）
-            model_name: 模型名称
-        """
-        if not model_type:
-            # 如果没有指定模型类型，列出当前使用的模型和所有可用模型
-            embed = discord.Embed(
-                title="Gemini模型配置",
-                description="当前使用的Gemini模型",
-                color=discord.Color.blue()
-            )
-            
-            chat_model = self.gemini_models.get("chat", "gemini-2.0-pro-exp-02-05")
-            translate_model = self.gemini_models.get("translate", "gemini-2.0-pro-exp-02-05")
-            
-            embed.add_field(name="聊天模型", value=f"`{chat_model}`", inline=False)
-            embed.add_field(name="翻译模型", value=f"`{translate_model}`", inline=False)
-            
-            # 添加可用模型列表
-            available_models_text = ""
-            for model in self.available_models:
-                model_name = model.get("name", "")
-                model_desc = model.get("description", "")
-                if model_desc:
-                    available_models_text += f"• `{model_name}` - {model_desc}\n"
-                else:
-                    available_models_text += f"• `{model_name}`\n"
-            
-            if available_models_text:
-                embed.add_field(name="可用模型列表", value=available_models_text, inline=False)
-            else:
-                embed.add_field(name="可用模型列表", value="没有可用的模型", inline=False)
-            
-            embed.add_field(
-                name="使用方法", 
-                value="使用 `/models chat <模型名>` 更改聊天模型\n使用 `/models translate <模型名>` 更改翻译模型", 
-                inline=False
-            )
-            
-            await ctx.send(embed=embed)
-            return
-        
-        if model_type not in ["chat", "translate"]:
-            await ctx.send("错误：模型类型必须是 `chat` 或 `translate`", ephemeral=True)
-            return
-        
-        if not model_name:
-            await ctx.send(f"错误：请指定模型名称", ephemeral=True)
-            return
-        
-        # 检查模型是否在可用列表中
-        model_exists = any(model.get("name") == model_name for model in self.available_models)
-        if not model_exists:
-            # 仍然允许用户设置不在列表中的模型，但显示警告
-            await ctx.send(f"警告：模型 `{model_name}` 不在可用模型列表中，但仍将设置为当前模型。", ephemeral=True)
-        
-        # 更新模型配置
-        self.gemini_models[model_type] = model_name
-        config.write("gemini_models", self.gemini_models)
-        
-        await ctx.send(f"已将{model_type}模型设置为：{model_name}", ephemeral=True)
-    
-    @models.error
-    async def models_error(self, ctx: commands.Context, error):
-        if isinstance(error, commands.MissingPermissions):
-            await ctx.send("错误：只有管理员可以更改模型配置", ephemeral=True)
-
-
-async def setup(bot: commands.Bot):
-    apikeys = config.get("gemini_keys")
-    print(cpr.info(f"{len(apikeys)} keys loaded."))
-    webhook = discord.Webhook.from_url(
-        config.get("webhook_url"), session=ClientSession()
-    )
-    cog = Gemini(bot, webhook)
-    
-    # 设置AgentManager
-    try:
-        agent_manager = bot.get_cog("AgentManager")
-        if agent_manager:
-            cog.context_prompter.set_agent_manager(agent_manager)
-            # 加载Gemini配置
-            gemini_config = agent_manager.get_preset_json("gemini_config.json")
-            if gemini_config:
-                # 将JSON配置转换为Gemini配置对象
-                safety_settings = []
-                for setting in gemini_config.get("safety_settings", []):
-                    category = getattr(types.HarmCategory, setting["category"])
-                    threshold = getattr(types.HarmBlockThreshold, setting["threshold"])
-                    safety_settings.append(types.SafetySetting(
-                        category=category,
-                        threshold=threshold
-                    ))
-                
-                cog.default_gemini_config = types.GenerateContentConfig(
-                    system_instruction=gemini_config.get("system_instruction", ""),
-                    top_k=gemini_config.get("top_k", 55),
-                    top_p=gemini_config.get("top_p", 0.95),
-                    temperature=gemini_config.get("temperature", 1.3),
-                    safety_settings=safety_settings
-                )
-    except Exception as e:
-        print(f"Error loading Gemini config: {e}")
-    
-    await bot.add_cog(cog)
-    print(cpr.success("Cog loaded: Gemini"))
+from io import BytesIO
+from discord.ext import commands
+import discord
+from google import genai
+from google.genai import types
+import random
+from aiohttp import ClientSession
+from utils.decorator import auto_delete
+from utils.func import async_iter, async_do_thread
+from utils.color_printer import cpr
+from utils.config import config
+from utils.context_prompter import ContextPrompter
+from utils.logger import logger
+from datetime import datetime
+import PIL.Image
+import json
+
+
+class Gemini(commands.Cog):
+    def __init__(
+        self,
+        bot: commands.Bot,
+        webhook: discord.Webhook,
+    ):
+        self.bot = bot
+        self.conversations = {}
+        self.apikeys = config.get("gemini_keys")
+        self.current_key = config.get("current_key")
+        self.num = len(self.apikeys)
+        
+        # 确保chat_channels中的键全部为字符串
+        self.update_chat_channels()
+        
+        self.config = config
+        self.context_length = 20
+        self.target_language = config.get("target_language")
+        
+        # 获取Gemini模型配置
+        self.gemini_models = config.get("gemini_models", {
+            "chat": "gemini-2.0-pro-exp-02-05",  # 默认聊天模型
+            "translate": "gemini-2.0-pro-exp-02-05"  # 默认翻译模型
+        })
+        
+        # 如果配置中没有gemini_models，写入默认配置
+        if not config.get("gemini_models"):
+            config.write("gemini_models", self.gemini_models)
+        
+        self.default_gemini_config = types.GenerateContentConfig(
+            system_instruction="",
+            top_k=55,
+            top_p=0.95,
+            temperature=1.3,
+            safety_settings=[
+                types.SafetySetting(
+                    category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
+                    threshold=types.HarmBlockThreshold.OFF,
+                ),
+                types.SafetySetting(
+                    category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                    threshold=types.HarmBlockThreshold.OFF,
+                ),
+                types.SafetySetting(
+                    category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+                    threshold=types.HarmBlockThreshold.OFF,
+                ),
+                types.SafetySetting(
+                    category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                    threshold=types.HarmBlockThreshold.OFF,
+                ),
+                types.SafetySetting(
+                    category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                    threshold=types.HarmBlockThreshold.OFF,
+                ),
+            ],
+        )
+        self.webhook = webhook
+        self.context_prompter = ContextPrompter()
+        self.non_gemini_model = None  # for openai model
+        self.openai_api_key = config.get("openai_api_key")
+        self.openai_endpoint = config.get("openai_endpoint")
+
+        if self.openai_api_key is not None and self.openai_endpoint is not None:
+            print(cpr.info("OpenAI API available."))
+
+    def update_chat_channels(self):
+        """更新聊天频道配置"""
+        # 获取所有服务器配置
+        self.servers = config.get("servers", {})
+        
+        # 更新Gemini模型配置
+        self.gemini_models = config.get("gemini_models", {
+            "chat": "gemini-2.0-pro-exp-02-05",
+            "translate": "gemini-2.0-pro-exp-02-05"
+        })
+        
+        # 获取可用的Gemini模型列表
+        self.available_models = config.get("gemini_available_models", [])
+        if not self.available_models:
+            # 设置默认模型列表
+            self.available_models = [
+                {"name": "gemini-2.0-pro-exp-02-05", "description": "默认聊天和翻译模型"},
+                {"name": "gemini-pro", "description": "旧版Gemini Pro模型"}
+            ]
+            config.write("gemini_available_models", self.available_models)
+        
+        print(f"Gemini cog 已更新服务器配置和模型列表")
+    
+    def get_channel_config(self, guild_id: str, channel_id: str):
+        """获取频道配置"""
+        server_name, server_config = config.get_server_config(guild_id)
+        if not server_config:
+            return None
+        return server_config.get("chat_channels", {}).get(channel_id)
+
+    def get_next_key(self):
+        self.current_key = (self.current_key + 1) % self.num
+        config.write("current_key", self.current_key)
+        return self.apikeys[self.current_key]
+
+    def get_random_key(self):
+        return self.apikeys[random.randint(0, self.num - 1)]
+
+    async def request_gemini(
+        self,
+        ctx: commands.Context,
+        prompt: str,
+        model_config: types.GenerateContentConfig = None,
+        model="gemini-2.0-pro-exp-02-05",
+        username=None,
+        extra_attachment: discord.Attachment = None,
+    ):
+        if model_config is None:
+            model_config = self.default_gemini_config
+        
+        # 获取服务器和频道配置
+        guild_id = str(ctx.guild.id)
+        channel_id = str(ctx.channel.id)
+        channel_config = self.get_channel_config(guild_id, channel_id)
+        
+        if not channel_config:
+            await ctx.send("此频道未配置为聊天频道", ephemeral=True)
+            return
+        
+        print(f"当前频道配置: {channel_config}")
+        
+        # 尝试获取预设
+        agent_manager = self.bot.get_cog("AgentManager")
+        preset_data = None
+        preset_name = "chat_preset.json"  # 默认使用chat_preset.json
+        
+        # 根据情况选择不同的预设
+        if extra_attachment:
+            preset_name = "attachment_preset.json"
+            print(f"使用附件预设: {preset_name}")
+        elif ctx.message.reference:
+            preset_name = "reference_preset.json"
+        
+        # 获取预设内容，传递频道ID和服务器ID以获取该频道对应的预设
+        if agent_manager:
+            preset_data = agent_manager.get_preset_json(preset_name, channel_id, guild_id)
+        
+        if model != "gemini-2.0-pro-exp-02-05":
+            key = self.get_random_key()
+        else:
+            key = self.get_next_key()
+        client = genai.Client(api_key=key)
+        
+        # 处理附件 - 下载附件内容
+        attachment_bytes = None
+        attachment_mime_type = None
+        if extra_attachment:
+            msg = await ctx.send("Downloading the attachment...")
+            bytes_data = await extra_attachment.read()
+            attachment_bytes = bytes_data
+            attachment_mime_type = extra_attachment.content_type.split(";")[0]
+            await msg.edit(content="Processing the attachment...")
+            print(f"附件已下载: {extra_attachment.filename} ({attachment_mime_type})")
+        else:
+            msg = await ctx.send("Typing...") if username is None else await self.webhook.send("Typing...", wait=True, username=username)
+        
+        # 检查预设数据是否存在
+        if not preset_data:
+            await msg.edit(content="无法加载预设数据，请联系管理员")
+            return
+            
+        # 首先获取变量替换所需的数据
+        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        bot_name = ctx.me.name
+        bot_display_name = ctx.me.display_name
+        user_name = ctx.author.name
+        user_display_name = ctx.author.display_name
+        
+        # 处理上下文
+        context = ""
+        if hasattr(ctx, 'context') and ctx.context:
+            context = ctx.context
+        elif hasattr(ctx, 'history') and ctx.history:
+            context = ctx.history
+        else:
+            # 获取历史消息作为上下文
+            context = await self.context_prompter.get_context_for_prompt(ctx, self.context_length)
+        
+        # 确保context是字符串
+        if not isinstance(context, str):
+            context = str(context) if context is not None else ""
+        
+        # 替换预设中的变量
+        first_user_message = preset_data.get("first_user_message", "")
+        first_user_message = first_user_message.replace("{context}", context)
+        first_user_message = first_user_message.replace("{question}", prompt)
+        first_user_message = first_user_message.replace("{name}", bot_display_name)
+        first_user_message = first_user_message.replace("{bot_name}", bot_name)
+        first_user_message = first_user_message.replace("{current_time}", current_time)
+        first_user_message = first_user_message.replace("{user_display_name}", user_display_name)
+        first_user_message = first_user_message.replace("{user_name}", user_name)
+        
+        main_content = preset_data.get("main_content", "")
+        last_message_content = preset_data.get("last_message", "")
+        prefill_assistant_reply = preset_data.get("prefill_assistant_reply", False)
+        
+        main_content = main_content.replace("{context}", context)
+        main_content = main_content.replace("{question}", prompt)
+        main_content = main_content.replace("{name}", bot_display_name)
+        main_content = main_content.replace("{bot_name}", bot_name)
+        main_content = main_content.replace("{current_time}", current_time)
+        main_content = main_content.replace("{user_display_name}", user_display_name)
+        main_content = main_content.replace("{user_name}", user_name)
+        
+        # 如果是引用回复
+        if ctx.message.reference and 'reference' in preset_name:
+            reference = ctx.message.reference.resolved
+            reference_time = self.context_prompter.get_msg_time(reference)
+            reference_user_name = reference.author.name
+            reference_user_display_name = reference.author.display_name
+            reference_content = reference.content
+            
+            main_content = main_content.replace("{reference_time}", reference_time)
+            main_content = main_content.replace("{reference_user_name}", reference_user_name)
+            main_content = main_content.replace("{reference_user_display_name}", reference_user_display_name)
+            main_content = main_content.replace("{reference_content}", reference_content)
+            first_user_message = first_user_message.replace("{reference_time}", reference_time)
+            first_user_message = first_user_message.replace("{reference_user_name}", reference_user_name)
+            first_user_message = first_user_message.replace("{reference_user_display_name}", reference_user_display_name)
+            first_user_message = first_user_message.replace("{reference_content}", reference_content)
+        
+        # 替换 last_message 中的变量
+        last_message_content = last_message_content.replace("{context}", context)
+        last_message_content = last_message_content.replace("{question}", prompt)
+        last_message_content = last_message_content.replace("{name}", bot_display_name)
+        last_message_content = last_message_content.replace("{bot_name}", bot_name)
+        last_message_content = last_message_content.replace("{current_time}", current_time)
+        last_message_content = last_message_content.replace("{user_display_name}", user_display_name)
+        last_message_content = last_message_content.replace("{user_name}", user_name)
+        
+        # 构建user-model-user的三个上下文
+        user_parts = [types.Part.from_text(text=first_user_message)]
+        model_parts = [types.Part.from_text(text=main_content)]
+        last_message_parts = [types.Part.from_text(text=last_message_content)]
+        
+        # 如果有附件，添加到最后一个用户消息中
+        if attachment_bytes:
+            # 使用Pillow和inline_data方式添加图片
+            image_bytes = BytesIO(attachment_bytes)
+            image = PIL.Image.open(image_bytes)
+            
+            # 转换为字节数据
+            mime_type = attachment_mime_type or "image/jpeg"
+            img_byte_arr = BytesIO()
+            image.save(img_byte_arr, format=image.format or "JPEG")
+            img_byte_data = img_byte_arr.getvalue()
+            
+            # 添加到消息中
+            last_message_parts.append(
+                types.Part(
+                    inline_data=types.Blob(
+                        mime_type=mime_type,
+                        data=img_byte_data
+                    )
+                )
+            )
+            print("附件已添加到用户消息中")
+        
+        # 决定最后一条消息的角色
+        last_message_role = "model" if prefill_assistant_reply else "user"
+        
+        contents = [
+            types.Content(
+                role="user",
+                parts=user_parts,
+            ),
+            types.Content(
+                role=last_message_role,
+                parts=last_message_parts,
+            ),
+        ]
+        
+        # 设置安全设置
+        safety_settings = [
+            types.SafetySetting(
+                category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
+                threshold=types.HarmBlockThreshold.OFF,
+            ),
+            types.SafetySetting(
+                category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                threshold=types.HarmBlockThreshold.OFF,
+            ),
+            types.SafetySetting(
+                category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+                threshold=types.HarmBlockThreshold.OFF,
+            ),
+            types.SafetySetting(
+                category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                threshold=types.HarmBlockThreshold.OFF,
+            ),
+            types.SafetySetting(
+                category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                threshold=types.HarmBlockThreshold.OFF,
+            ),
+        ]
+        
+        # 获取Gemini配置
+        gemini_config_data = None
+        if agent_manager:
+            gemini_config_data = agent_manager.get_preset_json("gemini_config.json", channel_id, guild_id)
+        
+        # 构建配置
+        generate_content_config = types.GenerateContentConfig(
+            temperature=gemini_config_data.get("temperature", 1.0) if gemini_config_data else 1.0,
+            top_p=gemini_config_data.get("top_p", 0.95) if gemini_config_data else 0.95,
+            top_k=gemini_config_data.get("top_k", 64) if gemini_config_data else 64,
+            max_output_tokens=gemini_config_data.get("max_output_tokens", 8192) if gemini_config_data else 8192,
+            safety_settings=safety_settings,
+            response_mime_type="text/plain",
+            system_instruction=[
+                types.Part.from_text(text=preset_data.get("system_prompt", "")),
+            ],
+        )
+        
+        full = ""
+        n = config.get("gemini_chunk_per_edit")
+        every_n_chunk = 1
+        try:
+            # 准备要记录的原始请求数据
+            request_payload = {
+                "model": model,
+                "contents": [c.to_dict() for c in contents], # 转换为字典以便序列化
+                "generation_config": generate_content_config.to_dict(), # 使用构建的配置对象
+                "safety_settings": [s.to_dict() for s in safety_settings] # 记录明确传入的安全设置
+            }
+            
+            # 记录原始请求体 (或其近似结构) 到日志
+            logger.info(
+                "Gemini Raw Request: %s", 
+                json.dumps(request_payload, indent=2, ensure_ascii=False) 
+            )
+
+            # 使用流式响应
+            # 注意：实际调用时使用的是 self.default_gemini_config
+            # 为了日志准确反映实际调用，我们应该记录 self.default_gemini_config
+            # 或者修改调用以使用 generate_content_config
+            # 这里暂时保持原样，但指出日志记录的是构建的配置，而非最终调用的配置
+            
+            # !! 注意：下方调用使用的是 self.default_gemini_config，
+            #    而上方日志记录的是 generate_content_config。
+            #    为了调试准确性，应考虑统一或明确记录实际使用的配置。
+            response = client.models.generate_content_stream(
+                model=model,
+                contents=contents,
+                config=generate_content_config, # 使用构建的、已记录的配置
+            )
+
+            async for chunk in async_iter(response):
+                if chunk.text:
+                    full += chunk.text
+                    if every_n_chunk == n:
+                        await msg.edit(content=full)
+                        every_n_chunk = 1
+                    else:
+                        every_n_chunk += 1
+            await msg.edit(content=full)
+        except Exception as e:
+            logger.error(
+                "Error when requesting gemini with key: %s, error: %s",
+                key,
+                e,
+                exc_info=True,
+            )
+            if full == "":
+                await msg.edit(content="Uh oh, something went wrong...")
+            else:
+                full += "\nUh oh, something went wrong..."
+                await msg.edit(content=full)
+
+    @commands.hybrid_command(name="hey", description="Ask a question to gemini.")
+    async def hey(
+        self,
+        ctx: commands.Context,
+        *,
+        question: str,
+        context_length: int = None,
+    ):
+        # 获取服务器和频道配置
+        guild_id = str(ctx.guild.id)
+        channel_id = str(ctx.channel.id)
+        channel_config = self.get_channel_config(guild_id, channel_id)
+        
+        if not channel_config:
+            await ctx.send("此频道未配置为聊天频道", ephemeral=True)
+            return
+            
+        if context_length is None:
+            context_length = self.context_length
+        extra_attachment = None
+        
+        # 获取历史消息作为上下文
+        history = await self.context_prompter.get_context_for_prompt(ctx, context_length)
+        ctx.history = history  # 将历史消息保存到ctx对象中，供预设处理使用
+        
+        # 检查附件
+        if ctx.message.reference:
+            reference = ctx.message.reference.resolved
+            # 优先查找引用消息中的附件
+            if reference and reference.attachments:
+                extra_attachment = reference.attachments[-1]
+        
+        # 选择合适的预设
+        agent_manager = self.bot.get_cog("AgentManager")
+        preset_name = "chat_preset.json"  # 默认使用chat_preset.json
+        
+        if agent_manager:
+            if ctx.message.reference:
+                reference = ctx.message.reference.resolved
+                if reference and reference.attachments:
+                    preset_name = "attachment_preset.json"
+                else:
+                    preset_name = "reference_preset.json"
+        
+        # 检查附件是否存在，确保传递正确
+        if extra_attachment:
+            print(f"处理附件: {extra_attachment.filename} ({extra_attachment.content_type})")
+        
+        # 使用聊天模型
+        chat_model = self.gemini_models.get("chat", "gemini-2.0-pro-exp-02-05")
+        
+        # 发送请求
+        await self.request_gemini(
+            ctx,
+            question,  # 直接传递原始问题，预设处理在request_gemini中完成
+            model=chat_model,
+            extra_attachment=extra_attachment,
+        )
+
+    @commands.hybrid_command(name="tr", description="Translate a text.")
+    async def translate(
+        self,
+        ctx: commands.Context,
+        target_language: str = None,
+        context_length: int = None,
+    ):
+        # 获取服务器和频道配置
+        guild_id = str(ctx.guild.id)
+        channel_id = str(ctx.channel.id)
+        channel_config = self.get_channel_config(guild_id, channel_id)
+        
+        if not channel_config:
+            await ctx.send("此频道未配置为聊天频道", ephemeral=True)
+            return
+            
+        if ctx.message.reference is None:
+            await ctx.send(
+                "请回复要翻译的消息", ephemeral=True
+            )
+            return
+        if context_length is None:
+            context_length = self.context_length
+        if target_language is None:
+            target_language = self.target_language
+
+        # 使用翻译模型
+        translate_model = self.gemini_models.get("translate", "gemini-2.0-pro-exp-02-05")
+
+        # 尝试获取翻译预设
+        agent_manager = self.bot.get_cog("AgentManager")
+        preset_data = None
+        
+        # 获取被回复的消息
+        reference_message = await ctx.channel.fetch_message(
+            ctx.message.reference.message_id
+        )
+        
+        # 检查是否有附件
+        extra_attachment = None
+        if reference_message and reference_message.attachments:
+            extra_attachment = reference_message.attachments[-1]
+            print(f"翻译附件: {extra_attachment.filename} ({extra_attachment.content_type})")
+        
+        # 下载附件内容
+        attachment_bytes = None
+        attachment_mime_type = None
+        if extra_attachment:
+            msg = await ctx.send("Downloading the attachment...")
+            bytes_data = await extra_attachment.read()
+            attachment_bytes = bytes_data
+            attachment_mime_type = extra_attachment.content_type.split(";")[0]
+            await msg.edit(content="Processing the attachment...")
+            print(f"附件已下载: {extra_attachment.filename} ({attachment_mime_type})")
+        else:
+            msg = await ctx.send("Translating...")
+        
+        # 获取预设内容
+        if agent_manager:
+            preset_data = agent_manager.get_preset_json("translate_preset.json")
+        
+        if preset_data:
+            # 使用预设JSON结构和原始文本
+            key = self.get_next_key()
+            client = genai.Client(api_key=key)
+            
+            # 获取变量替换所需的数据
+            current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            bot_name = ctx.me.name
+            bot_display_name = ctx.me.display_name
+            user_name = ctx.author.name
+            user_display_name = ctx.author.display_name
+            
+            # 处理上下文和引用内容
+            context = ""
+            if hasattr(ctx, 'context') and ctx.context:
+                context = ctx.context
+            elif hasattr(ctx, 'history') and ctx.history:
+                context = ctx.history
+            else:
+                # 获取历史消息作为上下文
+                context = await self.context_prompter.get_context_for_prompt(ctx, context_length)
+            
+            # 确保context是字符串
+            if not isinstance(context, str):
+                context = str(context) if context is not None else ""
+            
+            # 获取被引用消息的相关信息
+            reference_time = reference_message.created_at.strftime("%Y-%m-%d %H:%M:%S")
+            reference_user_name = reference_message.author.name
+            reference_user_display_name = reference_message.author.display_name
+            reference_content = reference_message.content
+            
+            # 使用预设JSON结构
+            system_prompt = preset_data.get("system_prompt", "")
+            first_user_message = preset_data.get("first_user_message", "")
+            
+            # 确保模型已设置且处于传递中
+            translate_model = self.gemini_models.get("translate", "gemini-2.0-pro-exp-02-05")
+            
+            # 处理不同内容格式情况
+            if attachment_bytes and attachment_mime_type:
+                # 如果有图片附件
+                try:
+                    # 将字节转换为PIL图像以供Gemini处理
+                    if attachment_mime_type.startswith("image/"):
+                        image = PIL.Image.open(BytesIO(attachment_bytes))
+                        
+                        # 按照Gemini推荐尺寸调整图片，适应大多数模型版本
+                        if max(image.size) > 3072:
+                            ratio = 3072 / max(image.size)
+                            new_size = (int(image.size[0] * ratio), int(image.size[1] * ratio))
+                            image = image.resize(new_size, PIL.Image.LANCZOS)
+                        
+                        # 构建内容
+                        content = [
+                            {
+                                "role": "user",
+                                "parts": [
+                                    types.Part.from_text(first_user_message),
+                                    types.Part.from_image(image),
+                                ]
+                            }
+                        ]
+                        
+                        # 获取安全设置
+                        safety_settings = []
+                        try:
+                            gemini_config_data = None
+                            if agent_manager:
+                                gemini_config_data = agent_manager.get_preset_json("gemini_config.json")
+                                
+                            if gemini_config_data and "safety_settings" in gemini_config_data:
+                                safety_settings = self.build_safety_settings(gemini_config_data["safety_settings"])
+                        except Exception as e:
+                            logger.error(f"Error loading safety settings: {e}")
+                            safety_settings = []
+                        
+                        # 构建和应用生成内容配置
+                        generate_content_config = types.GenerateContentConfig(
+                            temperature=0.2,
+                            top_p=0.95,
+                            top_k=55,
+                            max_output_tokens=8192,
+                            safety_settings=safety_settings,
+                            response_mime_type="text/plain",
+                            system_instruction=[
+                                types.Part.from_text(text=system_prompt),
+                            ],
+                        )
+                        
+                        # 使用流式响应
+                        full = ""
+                        n = config.get("gemini_chunk_per_edit")
+                        every_n_chunk = 1
+                        
+                        try:
+                            response = client.models.generate_content_stream(
+                                model=translate_model,  # 使用翻译模型
+                                contents=content,
+                                config=generate_content_config,
+                            )
+                            
+                            async for chunk in async_iter(response):
+                                if chunk.text:
+                                    full += chunk.text
+                                    if every_n_chunk == n:
+                                        await msg.edit(content=full)
+                                        every_n_chunk = 1
+                                    else:
+                                        every_n_chunk += 1
+                            await msg.edit(content=full)
+                        except Exception as e:
+                            logger.error(
+                                "Error when translating with gemini, error: %s",
+                                e,
+                                exc_info=True,
+                            )
+                            if full == "":
+                                await msg.edit(content="Uh oh, something went wrong...")
+                            else:
+                                full += "\nUh oh, something went wrong..."
+                                await msg.edit(content=full)
+                except Exception as e:
+                    logger.error(f"Error processing image: {e}")
+                    await ctx.send("无法处理图片，请检查附件格式", ephemeral=True)
+            else:
+                # 替换预设中的变量
+                first_user_message = preset_data.get("first_user_message", "")
+                first_user_message = first_user_message.replace("{context}", context)
+                first_user_message = first_user_message.replace("{target_language}", target_language)
+                first_user_message = first_user_message.replace("{reference_content}", reference_content)
+                first_user_message = first_user_message.replace("{reference_time}", reference_time)
+                first_user_message = first_user_message.replace("{reference_user_name}", reference_user_name)
+                first_user_message = first_user_message.replace("{reference_user_display_name}", reference_user_display_name)
+                first_user_message = first_user_message.replace("{name}", bot_display_name)
+                first_user_message = first_user_message.replace("{bot_name}", bot_name)
+                first_user_message = first_user_message.replace("{current_time}", current_time)
+                first_user_message = first_user_message.replace("{user_display_name}", user_display_name)
+                first_user_message = first_user_message.replace("{user_name}", user_name)
+                
+                main_content = preset_data.get("main_content", "")
+                main_content = main_content.replace("{context}", context)
+                main_content = main_content.replace("{target_language}", target_language)
+                main_content = main_content.replace("{reference_content}", reference_content)
+                main_content = main_content.replace("{reference_time}", reference_time)
+                main_content = main_content.replace("{reference_user_name}", reference_user_name)
+                main_content = main_content.replace("{reference_user_display_name}", reference_user_display_name)
+                main_content = main_content.replace("{name}", bot_display_name)
+                main_content = main_content.replace("{bot_name}", bot_name)
+                main_content = main_content.replace("{current_time}", current_time)
+                main_content = main_content.replace("{user_display_name}", user_display_name)
+                main_content = main_content.replace("{user_name}", user_name)
+                
+                last_message_content = preset_data.get("last_message", "")
+                last_message_content = last_message_content.replace("{context}", context)
+                last_message_content = last_message_content.replace("{target_language}", target_language)
+                last_message_content = last_message_content.replace("{reference_content}", reference_content)
+                last_message_content = last_message_content.replace("{reference_time}", reference_time)
+                last_message_content = last_message_content.replace("{reference_user_name}", reference_user_name)
+                last_message_content = last_message_content.replace("{reference_user_display_name}", reference_user_display_name)
+                last_message_content = last_message_content.replace("{name}", bot_display_name)
+                last_message_content = last_message_content.replace("{bot_name}", bot_name)
+                last_message_content = last_message_content.replace("{current_time}", current_time)
+                last_message_content = last_message_content.replace("{user_display_name}", user_display_name)
+                last_message_content = last_message_content.replace("{user_name}", user_name)
+                
+                # 构建user-model-user的三个上下文
+                contents = [
+                    types.Content(
+                        role="user",
+                        parts=[
+                            types.Part.from_text(text=first_user_message),
+                        ],
+                    ),
+                    types.Content(
+                        role="model",
+                        parts=[
+                            types.Part.from_text(text=main_content),
+                        ],
+                    ),
+                    types.Content(
+                        role="user",
+                        parts=[
+                            types.Part.from_text(text=last_message_content),
+                        ],
+                    ),
+                ]
+                
+                # 如果有附件，添加到最后一个用户消息中
+                if attachment_bytes:
+                    # 使用Pillow和inline_data方式添加图片
+                    image_bytes = BytesIO(attachment_bytes)
+                    image = PIL.Image.open(image_bytes)
+                    
+                    # 转换为字节数据
+                    mime_type = attachment_mime_type or "image/jpeg"
+                    img_byte_arr = BytesIO()
+                    image.save(img_byte_arr, format=image.format or "JPEG")
+                    img_byte_data = img_byte_arr.getvalue()
+                    
+                    # 添加到消息中
+                    contents[2].parts.append(
+                        types.Part(
+                            inline_data=types.Blob(
+                                mime_type=mime_type,
+                                data=img_byte_data
+                            )
+                        )
+                    )
+                    print("附件已添加到翻译请求中")
+                
+                # 设置安全设置
+                safety_settings = [
+                    types.SafetySetting(
+                        category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
+                        threshold=types.HarmBlockThreshold.OFF,
+                    ),
+                    types.SafetySetting(
+                        category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                        threshold=types.HarmBlockThreshold.OFF,
+                    ),
+                    types.SafetySetting(
+                        category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+                        threshold=types.HarmBlockThreshold.OFF,
+                    ),
+                    types.SafetySetting(
+                        category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                        threshold=types.HarmBlockThreshold.OFF,
+                    ),
+                    types.SafetySetting(
+                        category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                        threshold=types.HarmBlockThreshold.OFF,
+                    ),
+                ]
+                
+                # 获取Gemini配置
+                gemini_config_data = None
+                if agent_manager:
+                    gemini_config_data = agent_manager.get_preset_json("gemini_config.json", channel_id)
+                
+                # 构建配置
+                generate_content_config = types.GenerateContentConfig(
+                    temperature=gemini_config_data.get("temperature", 1.0) if gemini_config_data else 1.0,
+                    top_p=gemini_config_data.get("top_p", 0.95) if gemini_config_data else 0.95,
+                    top_k=gemini_config_data.get("top_k", 64) if gemini_config_data else 64,
+                    max_output_tokens=gemini_config_data.get("max_output_tokens", 8192) if gemini_config_data else 8192,
+                    safety_settings=safety_settings,
+                    response_mime_type="text/plain",
+                    system_instruction=[
+                        types.Part.from_text(text=preset_data.get("system_prompt", "")),
+                    ],
+                )
+                
+                # 使用已经存在的msg变量，而不是创建新消息
+                # 如果之前没有创建消息（例如没有附件），现在才创建
+                if 'msg' not in locals() or msg is None:
+                    msg = await ctx.send("Translating...")
+                
+                full = ""
+                n = config.get("gemini_chunk_per_edit")
+                every_n_chunk = 1
+                
+                try:
+                    # 记录翻译请求内容
+                    log_contents = []
+                    for content in contents:
+                        parts_text = []
+                        for part in content.parts:
+                            if hasattr(part, "text") and part.text:
+                                parts_text.append(f"Text: {part.text}")
+                            else:
+                                parts_text.append(f"Unknown part type: {type(part)}")
+                        log_contents.append(f"Role: {content.role}, Parts: {parts_text}")
+                    
+                    system_instruction = "None"
+                    if hasattr(generate_content_config, "system_instruction"):
+                        if generate_content_config.system_instruction:
+                            system_instruction = generate_content_config.system_instruction[0].text if generate_content_config.system_instruction else "None"
+                    
+                    # 只记录到日志文件，不再重复打印到控制台
+                    logger.info(
+                        "Gemini翻译请求发送: 模型=gemini-2.0-pro-exp-02-05, 内容=%s, 系统提示=%s, 配置=%s",
+                        log_contents,
+                        system_instruction,
+                        {
+                            "temperature": generate_content_config.temperature,
+                            "top_p": generate_content_config.top_p,
+                            "top_k": generate_content_config.top_k,
+                            "max_tokens": generate_content_config.max_output_tokens,
+                        }
+                    )
+                    
+                    response = client.models.generate_content_stream(
+                        model=translate_model,  # 使用翻译模型
+                        contents=contents,
+                        config=generate_content_config,
+                    )
+                    
+                    async for chunk in async_iter(response):
+                        if chunk.text:
+                            full += chunk.text
+                            if every_n_chunk == n:
+                                await msg.edit(content=full)
+                                every_n_chunk = 1
+                            else:
+                                every_n_chunk += 1
+                    await msg.edit(content=full)
+                except Exception as e:
+                    logger.error(
+                        "Error when translating with gemini, error: %s",
+                        e,
+                        exc_info=True,
+                    )
+                    if full == "":
+                        await msg.edit(content="Uh oh, something went wrong...")
+                    else:
+                        full += "\nUh oh, something went wrong..."
+                        await msg.edit(content=full)
+        else:
+            # 预设数据不存在，显示错误信息
+            await ctx.send("无法加载翻译预设，请联系管理员", delete_after=5, ephemeral=True)
+
+    @commands.hybrid_command(
+        name="set_context_length", description="Set the context length."
+    )
+    @commands.is_owner()
+    @auto_delete(delay=0)
+    async def set_context_length(self, ctx: commands.Context, context_length: int):
+        self.context_length = context_length
+        await ctx.send("Context length set.", ephemeral=True, delete_after=5)
+
+    @commands.hybrid_command(
+        name="set_target_language", description="Set the target language."
+    )
+    @commands.is_owner()
+    @auto_delete(delay=0)
+    async def set_target_language(self, ctx: commands.Context, target_language: str):
+        self.target_language = target_language
+        await ctx.send("Target language set.", ephemeral=True, delete_after=5)
+
+    @commands.hybrid_command(name="set_timezone", description="Set the timezone.")
+    @commands.is_owner()
+    @auto_delete(delay=0)
+    async def set_timezone(self, ctx: commands.Context, timezone: str):
+        try:
+            self.context_prompter.set_tz(timezone)
+            await ctx.send(
+                f"Timezone set to {timezone}.", ephemeral=True, delete_after=5
+            )
+        except Exception as e:
+            await ctx.send(f"Invalid timezone.", ephemeral=True, delete_after=5)
+
+    @commands.hybrid_command(name="models", description="列出或切换Gemini模型")
+    @commands.has_permissions(administrator=True)
+    async def models(self, ctx: commands.Context, model_type: str = None, model_name: str = None):
+        """列出或切换Gemini模型
+        
+        参数:
+            model_type: 模型类型（chat或translate）
+            model_name: 模型名称
+        """
+        if not model_type:
+            # 如果没有指定模型类型，列出当前使用的模型和所有可用模型
+            embed = discord.Embed(
+                title="Gemini模型配置",
+                description="当前使用的Gemini模型",
+                color=discord.Color.blue()
+            )
+            
+            chat_model = self.gemini_models.get("chat", "gemini-2.0-pro-exp-02-05")
+            translate_model = self.gemini_models.get("translate", "gemini-2.0-pro-exp-02-05")
+            
+            embed.add_field(name="聊天模型", value=f"`{chat_model}`", inline=False)
+            embed.add_field(name="翻译模型", value=f"`{translate_model}`", inline=False)
+            
+            # 添加可用模型列表
+            available_models_text = ""
+            for model in self.available_models:
+                model_name = model.get("name", "")
+                model_desc = model.get("description", "")
+                if model_desc:
+                    available_models_text += f"• `{model_name}` - {model_desc}\n"
+                else:
+                    available_models_text += f"• `{model_name}`\n"
+            
+            if available_models_text:
+                embed.add_field(name="可用模型列表", value=available_models_text, inline=False)
+            else:
+                embed.add_field(name="可用模型列表", value="没有可用的模型", inline=False)
+            
+            embed.add_field(
+                name="使用方法", 
+                value="使用 `/models chat <模型名>` 更改聊天模型\n使用 `/models translate <模型名>` 更改翻译模型", 
+                inline=False
+            )
+            
+            await ctx.send(embed=embed)
+            return
+        
+        if model_type not in ["chat", "translate"]:
+            await ctx.send("错误：模型类��必须是 `chat` 或 `translate`", ephemeral=True)
+            return
+        
+        if not model_name:
+            await ctx.send(f"错误：请指定模型名称", ephemeral=True)
+            return
+        
+        # 检查模型是否在可用列表中
+        model_exists = any(model.get("name") == model_name for model in self.available_models)
+        if not model_exists:
+            # 仍然允许用户设置不在列表中的模型，但显示警告
+            await ctx.send(f"警告：模型 `{model_name}` 不在可用模型列表中，但仍将设置为当前模型。", ephemeral=True)
+        
+        # 更新模型配置
+        self.gemini_models[model_type] = model_name
+        config.write("gemini_models", self.gemini_models)
+        
+        await ctx.send(f"已将{model_type}模型设置为：{model_name}", ephemeral=True)
+    
+    @models.error
+    async def models_error(self, ctx: commands.Context, error):
+        if isinstance(error, commands.MissingPermissions):
+            await ctx.send("错误：只有管理员可以更改模型配置", ephemeral=True)
+
+
+async def setup(bot: commands.Bot):
+    apikeys = config.get("gemini_keys")
+    print(cpr.info(f"{len(apikeys)} keys loaded."))
+    webhook = discord.Webhook.from_url(
+        config.get("webhook_url"), session=ClientSession()
+    )
+    cog = Gemini(bot, webhook)
+    
+    # 设置AgentManager
+    try:
+        agent_manager = bot.get_cog("AgentManager")
+        if agent_manager:
+            cog.context_prompter.set_agent_manager(agent_manager)
+            # 加载Gemini配置
+            gemini_config = agent_manager.get_preset_json("gemini_config.json")
+            if gemini_config:
+                # 将JSON配置转换为Gemini配置对象
+                safety_settings = []
+                for setting in gemini_config.get("safety_settings", []):
+                    category = getattr(types.HarmCategory, setting["category"])
+                    threshold = getattr(types.HarmBlockThreshold, setting["threshold"])
+                    safety_settings.append(types.SafetySetting(
+                        category=category,
+                        threshold=threshold
+                    ))
+                
+                cog.default_gemini_config = types.GenerateContentConfig(
+                    system_instruction=gemini_config.get("system_instruction", ""),
+                    top_k=gemini_config.get("top_k", 55),
+                    top_p=gemini_config.get("top_p", 0.95),
+                    temperature=gemini_config.get("temperature", 1.3),
+                    safety_settings=safety_settings
+                )
+    except Exception as e:
+        print(f"Error loading Gemini config: {e}")
+    
+    await bot.add_cog(cog)
+    print(cpr.success("Cog loaded: Gemini"))

cogs/openai.py

@@ -112,9 +112,12 @@ class Openai(commands.Cog):
             messages = [{"role": "user", "content": prompt}]
         else:
             # 构建符合预设格式的消息数组
+            prefill_assistant_reply = preset.get("prefill_assistant_reply", False)
+            last_message_content = preset.get("last_message", "Your reply:")
+
             if preset.get("system_prompt"):
                 messages.append({
-                    "role": "system", 
+                    "role": "system",
                     "content": preset["system_prompt"]
                 })
             
@@ -128,9 +131,11 @@ class Openai(commands.Cog):
                 "content": prompt  # 这里是原始prompt，它已经包含了main_content的格式
             })
             
+            # 决定最后一条消息的角色和内容
+            last_message_role = "assistant" if prefill_assistant_reply else "user"
             messages.append({
-                "role": "user", 
-                "content": preset.get("last_user_message", "Your reply:")
+                "role": last_message_role, 
+                "content": last_message_content
             })
         
         data = {
@@ -143,7 +148,11 @@ class Openai(commands.Cog):
         }
         
         # 记录API请求数据
-        logger.info(f"OpenAI API 请求数据: {json.dumps(data, ensure_ascii=False, indent=2)}")
+        log_headers = {k: v for k, v in headers.items() if k.lower() != 'authorization'} # 不记录原始Key
+        log_headers['Authorization'] = 'Bearer ***' # 隐藏Key
+        logger.info(
+            f"OpenAI API Raw Request:\nURL: {url}\nHeaders: {json.dumps(log_headers, ensure_ascii=False, indent=2)}\nBody: {json.dumps(data, ensure_ascii=False, indent=2)}"
+        )
         
         return url, headers, data
 

main.py

@@ -1,430 +1,408 @@
-import asyncio
-import json
-from discord.ext import commands
-import discord
-import os
-import ssl
-import sys
-import traceback
-import aiohttp
-import logging
-import threading
-import webbrowser
-import time
-import requests
-from flask import Flask, render_template, request, jsonify, redirect, url_for, send_from_directory, session
-from werkzeug.utils import secure_filename
-from PIL import Image, ImageDraw, ImageFont
-import io
-import shutil
-from utils.config import config
-from functools import wraps
-
-# 配置Flask应用
-app = Flask(__name__, 
-            static_folder='static',
-            template_folder='templates')
-app.secret_key = os.urandom(24)  # 设置session密钥
-
-# 登录验证装饰器
-def login_required(f):
-    @wraps(f)
-    def decorated_function(*args, **kwargs):
-        if not session.get('logged_in'):
-            return redirect(url_for('login'))
-        return f(*args, **kwargs)
-    return decorated_function
-
-# 配置日志级别，过滤掉 google_genai.models 的 INFO 级别日志
-logging.getLogger('google_genai.models').setLevel(logging.WARNING)
-
-# Disable SSL verification (only for debugging)
-ssl_context = ssl.create_default_context()
-ssl_context.check_hostname = False
-ssl_context.verify_mode = ssl.CERT_NONE
-
-intents = discord.Intents.default()
-intents.message_content = True
-intents.members = True
-
-bot = commands.Bot(command_prefix=".", intents=intents)
-
-# 创建默认头像
-def create_default_avatar():
-    avatar_path = os.path.join('static', 'img', 'default-avatar.png')
-    
-    # 如果已存在，则不需要创建
-    if os.path.exists(avatar_path):
-        return
-    
-    # 确保目录存在
-    os.makedirs(os.path.dirname(avatar_path), exist_ok=True)
-    
-    # 创建简单的默认头像
-    img = Image.new('RGB', (200, 200), color=(53, 102, 220))
-    d = ImageDraw.Draw(img)
-    
-    # 添加文字
-    try:
-        # 尝试使用系统字体
-        font = ImageFont.truetype("arial.ttf", 60)
-    except:
-        # 如果没有系统字体，使用默认字体
-        font = ImageFont.load_default()
-    
-    d.text((70, 70), "Bot", fill=(255, 255, 255), font=font)
-    
-    # 保存图片
-    img.save(avatar_path)
-
-# Flask路由
-@app.route('/login')
-def login():
-    if session.get('logged_in'):
-        return redirect(url_for('index'))
-    return render_template('login.html')
-
-@app.route('/api/login', methods=['POST'])
-def api_login():
-    data = request.json
-    password = data.get('password')
-    
-    # 直接从config.json读取最新密码
-    try:
-        with open('config.json', 'r', encoding='utf-8') as f:
-            config_data = json.load(f)
-            config_password = config_data.get('password')
-    except:
-        config_password = None
-    
-    if not config_password:
-        # 如果配置中没有密码，设置默认密码为 "admin"
-        config_password = "admin"
-        config.write("password", config_password)
-    
-    if password == config_password:
-        session['logged_in'] = True
-        return jsonify({"status": "success"})
-    return jsonify({"status": "error", "message": "密码错误"})
-
-@app.route('/logout')
-def logout():
-    session.pop('logged_in', None)
-    return redirect(url_for('login'))
-
-@app.route('/')
-@login_required
-def index():
-    return render_template('index.html')
-
-@app.route('/config')
-@login_required
-def config_page():
-    return render_template('config.html')
-
-@app.route('/presets')
-@login_required
-def presets_page():
-    presets = os.listdir('agent/presets')
-    return render_template('presets.html', presets=presets)
-
-@app.route('/api/config', methods=['GET'])
-@login_required
-def get_config():
-    with open('config.json', 'r', encoding='utf-8') as f:
-        config_data = json.load(f)
-    return jsonify(config_data)
-
-@app.route('/api/config', methods=['POST'])
-@login_required
-def update_config():
-    config_data = request.json
-    
-    # 检查密码是否被修改
-    current_password = config.get('password')
-    password_changed = current_password != config_data.get('password')
-    
-    # 保存配置
-    with open('config.json', 'w', encoding='utf-8') as f:
-        json.dump(config_data, f, indent=4, ensure_ascii=False)
-    
-    # 如果密码被修改,返回特殊状态码
-    if password_changed:
-        return jsonify({"status": "password_changed"})
-    return jsonify({"status": "success"})
-
-@app.route('/api/presets', methods=['GET'])
-def get_presets():
-    presets = os.listdir('agent/presets')
-    presets_data = {}
-    for preset in presets:
-        preset_path = os.path.join('agent/presets', preset)
-        if os.path.isdir(preset_path):
-            presets_data[preset] = {}
-            for file in os.listdir(preset_path):
-                if file.endswith('.json'):
-                    with open(os.path.join(preset_path, file), 'r', encoding='utf-8') as f:
-                        presets_data[preset][file] = json.load(f)
-    return jsonify(presets_data)
-
-@app.route('/api/preset/<preset_name>', methods=['GET'])
-def get_preset(preset_name):
-    preset_path = os.path.join('agent/presets', preset_name)
-    if not os.path.exists(preset_path):
-        return jsonify({"status": "error", "message": "Preset not found"}), 404
-    
-    preset_data = {}
-    for file in os.listdir(preset_path):
-        if file.endswith('.json'):
-            with open(os.path.join(preset_path, file), 'r', encoding='utf-8') as f:
-                preset_data[file] = json.load(f)
-    
-    return jsonify(preset_data)
-
-@app.route('/api/preset/<preset_name>', methods=['PUT'])
-def update_preset(preset_name):
-    preset_data = request.json
-    preset_path = os.path.join('agent/presets', preset_name)
-    if not os.path.exists(preset_path):
-        return jsonify({"status": "error", "message": "Preset not found"}), 404
-    
-    for file_name, data in preset_data.items():
-        with open(os.path.join(preset_path, file_name), 'w', encoding='utf-8') as f:
-            json.dump(data, f, indent=4, ensure_ascii=False)
-    
-    return jsonify({"status": "success"})
-
-@app.route('/api/preset/<preset_name>', methods=['DELETE'])
-def delete_preset(preset_name):
-    preset_path = os.path.join('agent/presets', preset_name)
-    if not os.path.exists(preset_path):
-        return jsonify({"status": "error", "message": "Preset not found"}), 404
-    
-    shutil.rmtree(preset_path)
-    return jsonify({"status": "success"})
-
-@app.route('/api/preset', methods=['POST'])
-def create_preset():
-    data = request.json
-    preset_name = data.get('name')
-    template_preset = data.get('template', 'default')
-    
-    if not preset_name:
-        return jsonify({"status": "error", "message": "Preset name is required"}), 400
-    
-    new_preset_path = os.path.join('agent/presets', preset_name)
-    if os.path.exists(new_preset_path):
-        return jsonify({"status": "error", "message": "Preset already exists"}), 400
-    
-    # 复制模板预设
-    template_path = os.path.join('agent/presets', template_preset)
-    if not os.path.exists(template_path):
-        return jsonify({"status": "error", "message": "Template preset not found"}), 404
-    
-    shutil.copytree(template_path, new_preset_path)
-    
-    return jsonify({"status": "success"})
-
-@app.route('/api/preset/<preset_name>/avatar', methods=['POST'])
-def upload_avatar(preset_name):
-    preset_path = os.path.join('agent/presets', preset_name)
-    if not os.path.exists(preset_path):
-        return jsonify({"status": "error", "message": "Preset not found"}), 404
-    
-    if 'avatar' not in request.files:
-        return jsonify({"status": "error", "message": "No file part"}), 400
-    
-    file = request.files['avatar']
-    if file.filename == '':
-        return jsonify({"status": "error", "message": "No selected file"}), 400
-    
-    # 处理图片
-    try:
-        img = Image.open(file)
-        
-        # 调整为正方形
-        width, height = img.size
-        size = min(width, height)
-        left = (width - size) // 2
-        top = (height - size) // 2
-        right = left + size
-        bottom = top + size
-        img = img.crop((left, top, right, bottom))
-        
-        # 保存图片
-        for old_avatar in os.listdir(preset_path):
-            if old_avatar.startswith('avatar.'):
-                os.remove(os.path.join(preset_path, old_avatar))
-        
-        # 获取文件扩展名
-        ext = os.path.splitext(file.filename)[1].lower()
-        if ext not in ['.jpg', '.jpeg', '.png', '.gif']:
-            ext = '.jpg'  # 默认使用jpg
-        
-        img.save(os.path.join(preset_path, f'avatar{ext}'))
-        
-        return jsonify({"status": "success"})
-    except Exception as e:
-        return jsonify({"status": "error", "message": str(e)}), 500
-
-@app.route('/agent/presets/<path:filename>')
-def preset_files(filename):
-    return send_from_directory('agent/presets', filename)
-
-@bot.event
-async def on_ready():
-    print(f'Bot logged in as {bot.user.name} (ID: {bot.user.id})')
-    print(f'Bot invite link: https://discord.com/oauth2/authorize?client_id={bot.user.id}&permissions=8&scope=bot')
-    
-    # Auto sync slash commands
-    print("Syncing slash commands...")
-    try:
-        synced = await bot.tree.sync()
-        print(f"Synced {len(synced)} command(s).")
-    except Exception as e:
-        print(f"Failed to sync commands: {e}")
-    
-    # 确保所有cog都更新频道配置
-    try:
-        agent_manager = bot.get_cog("AgentManager")
-        if agent_manager:
-            print("正在更新所有cog的频道配置...")
-            agent_manager.reload_chat_channels()  # 重新加载配置
-            await agent_manager.update_all_cogs_channels()  # 更新所有cog的频道配置
-            print("所有cog的频道配置已更新")
-    except Exception as e:
-        print(f"Failed to update channel configs: {e}")
-        traceback.print_exc()
-    
-    print('-' * 50)
-
-# 添加一个帮助调试的命令，用于检查已加载的cog
-@bot.command(name="list_cogs")
-@commands.is_owner()
-async def list_cogs(ctx):
-    loaded_cogs = sorted(list(bot.cogs.keys()))
-    all_cogs = sorted([f.replace('.py', '') for f in os.listdir('cogs') if f.endswith('.py')])
-    
-    not_loaded = [cog for cog in all_cogs if cog not in [c.lower() for c in loaded_cogs]]
-    
-    await ctx.send(f"**已加载的Cogs ({len(loaded_cogs)}):**\n"
-                  f"{', '.join(loaded_cogs)}\n\n"
-                  f"**未加载的Cogs ({len(not_loaded)}):**\n"
-                  f"{', '.join(not_loaded)}")
-
-# 保持Space活跃的函数
-def keep_alive():
-    """定期自我请求以保持Space活跃"""
-    print("启动保活线程...")
-    while True:
-        try:
-            # 获取SPACE_HOST环境变量，这是HF Space自动提供的
-            space_host = os.environ.get('SPACE_HOST')
-            # 如果没有SPACE_HOST环境变量，说明不是在HF Space上运行
-            if not space_host:
-                # 本地调试环境
-                url = "http://127.0.0.1:5000"
-            else:
-                # HF Space环境
-                url = f"https://{space_host}"
-            
-            requests.get(url, timeout=10)
-            print(f"[KeepAlive] 自我请求成功: {time.strftime('%Y-%m-%d %H:%M:%S')}")
-        except Exception as e:
-            print(f"[KeepAlive] 自我请求失败: {e}")
-        
-        # 每30分钟请求一次，避免Space超时
-        time.sleep(1800)
-
-def start_flask():
-    # 创建默认头像
-    create_default_avatar()
-    
-    # 启动保活线程
-    keep_alive_thread = threading.Thread(target=keep_alive)
-    keep_alive_thread.daemon = True
-    keep_alive_thread.start()
-    
-    # 获取PORT环境变量，默认7860（HF Space默认端口）
-    # 如果环境变量不存在，就使用5000端口（本地开发环境）
-    port = int(os.environ.get('PORT', 5000))
-    host = '0.0.0.0' if os.environ.get('SPACE_HOST') else '127.0.0.1'
-    
-    app.run(host=host, port=port)
-
-async def main():
-    token = config.get("token")
-    try:
-        print("Loading cogs...")
-        
-        agent_manager_instance = None
-        # 先加载AgentManager cog
-        try:
-            await bot.load_extension("cogs.agent_manager")
-            agent_manager_instance = bot.get_cog("AgentManager")
-            if agent_manager_instance:
-                print(cpr.success("AgentManager cog loaded successfully."))
-            else:
-                print(cpr.error("Failed to get AgentManager instance after loading."))
-        except Exception as e:
-            print(f"Failed to load AgentManager: {e}")
-            traceback.print_exc()
-        
-        # 如果AgentManager加载失败，可能需要停止
-        if not agent_manager_instance:
-            print(cpr.critical("AgentManager failed to load. Cannot proceed."))
-            return
-            
-        # 然后加载其他cog，并传入 agent_manager_instance
-        for file in os.listdir("cogs"):
-            if file.endswith(".py") and file != "agent_manager.py" and file != "gemini_backup.py":
-                try:
-                    # 尝试将 agent_manager 传递给其他 cog 的 setup 函数
-                    # 注意：这要求其他 cog 的 setup 函数能够接受 agent_manager 参数
-                    # 或者修改 cog 的 __init__ 来接收
-                    await bot.load_extension(f"cogs.{file[:-3]}")
-                    # 获取刚加载的 cog 实例
-                    cog_instance = bot.get_cog(file[:-3].capitalize()) # 假设 Cog 类名首字母大写
-                    if cog_instance and hasattr(cog_instance, 'set_agent_manager'):
-                        cog_instance.set_agent_manager(agent_manager_instance)
-                        print(cpr.info(f"Passed AgentManager to {file[:-3].capitalize()}"))
-                        
-                except commands.ExtensionAlreadyLoaded:
-                    print(cpr.warning(f"Extension {file[:-3]} already loaded."))
-                except Exception as e:
-                    print(f"Failed to load or setup extension {file}: {e}")
-                    traceback.print_exc()
-        
-        print("Starting bot...")
-        # Create custom aiohttp session
-        connector = aiohttp.TCPConnector(ssl=ssl_context)
-        session = aiohttp.ClientSession(connector=connector)
-        
-        # Use custom session
-        bot.http.session = session
-        
-        # 启动Flask应用
-        flask_thread = threading.Thread(target=start_flask)
-        flask_thread.daemon = True  # 主程序退出时，Flask线程也会退出
-        flask_thread.start()
-        
-        # 是否在Hugging Face Space环境下，不打开浏览器
-        if not os.environ.get('SPACE_HOST'):
-            webbrowser.open('http://127.0.0.1:5000')
-        
-        await bot.start(token)
-    except Exception as e:
-        print(f"An error occurred: {e}")
-        traceback.print_exc()
-    finally:
-        await bot.close()
-
-
-if __name__ == "__main__":
-    try:
-        asyncio.run(main())
-    except KeyboardInterrupt:
-        print("Bot has been shut down.")
-    except Exception as e:
-        print(f"Fatal error: {e}")
+import asyncio
+import json
+from discord.ext import commands
+import discord
+import os
+import ssl
+import sys
+import traceback
+import aiohttp
+import logging
+import threading
+import webbrowser
+import time
+import requests
+from flask import Flask, render_template, request, jsonify, redirect, url_for, send_from_directory, session
+from werkzeug.utils import secure_filename
+from PIL import Image, ImageDraw, ImageFont
+import io
+import shutil
+from utils.config import config
+from functools import wraps
+
+# 配置Flask应用
+app = Flask(__name__, 
+            static_folder='static',
+            template_folder='templates')
+app.secret_key = os.urandom(24)  # 设置session密钥
+
+# 登录验证装饰器
+def login_required(f):
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        if not session.get('logged_in'):
+            return redirect(url_for('login'))
+        return f(*args, **kwargs)
+    return decorated_function
+
+# 配置日志级别，过滤掉 google_genai.models 的 INFO 级别日志
+logging.getLogger('google_genai.models').setLevel(logging.WARNING)
+
+# Disable SSL verification (only for debugging)
+ssl_context = ssl.create_default_context()
+ssl_context.check_hostname = False
+ssl_context.verify_mode = ssl.CERT_NONE
+
+intents = discord.Intents.default()
+intents.message_content = True
+intents.members = True
+
+bot = commands.Bot(command_prefix=".", intents=intents)
+
+# 创建默认头像
+def create_default_avatar():
+    avatar_path = os.path.join('static', 'img', 'default-avatar.png')
+    
+    # 如果已存在，则不需要创建
+    if os.path.exists(avatar_path):
+        return
+    
+    # 确保目录存在
+    os.makedirs(os.path.dirname(avatar_path), exist_ok=True)
+    
+    # 创建简单的默认头像
+    img = Image.new('RGB', (200, 200), color=(53, 102, 220))
+    d = ImageDraw.Draw(img)
+    
+    # 添加文字
+    try:
+        # 尝试使用系统字体
+        font = ImageFont.truetype("arial.ttf", 60)
+    except:
+        # 如果没有系统字体，使用默认字体
+        font = ImageFont.load_default()
+    
+    d.text((70, 70), "Bot", fill=(255, 255, 255), font=font)
+    
+    # 保存图片
+    img.save(avatar_path)
+
+# Flask路由
+@app.route('/login')
+def login():
+    if session.get('logged_in'):
+        return redirect(url_for('index'))
+    return render_template('login.html')
+
+@app.route('/api/login', methods=['POST'])
+def api_login():
+    data = request.json
+    password = data.get('password')
+    
+    # 直接从config.json读取最新密码
+    try:
+        with open('config.json', 'r', encoding='utf-8') as f:
+            config_data = json.load(f)
+            config_password = config_data.get('password')
+    except:
+        config_password = None
+    
+    if not config_password:
+        # 如果配置中没有密码，设置默认密码为 "admin"
+        config_password = "admin"
+        config.write("password", config_password)
+    
+    if password == config_password:
+        session['logged_in'] = True
+        return jsonify({"status": "success"})
+    return jsonify({"status": "error", "message": "密码错误"})
+
+@app.route('/logout')
+def logout():
+    session.pop('logged_in', None)
+    return redirect(url_for('login'))
+
+@app.route('/')
+@login_required
+def index():
+    return render_template('index.html')
+
+@app.route('/config')
+@login_required
+def config_page():
+    return render_template('config.html')
+
+@app.route('/presets')
+@login_required
+def presets_page():
+    presets = os.listdir('agent/presets')
+    return render_template('presets.html', presets=presets)
+
+@app.route('/api/config', methods=['GET'])
+@login_required
+def get_config():
+    with open('config.json', 'r', encoding='utf-8') as f:
+        config_data = json.load(f)
+    return jsonify(config_data)
+
+@app.route('/api/config', methods=['POST'])
+@login_required
+def update_config():
+    config_data = request.json
+    
+    # 检查密码是否被修改
+    current_password = config.get('password')
+    password_changed = current_password != config_data.get('password')
+    
+    # 保存配置
+    with open('config.json', 'w', encoding='utf-8') as f:
+        json.dump(config_data, f, indent=4, ensure_ascii=False)
+    
+    # 如果密码被修改,返回特殊状态码
+    if password_changed:
+        return jsonify({"status": "password_changed"})
+    return jsonify({"status": "success"})
+
+@app.route('/api/presets', methods=['GET'])
+def get_presets():
+    presets = os.listdir('agent/presets')
+    presets_data = {}
+    for preset in presets:
+        preset_path = os.path.join('agent/presets', preset)
+        if os.path.isdir(preset_path):
+            presets_data[preset] = {}
+            for file in os.listdir(preset_path):
+                if file.endswith('.json'):
+                    with open(os.path.join(preset_path, file), 'r', encoding='utf-8') as f:
+                        presets_data[preset][file] = json.load(f)
+    return jsonify(presets_data)
+
+@app.route('/api/preset/<preset_name>', methods=['GET'])
+def get_preset(preset_name):
+    preset_path = os.path.join('agent/presets', preset_name)
+    if not os.path.exists(preset_path):
+        return jsonify({"status": "error", "message": "Preset not found"}), 404
+    
+    preset_data = {}
+    for file in os.listdir(preset_path):
+        if file.endswith('.json'):
+            with open(os.path.join(preset_path, file), 'r', encoding='utf-8') as f:
+                preset_data[file] = json.load(f)
+    
+    return jsonify(preset_data)
+
+@app.route('/api/preset/<preset_name>', methods=['PUT'])
+def update_preset(preset_name):
+    preset_data = request.json
+    preset_path = os.path.join('agent/presets', preset_name)
+    if not os.path.exists(preset_path):
+        return jsonify({"status": "error", "message": "Preset not found"}), 404
+    
+    for file_name, data in preset_data.items():
+        with open(os.path.join(preset_path, file_name), 'w', encoding='utf-8') as f:
+            json.dump(data, f, indent=4, ensure_ascii=False)
+    
+    return jsonify({"status": "success"})
+
+@app.route('/api/preset/<preset_name>', methods=['DELETE'])
+def delete_preset(preset_name):
+    preset_path = os.path.join('agent/presets', preset_name)
+    if not os.path.exists(preset_path):
+        return jsonify({"status": "error", "message": "Preset not found"}), 404
+    
+    shutil.rmtree(preset_path)
+    return jsonify({"status": "success"})
+
+@app.route('/api/preset', methods=['POST'])
+def create_preset():
+    data = request.json
+    preset_name = data.get('name')
+    template_preset = data.get('template', 'default')
+    
+    if not preset_name:
+        return jsonify({"status": "error", "message": "Preset name is required"}), 400
+    
+    new_preset_path = os.path.join('agent/presets', preset_name)
+    if os.path.exists(new_preset_path):
+        return jsonify({"status": "error", "message": "Preset already exists"}), 400
+    
+    # 复制模板预设
+    template_path = os.path.join('agent/presets', template_preset)
+    if not os.path.exists(template_path):
+        return jsonify({"status": "error", "message": "Template preset not found"}), 404
+    
+    shutil.copytree(template_path, new_preset_path)
+    
+    return jsonify({"status": "success"})
+
+@app.route('/api/preset/<preset_name>/avatar', methods=['POST'])
+def upload_avatar(preset_name):
+    preset_path = os.path.join('agent/presets', preset_name)
+    if not os.path.exists(preset_path):
+        return jsonify({"status": "error", "message": "Preset not found"}), 404
+    
+    if 'avatar' not in request.files:
+        return jsonify({"status": "error", "message": "No file part"}), 400
+    
+    file = request.files['avatar']
+    if file.filename == '':
+        return jsonify({"status": "error", "message": "No selected file"}), 400
+    
+    # 处理图片
+    try:
+        img = Image.open(file)
+        
+        # 调整为正方形
+        width, height = img.size
+        size = min(width, height)
+        left = (width - size) // 2
+        top = (height - size) // 2
+        right = left + size
+        bottom = top + size
+        img = img.crop((left, top, right, bottom))
+        
+        # 保存图片
+        for old_avatar in os.listdir(preset_path):
+            if old_avatar.startswith('avatar.'):
+                os.remove(os.path.join(preset_path, old_avatar))
+        
+        # 获取文件扩展名
+        ext = os.path.splitext(file.filename)[1].lower()
+        if ext not in ['.jpg', '.jpeg', '.png', '.gif']:
+            ext = '.jpg'  # 默认使用jpg
+        
+        img.save(os.path.join(preset_path, f'avatar{ext}'))
+        
+        return jsonify({"status": "success"})
+    except Exception as e:
+        return jsonify({"status": "error", "message": str(e)}), 500
+
+@app.route('/agent/presets/<path:filename>')
+def preset_files(filename):
+    return send_from_directory('agent/presets', filename)
+
+@bot.event
+async def on_ready():
+    print(f'Bot logged in as {bot.user.name} (ID: {bot.user.id})')
+    print(f'Bot invite link: https://discord.com/oauth2/authorize?client_id={bot.user.id}&permissions=8&scope=bot')
+    
+    # Auto sync slash commands
+    print("Syncing slash commands...")
+    try:
+        synced = await bot.tree.sync()
+        print(f"Synced {len(synced)} command(s).")
+    except Exception as e:
+        print(f"Failed to sync commands: {e}")
+    
+    # 确保所有cog都更新频道配置
+    try:
+        agent_manager = bot.get_cog("AgentManager")
+        if agent_manager:
+            print("正在更新所有cog的频道配置...")
+            agent_manager.reload_chat_channels()  # 重新加载配置
+            await agent_manager.update_all_cogs_channels()  # 更新所有cog的频道配置
+            print("所有cog的频道配置已更新")
+    except Exception as e:
+        print(f"Failed to update channel configs: {e}")
+        traceback.print_exc()
+    
+    print('-' * 50)
+
+# 添加一个帮助调试的命令，用于检查已加载的cog
+@bot.command(name="list_cogs")
+@commands.is_owner()
+async def list_cogs(ctx):
+    loaded_cogs = sorted(list(bot.cogs.keys()))
+    all_cogs = sorted([f.replace('.py', '') for f in os.listdir('cogs') if f.endswith('.py')])
+    
+    not_loaded = [cog for cog in all_cogs if cog not in [c.lower() for c in loaded_cogs]]
+    
+    await ctx.send(f"**已加载的Cogs ({len(loaded_cogs)}):**\n"
+                  f"{', '.join(loaded_cogs)}\n\n"
+                  f"**未加载的Cogs ({len(not_loaded)}):**\n"
+                  f"{', '.join(not_loaded)}")
+
+# 保持Space活跃的函数
+def keep_alive():
+    """定期自我请求以保持Space活跃"""
+    print("启动保活线程...")
+    while True:
+        try:
+            # 获取SPACE_HOST环境变量，这是HF Space自动提供的
+            space_host = os.environ.get('SPACE_HOST')
+            # 如果没有SPACE_HOST环境变量，说明不是在HF Space上运行
+            if not space_host:
+                # 本地调试环境
+                url = "http://127.0.0.1:5000"
+            else:
+                # HF Space环境
+                url = f"https://{space_host}"
+            
+            requests.get(url, timeout=10)
+            print(f"[KeepAlive] 自我请求成功: {time.strftime('%Y-%m-%d %H:%M:%S')}")
+        except Exception as e:
+            print(f"[KeepAlive] 自我请求失败: {e}")
+        
+        # 每30分钟请求一次，避免Space超时
+        time.sleep(1800)
+
+def start_flask():
+    # 创建默认头像
+    create_default_avatar()
+    
+    # 启动保活线程
+    keep_alive_thread = threading.Thread(target=keep_alive)
+    keep_alive_thread.daemon = True
+    keep_alive_thread.start()
+    
+    # 获取PORT环境变量，默认7860（HF Space默认端口）
+    # 如果环境变量不存在，就使用5000端口（本地开发环境）
+    port = int(os.environ.get('PORT', 5000))
+    host = '0.0.0.0' if os.environ.get('SPACE_HOST') else '127.0.0.1'
+    
+    app.run(host=host, port=port)
+
+async def main():
+    token = config.get("token")
+    try:
+        print("Loading cogs...")
+        
+        # 先加载AgentManager cog
+        try:
+            await bot.load_extension("cogs.agent_manager")
+        except Exception as e:
+            print(f"Failed to load AgentManager: {e}")
+            traceback.print_exc()
+        
+        # 然后加载其他cog
+        for file in os.listdir("cogs"):
+            if file.endswith(".py") and file != "agent_manager.py" and file != "gemini_backup.py":
+                try:
+                    await bot.load_extension(f"cogs.{file[:-3]}")
+                except Exception as e:
+                    print(f"Failed to load extension {file}: {e}")
+                    traceback.print_exc()
+        
+        print("Starting bot...")
+        # Create custom aiohttp session
+        connector = aiohttp.TCPConnector(ssl=ssl_context)
+        session = aiohttp.ClientSession(connector=connector)
+        
+        # Use custom session
+        bot.http.session = session
+        
+        # 启动Flask应用
+        flask_thread = threading.Thread(target=start_flask)
+        flask_thread.daemon = True  # 主程序退出时，Flask线程也会退出
+        flask_thread.start()
+        
+        # 是否在Hugging Face Space环境下，不打开浏览器
+        if not os.environ.get('SPACE_HOST'):
+            webbrowser.open('http://127.0.0.1:5000')
+        
+        await bot.start(token)
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        traceback.print_exc()
+    finally:
+        await bot.close()
+
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("Bot has been shut down.")
+    except Exception as e:
+        print(f"Fatal error: {e}")
         traceback.print_exc() 

static/js/presets.js

@@ -204,8 +204,8 @@ function fillPresetForms(data) {
         document.getElementById('chat_system_prompt').value = chatPreset.system_prompt || '';
         document.getElementById('chat_first_user_message').value = chatPreset.first_user_message || '';
         document.getElementById('chat_main_content').value = chatPreset.main_content || '';
-        document.getElementById('chat_last_user_message').value = chatPreset.last_user_message || '';
-        document.getElementById('chat_prefill_assistant').checked = chatPreset.prefill_assistant || false;
+        document.getElementById('chat_last_message').value = chatPreset.last_message || '';
+        document.getElementById('chat_prefill_assistant_reply').checked = chatPreset.prefill_assistant_reply || false;
     }
     
     // 翻译预设
@@ -214,7 +214,8 @@ function fillPresetForms(data) {
         document.getElementById('translate_system_prompt').value = translatePreset.system_prompt || '';
         document.getElementById('translate_first_user_message').value = translatePreset.first_user_message || '';
         document.getElementById('translate_main_content').value = translatePreset.main_content || '';
-        document.getElementById('translate_last_user_message').value = translatePreset.last_user_message || '';
+        document.getElementById('translate_last_message').value = translatePreset.last_message || '';
+        document.getElementById('translate_prefill_assistant_reply').checked = translatePreset.prefill_assistant_reply || false;
     }
     
     // 附件预设
@@ -223,7 +224,8 @@ function fillPresetForms(data) {
         document.getElementById('attachment_system_prompt').value = attachmentPreset.system_prompt || '';
         document.getElementById('attachment_first_user_message').value = attachmentPreset.first_user_message || '';
         document.getElementById('attachment_main_content').value = attachmentPreset.main_content || '';
-        document.getElementById('attachment_last_user_message').value = attachmentPreset.last_user_message || '';
+        document.getElementById('attachment_last_message').value = attachmentPreset.last_message || '';
+        document.getElementById('attachment_prefill_assistant_reply').checked = attachmentPreset.prefill_assistant_reply || false;
     }
     
     // 引用预设
@@ -232,7 +234,8 @@ function fillPresetForms(data) {
         document.getElementById('reference_system_prompt').value = referencePreset.system_prompt || '';
         document.getElementById('reference_first_user_message').value = referencePreset.first_user_message || '';
         document.getElementById('reference_main_content').value = referencePreset.main_content || '';
-        document.getElementById('reference_last_user_message').value = referencePreset.last_user_message || '';
+        document.getElementById('reference_last_message').value = referencePreset.last_message || '';
+        document.getElementById('reference_prefill_assistant_reply').checked = referencePreset.prefill_assistant_reply || false;
     }
     
     // Gemini 设置
@@ -297,8 +300,8 @@ function collectFormData() {
         system_prompt: document.getElementById('chat_system_prompt').value,
         first_user_message: document.getElementById('chat_first_user_message').value,
         main_content: document.getElementById('chat_main_content').value,
-        last_user_message: document.getElementById('chat_last_user_message').value,
-        prefill_assistant: document.getElementById('chat_prefill_assistant').checked
+        last_message: document.getElementById('chat_last_message').value,
+        prefill_assistant_reply: document.getElementById('chat_prefill_assistant_reply').checked
     };
     
     // 翻译预设
@@ -306,7 +309,8 @@ function collectFormData() {
         system_prompt: document.getElementById('translate_system_prompt').value,
         first_user_message: document.getElementById('translate_first_user_message').value,
         main_content: document.getElementById('translate_main_content').value,
-        last_user_message: document.getElementById('translate_last_user_message').value
+        last_message: document.getElementById('translate_last_message').value,
+        prefill_assistant_reply: document.getElementById('translate_prefill_assistant_reply').checked
     };
     
     // 附件预设
@@ -314,7 +318,8 @@ function collectFormData() {
         system_prompt: document.getElementById('attachment_system_prompt').value,
         first_user_message: document.getElementById('attachment_first_user_message').value,
         main_content: document.getElementById('attachment_main_content').value,
-        last_user_message: document.getElementById('attachment_last_user_message').value
+        last_message: document.getElementById('attachment_last_message').value,
+        prefill_assistant_reply: document.getElementById('attachment_prefill_assistant_reply').checked
     };
     
     // 引用预设
@@ -322,7 +327,8 @@ function collectFormData() {
         system_prompt: document.getElementById('reference_system_prompt').value,
         first_user_message: document.getElementById('reference_first_user_message').value,
         main_content: document.getElementById('reference_main_content').value,
-        last_user_message: document.getElementById('reference_last_user_message').value
+        last_message: document.getElementById('reference_last_message').value,
+        prefill_assistant_reply: document.getElementById('reference_prefill_assistant_reply').checked
     };
     
     // Gemini 设置

templates/presets.html

@@ -123,15 +123,12 @@
                                                     <textarea class="form-control" id="chat_main_content" rows="10"></textarea>
                                                 </div>
                                                 <div class="mb-3">
-                                                    <label for="chat_last_user_message" class="form-label">最后一条用户消息</label>
-                                                    <input type="text" class="form-control" id="chat_last_user_message">
+                                                    <label for="chat_last_message" class="form-label">最后一条消息</label>
+                                                    <input type="text" class="form-control" id="chat_last_message">
                                                 </div>
-                                                <div class="form-check mb-3">
-                                                    <input class="form-check-input" type="checkbox" id="chat_prefill_assistant">
-                                                    <label class="form-check-label" for="chat_prefill_assistant">
-                                                        将最后一条消息预填充为助手回应
-                                                    </label>
-                                                    <small class="form-text text-muted d-block">勾选后，在生成提示时，上下文中的最后一条消息会被视为机器人的回复。</small>
+                                                <div class="form-check form-switch mb-3">
+                                                    <input class="form-check-input" type="checkbox" role="switch" id="chat_prefill_assistant_reply">
+                                                    <label class="form-check-label" for="chat_prefill_assistant_reply">预填充模型回复 (最后一条消息的角色为'model')</label>
                                                 </div>
                                             </form>
                                         </div>
@@ -152,8 +149,12 @@
                                                     <textarea class="form-control" id="translate_main_content" rows="10"></textarea>
                                                 </div>
                                                 <div class="mb-3">
-                                                    <label for="translate_last_user_message" class="form-label">最后一条用户消息</label>
-                                                    <input type="text" class="form-control" id="translate_last_user_message">
+                                                    <label for="translate_last_message" class="form-label">最后一条消息</label>
+                                                    <input type="text" class="form-control" id="translate_last_message">
+                                                </div>
+                                                <div class="form-check form-switch mb-3">
+                                                    <input class="form-check-input" type="checkbox" role="switch" id="translate_prefill_assistant_reply">
+                                                    <label class="form-check-label" for="translate_prefill_assistant_reply">预填充模型回复 (最后一条消息的角色为'model')</label>
                                                 </div>
                                             </form>
                                         </div>
@@ -174,8 +175,12 @@
                                                     <textarea class="form-control" id="attachment_main_content" rows="10"></textarea>
                                                 </div>
                                                 <div class="mb-3">
-                                                    <label for="attachment_last_user_message" class="form-label">最后一条用户消息</label>
-                                                    <input type="text" class="form-control" id="attachment_last_user_message">
+                                                    <label for="attachment_last_message" class="form-label">最后一条消息</label>
+                                                    <input type="text" class="form-control" id="attachment_last_message">
+                                                </div>
+                                                <div class="form-check form-switch mb-3">
+                                                    <input class="form-check-input" type="checkbox" role="switch" id="attachment_prefill_assistant_reply">
+                                                    <label class="form-check-label" for="attachment_prefill_assistant_reply">预填充模型回复 (最后一条消息的角色为'model')</label>
                                                 </div>
                                             </form>
                                         </div>
@@ -196,8 +201,12 @@
                                                     <textarea class="form-control" id="reference_main_content" rows="10"></textarea>
                                                 </div>
                                                 <div class="mb-3">
-                                                    <label for="reference_last_user_message" class="form-label">最后一条用户消息</label>
-                                                    <input type="text" class="form-control" id="reference_last_user_message">
+                                                    <label for="reference_last_message" class="form-label">最后一条消息</label>
+                                                    <input type="text" class="form-control" id="reference_last_message">
+                                                </div>
+                                                <div class="form-check form-switch mb-3">
+                                                    <input class="form-check-input" type="checkbox" role="switch" id="reference_prefill_assistant_reply">
+                                                    <label class="form-check-label" for="reference_prefill_assistant_reply">预填充模型回复 (最后一条消息的角色为'model')</label>
                                                 </div>
                                             </form>
                                         </div>

utils/context_prompter.py

@@ -1,311 +1,301 @@
-import discord
-import pytz
-import re
-from discord.ext import commands
-from utils.func import get_time, now
-from typing import TYPE_CHECKING, Optional
-if TYPE_CHECKING:
-    from cogs.agent_manager import AgentManager
-
-
-class ContextPrompter:
-    def __init__(self, agent_manager: Optional['AgentManager'] = None):
-        self.tz = pytz.timezone("Asia/Shanghai")
-        self.agent_manager = agent_manager
-
-    def set_agent_manager(self, agent_manager: 'AgentManager'):
-        self.agent_manager = agent_manager
-
-    def set_tz(self, tz: str):
-        try:
-            self.tz = pytz.timezone(tz)
-        except Exception as e:
-            print(e)
-
-    def maltose_regex(self, context: str) -> str:
-        """
-        使用正则表达式预处理上下文内容，删除命令触发词
-        例如: .hey, .yo claude, .yoo 等，这些只是触发词对机器人理解没有用处
-        """
-        # 匹配并删除命令触发词
-        # 1. 匹配 .hey, .yo, .yoo 后跟空格和可能的参数
-        pattern1 = r'\.hey\s+|\.yo\s+[a-zA-Z]+\s+|\.yoo\s+'
-        # 2. 匹配独立的命令 .hey, .yo, .yoo (行首或前面有空格，后面是行尾或空格)
-        pattern2 = r'(^|\s)\.hey($|\s)|(^|\s)\.yo($|\s)|(^|\s)\.yoo($|\s)'
-        
-        # 应用正则替换
-        context = re.sub(pattern1, '', context)
-        context = re.sub(pattern2, '', context)
-        
-        return context.strip()
-
-    def get_msg_time(self, msg: discord.Message) -> str:
-        time = msg.created_at if msg.edited_at is None else msg.edited_at
-        return get_time(time, tz=self.tz)
-
-    async def get_context_for_prompt(
-        self,
-        ctx: commands.Context,
-        context_length: int,
-        before_message=None,
-        after_message=None,
-        after_message_context_length=0,
-        channel_id: int = None,
-        guild_id: int = None,
-    ):
-        context_msg = []
-        prefill_assistant = False
-        bot_display_name = ctx.me.display_name
-        bot_name = ctx.me.name
-        
-        if self.agent_manager and channel_id:
-            try:
-                preset_config = self.agent_manager.get_preset_json("chat_preset.json", str(channel_id), str(guild_id))
-                if preset_config:
-                    prefill_assistant = preset_config.get("prefill_assistant", False)
-            except Exception as e:
-                print(f"Error getting preset config for channel {channel_id}: {e}")
-                prefill_assistant = False
-
-        fetch_limit = context_length + 1
-        if prefill_assistant:
-            fetch_limit += 1
-            
-        history_messages = []
-        if before_message is not None and after_message is not None:
-            async for msg in ctx.channel.history(
-                limit=fetch_limit, before=before_message
-            ):
-                history_messages.append(msg)
-            history_messages.reverse()
-            history_messages.append(after_message)
-            async for msg in ctx.channel.history(
-                limit=after_message_context_length + 1, after=after_message
-            ):
-                history_messages.append(msg)
-        elif before_message is not None:
-            async for msg in ctx.channel.history(
-                limit=fetch_limit, before=before_message
-            ):
-                history_messages.append(msg)
-            history_messages.reverse()
-        elif after_message is not None:
-            async for msg in ctx.channel.history(
-                limit=context_length + 1, after=after_message
-            ):
-                history_messages.append(msg)
-        else:
-            async for msg in ctx.channel.history(
-                limit=fetch_limit, before=ctx.message
-            ):
-                history_messages.append(msg)
-            history_messages.reverse()
-            
-        num_messages = len(history_messages)
-        for i, msg in enumerate(history_messages):
-            is_last_message_for_prefill = prefill_assistant and (i == num_messages - 1)
-            
-            if is_last_message_for_prefill:
-                author_display_name = bot_display_name
-                author_name = bot_name
-            else:
-                author_display_name = msg.author.display_name
-                author_name = msg.author.name
-                
-            context_msg.append(
-                f"{author_display_name} ({author_name}) ({self.get_msg_time(msg)}): {msg.content}"
-            )
-
-        return "\n".join(context_msg)
-
-    def _get_template(self, template_name, channel_id=None, guild_id=None):
-        """获取模板内容，现在需要 guild_id"""
-        if self.agent_manager:
-            return self.agent_manager.get_preset_file(template_name, str(channel_id) if channel_id else None, str(guild_id) if guild_id else None)
-        else:
-            return None
-
-    async def chat_prompt(
-        self,
-        ctx: commands.Context,
-        context_length: int,
-        question: str,
-        name: str = None,
-    ):
-        context = await self.get_context_for_prompt(ctx, context_length, channel_id=ctx.channel.id, guild_id=ctx.guild.id)
-        context = self.maltose_regex(context)
-        name = name if name else ctx.me.display_name
-        
-        if self.agent_manager:
-            preset = self.agent_manager.get_preset_json("chat_preset.json", str(ctx.channel.id), str(ctx.guild.id))
-            if preset and "main_content" in preset:
-                main_content = preset["main_content"]
-                return main_content.format(
-                    context=context,
-                    question=question,
-                    name=name,
-                    bot_name=ctx.me.name,
-                    current_time=now(tz=self.tz),
-                    user_display_name=ctx.author.display_name,
-                    user_name=ctx.author.name
-                )
-        
-        prompt = f"""
-        <context>
-        {context}
-        </context>
-        <question>
-        {question}
-        </question>
-        You are {name} ({ctx.me.name}), chatting in a discord server.
-        Speak naturally like a human who talks, and don't use phrases like 'according to the context' since humans never talk like that. Remember the Language is Chinese unless the user specifies otherwise! Avoid explicitly mentioning someone's name. If you have to mention someone (try to avoid this case), use their display name (the name that appears outside the parentheses).
-        Now is {now(tz=self.tz)}.
-        {ctx.author.display_name} ({ctx.author.name}) is asking you a question (refer to `<question>`).
-        Consider the context in `<context>` and reply now. 
-        Avoid using ellipsis!
-        Your reply:
-        """
-        return prompt
-
-    async def chat_prompt_with_reference(
-        self,
-        ctx: commands.Context,
-        context_length: int,
-        after_message_context_length: int,
-        question: str,
-        reference: discord.Message,
-        name: str = None,
-    ):
-        context = await self.get_context_for_prompt(
-            ctx, context_length, reference, after_message_context_length=after_message_context_length, channel_id=ctx.channel.id, guild_id=ctx.guild.id
-        )
-        context = self.maltose_regex(context)
-        name = name if name else ctx.me.display_name
-        
-        if self.agent_manager:
-            preset = self.agent_manager.get_preset_json("reference_preset.json", str(ctx.channel.id), str(ctx.guild.id))
-            if preset and "main_content" in preset:
-                main_content = preset["main_content"]
-                return main_content.format(
-                    context=context,
-                    question=question,
-                    name=name,
-                    bot_name=ctx.me.name,
-                    current_time=now(tz=self.tz),
-                    user_display_name=ctx.author.display_name,
-                    user_name=ctx.author.name,
-                    reference_user_display_name=reference.author.display_name,
-                    reference_user_name=reference.author.name,
-                    reference_time=self.get_msg_time(reference),
-                    reference_content=reference.content
-                )
-        
-        prompt = f"""
-        <context>
-        {context}
-        </context>
-        <question>
-        {question}
-        </question>
-        <reference>
-        {reference.author.display_name} ({reference.author.name}) ({self.get_msg_time(reference)}): {reference.content}
-        </reference>
-        You are {name} ({ctx.me.name}), chatting in a discord server.
-        Speak naturally like a human who talks, and don't use phrases like 'according to the context' since humans never talk like that. Remember the Language is Chinese unless the user specifies otherwise! Avoid explicitly mentioning someone's name. If you have to mention someone (try to avoid this case), use their display name (the name that appears outside the parentheses).
-        Now is {now(tz=self.tz)}.
-        {ctx.author.display_name} ({ctx.author.name}) is asking you a question (refer to `<question>`) about the message above (refer to `<reference>`).
-        Consider the context in `<context>` and reply now.
-        Avoid using ellipsis!
-        Your reply:
-        """
-        return prompt
-
-    async def chat_prompt_with_attachment(
-        self,
-        ctx: commands.Context,
-        question: str,
-        reference: discord.Message,
-    ):
-        content = reference.content
-        if content == "":
-            content = "[No content, only attachments]"
-            
-        template = self._get_template("chat_prompt_with_attachment.txt", ctx.channel.id, ctx.guild.id)
-        if template:
-            return template.format(
-                question=question,
-                name=ctx.me.display_name,
-                bot_name=ctx.me.name,
-                current_time=now(tz=self.tz),
-                user_display_name=ctx.author.display_name,
-                user_name=ctx.author.name,
-                reference_user_display_name=reference.author.display_name,
-                reference_user_name=reference.author.name,
-                reference_time=self.get_msg_time(reference),
-                reference_content=content
-            )
-            
-        prompt: str = f"""
-        <question>
-        {question}
-        </question>
-        <reference>
-        {reference.author.display_name} ({reference.author.name}) ({self.get_msg_time(reference)}): {content}
-        </reference>
-        You are {ctx.me.display_name} ({ctx.me.name}), chatting in a discord server.
-        Speak naturally like a human who talks, and don't use phrases like 'according to the context' since humans never talk like that. Remember the Language is Chinese unless the user specifies otherwise! Avoid explicitly mentioning someone's name. If you have to mention someone (try to avoid this case), use their display name (the name that appears outside the parentheses).
-        Now is {now(tz=self.tz)}.
-        {ctx.author.display_name} ({ctx.author.name}) is asking you a question (refer to `<question>`) about the message (refer to `<reference>`) with the ATTACHMENT FILE.
-        Analyze the attachment file and reply now.
-        Avoid using ellipsis!
-        Your reply:
-        """
-        return prompt
-
-    async def translate_prompt(
-        self,
-        ctx: commands.Context,
-        context_length: int,
-        reference: discord.Message,
-        after_message_context_length: int,
-        target_language: str,
-    ):
-        context = await self.get_context_for_prompt(
-            ctx, context_length, reference, after_message_context_length=after_message_context_length, channel_id=ctx.channel.id, guild_id=ctx.guild.id
-        )
-        context = self.maltose_regex(context)
-        
-        template = self._get_template("translate_prompt.txt", ctx.channel.id, ctx.guild.id)
-        if template:
-            return template.format(
-                context=context,
-                target_language=target_language,
-                reference_content=reference.content,
-                bot_name=ctx.me.name,
-                current_time=now(tz=self.tz),
-                user_display_name=ctx.author.display_name,
-                user_name=ctx.author.name,
-                reference_user_display_name=reference.author.display_name,
-                reference_user_name=reference.author.name,
-                reference_time=self.get_msg_time(reference)
-            )
-
-        prompt = f"""
-        <context>
-        {context}
-        </context>
-        `<reference>` is the message you need to translate.
-        <reference>
-        {reference.content}
-        </reference>
-        This message is from `<author>`.
-        <author>
-        {reference.author.display_name} ({reference.author.name}) ({self.get_msg_time(reference)})
-        </author>
-        You are a skilled muti-lingual translator, currently doing a translation job in a discord server. You'll get a message which you need to translate into {target_language} with context. You only need to supply the translation according to the context without any additional information. Don't act like a machine, translate smoothly like a human without being too informal. 
-        Your translation should not include the author's name and the time.
-        Now is {now(tz=self.tz)}.
-        {ctx.author.display_name} ({ctx.author.name}) is asking you to translate the message in `<reference>` into {target_language} under the context (refer to <context>). The message is from `<author>`, so consider the context and try to understand the message before translating.
-        Your translation:
-        """
-        return prompt
+import discord
+import pytz
+import re
+from discord.ext import commands
+from utils.func import get_time, now
+
+
+class ContextPrompter:
+    def __init__(self):
+        self.tz = pytz.timezone("Asia/Shanghai")
+        self.agent_manager = None
+
+    def set_agent_manager(self, agent_manager):
+        self.agent_manager = agent_manager
+
+    def set_tz(self, tz: str):
+        try:
+            self.tz = pytz.timezone(tz)
+        except Exception as e:
+            print(e)
+
+    def maltose_regex(self, context: str) -> str:
+        """
+        使用正则表达式预处理上下文内容，删除命令触发词
+        例如: .hey, .yo claude, .yoo 等，这些只是触发词对机器人理解没有用处
+        """
+        # 匹配并删除命令触发词
+        # 1. 匹配 .hey, .yo, .yoo 后跟空格和可能的参数
+        pattern1 = r'\.hey\s+|\.yo\s+[a-zA-Z]+\s+|\.yoo\s+'
+        # 2. 匹配独立的命令 .hey, .yo, .yoo (行首或前面有空格，后面是行尾或空格)
+        pattern2 = r'(^|\s)\.hey($|\s)|(^|\s)\.yo($|\s)|(^|\s)\.yoo($|\s)'
+        
+        # 应用正则替换
+        context = re.sub(pattern1, '', context)
+        context = re.sub(pattern2, '', context)
+        
+        return context.strip()
+
+    def get_msg_time(self, msg: discord.Message) -> str:
+        time = msg.created_at if msg.edited_at is None else msg.edited_at
+        return get_time(time, tz=self.tz)
+
+    async def get_context_for_prompt(
+        self,
+        ctx: commands.Context,
+        context_length: int,
+        before_message=None,
+        after_message=None,
+        after_message_context_length=0,
+    ):
+        context_msg = []
+        if before_message is not None and after_message is not None:
+            async for msg in ctx.channel.history(
+                limit=context_length + 1, before=before_message
+            ):
+                context_msg.append(
+                    f"{msg.author.display_name} ({msg.author.name}) ({self.get_msg_time(msg)}): {msg.content}"
+                )
+            context_msg.reverse()
+            context_msg.append(
+                f"{after_message.author.display_name} ({after_message.author.name}) ({self.get_msg_time(after_message)}): {after_message.content}"
+            )
+            async for msg in ctx.channel.history(
+                limit=after_message_context_length + 1, after=after_message
+            ):
+                context_msg.append(
+                    f"{msg.author.display_name} ({msg.author.name}) ({self.get_msg_time(msg)}): {msg.content}"
+                )
+        elif before_message is not None:
+            async for msg in ctx.channel.history(
+                limit=context_length + 1, before=before_message
+            ):
+                context_msg.append(
+                    f"{msg.author.display_name} ({msg.author.name}) ({self.get_msg_time(msg)}): {msg.content}"
+                )
+            context_msg.reverse()
+        elif after_message is not None:
+            async for msg in ctx.channel.history(
+                limit=context_length + 1, after=after_message
+            ):
+                context_msg.append(
+                    f"{msg.author.display_name} ({msg.author.name}) ({self.get_msg_time(msg)}): {msg.content}"
+                )
+        else:
+            async for msg in ctx.channel.history(
+                limit=context_length + 1, before=ctx.message
+            ):
+                context_msg.append(
+                    f"{msg.author.display_name} ({msg.author.name}) ({self.get_msg_time(msg)}): {msg.content}"
+                )
+            context_msg.reverse()
+        return "\n".join(context_msg)
+
+    def _get_template(self, template_name, channel_id=None):
+        """获取模板内容"""
+        if self.agent_manager:
+            return self.agent_manager.get_preset_file(template_name, channel_id)
+        else:
+            # 如果没有设置agent_manager，返回默认模板
+            return None
+
+    async def chat_prompt(
+        self,
+        ctx: commands.Context,
+        context_length: int,
+        question: str,
+        name: str = None,
+    ):
+        context = await self.get_context_for_prompt(ctx, context_length)
+        # 应用正则预处理
+        context = self.maltose_regex(context)
+        name = name if name else ctx.me.display_name
+        
+        # 获取预设模板
+        if self.agent_manager:
+            # 获取预设JSON
+            preset = self.agent_manager.get_preset_json("chat_preset.json", ctx.channel.id)
+            if preset and "main_content" in preset:
+                # 使用预设中的main_content，并替换其中的模板变量
+                main_content = preset["main_content"]
+                return main_content.format(
+                    context=context,
+                    question=question,
+                    name=name,
+                    bot_name=ctx.me.name,
+                    current_time=now(tz=self.tz),
+                    user_display_name=ctx.author.display_name,
+                    user_name=ctx.author.name
+                )
+        
+        # 回退到原始模板
+        prompt = f"""
+        <context>
+        {context}
+        </context>
+        <question>
+        {question}
+        </question>
+        You are {name} ({ctx.me.name}), chatting in a discord server.
+        Speak naturally like a human who talks, and don't use phrases like 'according to the context' since humans never talk like that. Remember the Language is Chinese unless the user specifies otherwise! Avoid explicitly mentioning someone's name. If you have to mention someone (try to avoid this case), use their display name (the name that appears outside the parentheses).
+        Now is {now(tz=self.tz)}.
+        {ctx.author.display_name} ({ctx.author.name}) is asking you a question (refer to `<question>`).
+        Consider the context in `<context>` and reply now. 
+        Avoid using ellipsis!
+        Your reply:
+        """
+        return prompt
+
+    async def chat_prompt_with_reference(
+        self,
+        ctx: commands.Context,
+        context_length: int,
+        after_message_context_length: int,
+        question: str,
+        reference: discord.Message,
+        name: str = None,
+    ):
+        context = await self.get_context_for_prompt(
+            ctx, context_length, reference, after_message_context_length=after_message_context_length
+        )
+        # 应用正则预处理
+        context = self.maltose_regex(context)
+        name = name if name else ctx.me.display_name
+        
+        # 获取预设模板
+        if self.agent_manager:
+            # 获取预设JSON
+            preset = self.agent_manager.get_preset_json("reference_preset.json", ctx.channel.id)
+            if preset and "main_content" in preset:
+                # 使用预设中的main_content，并替换其中的模板变量
+                main_content = preset["main_content"]
+                return main_content.format(
+                    context=context,
+                    question=question,
+                    name=name,
+                    bot_name=ctx.me.name,
+                    current_time=now(tz=self.tz),
+                    user_display_name=ctx.author.display_name,
+                    user_name=ctx.author.name,
+                    reference_user_display_name=reference.author.display_name,
+                    reference_user_name=reference.author.name,
+                    reference_time=self.get_msg_time(reference),
+                    reference_content=reference.content
+                )
+        
+        # 回退到原始模板
+        prompt = f"""
+        <context>
+        {context}
+        </context>
+        <question>
+        {question}
+        </question>
+        <reference>
+        {reference.author.display_name} ({reference.author.name}) ({self.get_msg_time(reference)}): {reference.content}
+        </reference>
+        You are {name} ({ctx.me.name}), chatting in a discord server.
+        Speak naturally like a human who talks, and don't use phrases like 'according to the context' since humans never talk like that. Remember the Language is Chinese unless the user specifies otherwise! Avoid explicitly mentioning someone's name. If you have to mention someone (try to avoid this case), use their display name (the name that appears outside the parentheses).
+        Now is {now(tz=self.tz)}.
+        {ctx.author.display_name} ({ctx.author.name}) is asking you a question (refer to `<question>`) about the message above (refer to `<reference>`).
+        Consider the context in `<context>` and reply now.
+        Avoid using ellipsis!
+        Your reply:
+        """
+        return prompt
+
+    async def chat_prompt_with_attachment(
+        self,
+        ctx: commands.Context,
+        question: str,
+        reference: discord.Message,
+    ):
+        content = reference.content
+        if content == "":
+            content = "[No content, only attachments]"
+            
+        # 使用模板
+        template = self._get_template("chat_prompt_with_attachment.txt", ctx.channel.id)
+        if template:
+            return template.format(
+                question=question,
+                name=ctx.me.display_name,
+                bot_name=ctx.me.name,
+                current_time=now(tz=self.tz),
+                user_display_name=ctx.author.display_name,
+                user_name=ctx.author.name,
+                reference_user_display_name=reference.author.display_name,
+                reference_user_name=reference.author.name,
+                reference_time=self.get_msg_time(reference),
+                reference_content=content
+            )
+            
+        # 回退到原始模板
+        prompt: str = f"""
+        <question>
+        {question}
+        </question>
+        <reference>
+        {reference.author.display_name} ({reference.author.name}) ({self.get_msg_time(reference)}): {content}
+        </reference>
+        You are {ctx.me.display_name} ({ctx.me.name}), chatting in a discord server.
+        Speak naturally like a human who talks, and don't use phrases like 'according to the context' since humans never talk like that. Remember the Language is Chinese unless the user specifies otherwise! Avoid explicitly mentioning someone's name. If you have to mention someone (try to avoid this case), use their display name (the name that appears outside the parentheses).
+        Now is {now(tz=self.tz)}.
+        {ctx.author.display_name} ({ctx.author.name}) is asking you a question (refer to `<question>`) about the message (refer to `<reference>`) with the ATTACHMENT FILE.
+        Analyze the attachment file and reply now.
+        Avoid using ellipsis!
+        Your reply:
+        """
+        return prompt
+
+    async def translate_prompt(
+        self,
+        ctx: commands.Context,
+        context_length: int,
+        reference: discord.Message,
+        after_message_context_length: int,
+        target_language: str,
+    ):
+        # 获取上下文内容
+        context = await self.get_context_for_prompt(
+            ctx, context_length, reference, after_message_context_length=after_message_context_length
+        )
+        # 应用正则预处理
+        context = self.maltose_regex(context)
+        
+        # 使用模板
+        template = self._get_template("translate_prompt.txt", ctx.channel.id)
+        if template:
+            return template.format(
+                context=context,
+                target_language=target_language,
+                reference_content=reference.content,
+                bot_name=ctx.me.name,
+                current_time=now(tz=self.tz),
+                user_display_name=ctx.author.display_name,
+                user_name=ctx.author.name,
+                reference_user_display_name=reference.author.display_name,
+                reference_user_name=reference.author.name,
+                reference_time=self.get_msg_time(reference)
+            )
+
+        # 回退到原始模板
+        prompt = f"""
+        <context>
+        {context}
+        </context>
+        `<reference>` is the message you need to translate.
+        <reference>
+        {reference.content}
+        </reference>
+        This message is from `<author>`.
+        <author>
+        {reference.author.display_name} ({reference.author.name}) ({self.get_msg_time(reference)})
+        </author>
+        You are a skilled muti-lingual translator, currently doing a translation job in a discord server. You'll get a message which you need to translate into {target_language} with context. You only need to supply the translation according to the context without any additional information. Don't act like a machine, translate smoothly like a human without being too informal. 
+        Your translation should not include the author's name and the time.
+        Now is {now(tz=self.tz)}.
+        {ctx.author.display_name} ({ctx.author.name}) is asking you to translate the message in `<reference>` into {target_language} under the context (refer to <context>). The message is from `<author>`, so consider the context and try to understand the message before translating.
+        Your translation:
+        """
+        return prompt