ruff说他有好点子

src/common/database.py

@@ -1,6 +1,8 @@
-from pymongo import MongoClient
 from typing import Optional
 
+from pymongo import MongoClient
+
+
 class Database:
     _instance: Optional["Database"] = None
     

src/gui/reasoning_gui.py

@@ -1,12 +1,12 @@
-import customtkinter as ctk
-from typing import Dict, List
-import json
-from datetime import datetime
-import time
-import threading
+import os
 import queue
 import sys
-import os
+import threading
+import time
+from datetime import datetime
+from typing import Dict, List
+
+import customtkinter as ctk
 from dotenv import load_dotenv
 
 # 获取当前文件的目录
@@ -25,9 +25,11 @@ else:
     print("未找到环境配置文件")
     sys.exit(1)
 
-from pymongo import MongoClient
 from typing import Optional
 
+from pymongo import MongoClient
+
+
 class Database:
     _instance: Optional["Database"] = None
     

src/plugins/chat/Segment_builder.py

@@ -1,6 +1,5 @@
-from typing import Dict, List, Union, Optional, Any
 import base64
-import os
+from typing import Any, Dict, List, Union
 
 """
 OneBot v11 Message Segment Builder

src/plugins/chat/__init__.py

@@ -1,21 +1,23 @@
-from loguru import logger
-from nonebot import on_message, on_command, require, get_driver
-from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message, MessageSegment
-from nonebot.typing import T_State
-from ...common.database import Database
-from .config import global_config
-import os
 import asyncio
+import os
 import random
-from .relationship_manager import relationship_manager
-from ..schedule.schedule_generator import bot_schedule
-from .willing_manager import willing_manager
-from nonebot.rule import to_me
-from .bot import chat_bot
-from .emoji_manager import emoji_manager
-from ..moods.moods import MoodManager  # 导入情绪管理器
 import time
+
+from loguru import logger
+from nonebot import get_driver, on_command, on_message, require
+from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message, MessageSegment
+from nonebot.rule import to_me
+from nonebot.typing import T_State
+
+from ...common.database import Database
+from ..moods.moods import MoodManager  # 导入情绪管理器
+from ..schedule.schedule_generator import bot_schedule
 from ..utils.statistic import LLMStatistics
+from .bot import chat_bot
+from .config import global_config
+from .emoji_manager import emoji_manager
+from .relationship_manager import relationship_manager
+from .willing_manager import willing_manager
 
 # 创建LLM统计实例
 llm_stats = LLMStatistics("llm_statistics.txt")
@@ -39,12 +41,11 @@ print("\033[1;32m[初始化数据库完成]\033[0m")
 
 
 # 导入其他模块
+from ..memory_system.memory import hippocampus, memory_graph
 from .bot import ChatBot
-from .emoji_manager import emoji_manager
+
 # from .message_send_control import message_sender
-from .relationship_manager import relationship_manager
-from .message_sender import message_manager,message_sender
-from ..memory_system.memory import memory_graph,hippocampus
+from .message_sender import message_manager, message_sender
 
 # 初始化表情管理器
 emoji_manager.initialize()

src/plugins/chat/bot.py

@@ -1,23 +1,29 @@
-from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message as EventMessage, Bot
-from .message import Message, MessageSet, Message_Sending
-from .config import BotConfig, global_config
-from .storage import MessageStorage
-from .llm_generator import ResponseGenerator
-# from .message_stream import MessageStream, MessageStreamContainer
-from .topic_identifier import topic_identifier
-from random import random, choice
-from .emoji_manager import emoji_manager  # 导入表情包管理器
-from ..moods.moods import MoodManager  # 导入情绪管理器
 import time
-import os
-from .cq_code import CQCode  # 导入CQCode模块
-from .message_sender import message_manager  # 导入新的消息管理器
-from .message import Message_Thinking  # 导入 Message_Thinking 类
-from .relationship_manager import relationship_manager
-from .willing_manager import willing_manager  # 导入意愿管理器
-from .utils import is_mentioned_bot_in_txt, calculate_typing_time
-from ..memory_system.memory import memory_graph,hippocampus
+
+# from .message_stream import MessageStream, MessageStreamContainer
+from random import random
+
 from loguru import logger
+from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent
+
+from ..memory_system.memory import hippocampus
+from ..moods.moods import MoodManager  # 导入情绪管理器
+from .config import global_config
+from .cq_code import CQCode  # 导入CQCode模块
+from .emoji_manager import emoji_manager  # 导入表情包管理器
+from .llm_generator import ResponseGenerator
+from .message import (
+    Message,
+    Message_Sending,
+    Message_Thinking,  # 导入 Message_Thinking 类
+    MessageSet,
+)
+from .message_sender import message_manager  # 导入新的消息管理器
+from .relationship_manager import relationship_manager
+from .storage import MessageStorage
+from .utils import calculate_typing_time, is_mentioned_bot_in_txt
+from .willing_manager import willing_manager  # 导入意愿管理器
+
 
 class ChatBot:
     def __init__(self):

src/plugins/chat/config.py

@@ -1,12 +1,9 @@
-from dataclasses import dataclass, field
-from typing import Dict, Any, Optional, Set
 import os
-import configparser
-import tomli
-import sys
-from loguru import logger
-from nonebot import get_driver
+from dataclasses import dataclass, field
+from typing import Dict, Optional
 
+import tomli
+from loguru import logger
 
 
 @dataclass

src/plugins/chat/cq_code.py

@@ -1,24 +1,23 @@
-from dataclasses import dataclass
-from typing import Dict, Optional, List, Union
-import html
-import requests
 import base64
-from PIL import Image
+import html
 import os
-from random import random
-from nonebot.adapters.onebot.v11 import Bot
-from .config import global_config
 import time
-import asyncio
-from .utils_image import storage_image, storage_emoji
-from .utils_user import get_user_nickname
-from ..models.utils_model import LLM_request
-from .mapper import emojimapper
+from dataclasses import dataclass
+from typing import Dict, Optional
+
+import requests
+
 # 解析各种CQ码
 # 包含CQ码类
 import urllib3
-from urllib3.util import create_urllib3_context
 from nonebot import get_driver
+from urllib3.util import create_urllib3_context
+
+from ..models.utils_model import LLM_request
+from .config import global_config
+from .mapper import emojimapper
+from .utils_image import storage_emoji, storage_image
+from .utils_user import get_user_nickname
 
 driver = get_driver()
 config = driver.config
@@ -81,7 +80,7 @@ class CQCode:
             if user_nickname:
                 self.translated_plain_text = f"[@{user_nickname}]"
             else:
-                self.translated_plain_text = f"@某人"
+                self.translated_plain_text = "@某人"
         elif self.type == 'reply':
             self.translated_plain_text = await self.translate_reply()
         elif self.type == 'face':
@@ -308,7 +307,7 @@ class CQCode:
                 return f"[回复 {self.reply_message.sender.nickname} 的消息: {message_obj.processed_plain_text}]"
 
         else:
-            print(f"\033[1;31m[错误]\033[0m 回复消息的sender.user_id为空")
+            print("\033[1;31m[错误]\033[0m 回复消息的sender.user_id为空")
             return '[回复某人消息]'
 
     @staticmethod

src/plugins/chat/emoji_manager.py

@@ -1,27 +1,18 @@
-from typing import List, Dict, Optional
-import random
-from ...common.database import Database
-import os
-import json
-from dataclasses import dataclass
-import jieba.analyse as jieba_analyse
-import aiohttp
-import hashlib
-from datetime import datetime
-import base64
-import shutil
 import asyncio
+import os
+import random
 import time
-from PIL import Image
-import io
-from loguru import logger
 import traceback
+from typing import Optional
 
+from loguru import logger
 from nonebot import get_driver
+
+from ...common.database import Database
 from ..chat.config import global_config
-from ..models.utils_model import LLM_request
-from ..chat.utils_image import image_path_to_base64
 from ..chat.utils import get_embedding
+from ..chat.utils_image import image_path_to_base64
+from ..models.utils_model import LLM_request
 
 driver = get_driver()
 config = driver.config
@@ -275,7 +266,7 @@ class EmojiManager:
     async def _periodic_scan(self, interval_MINS: int = 10):
         """定期扫描新表情包"""
         while True:
-            print(f"\033[1;36m[表情包]\033[0m 开始扫描新表情包...")
+            print("\033[1;36m[表情包]\033[0m 开始扫描新表情包...")
             await self.scan_new_emojis()
             await asyncio.sleep(interval_MINS * 60)  # 每600秒扫描一次
 

src/plugins/chat/llm_generator.py

@@ -1,19 +1,16 @@
-from typing import Dict, Any, List, Optional, Union, Tuple
-from openai import OpenAI
-import asyncio
-from functools import partial
-from .message import Message
-from .config import global_config
-from ...common.database import Database
 import random
 import time
-import numpy as np
-from .relationship_manager import relationship_manager
-from .prompt_builder import prompt_builder
-from .config import global_config
-from .utils import process_llm_response
+from typing import List, Optional, Tuple, Union
+
 from nonebot import get_driver
+
+from ...common.database import Database
 from ..models.utils_model import LLM_request
+from .config import global_config
+from .message import Message
+from .prompt_builder import prompt_builder
+from .relationship_manager import relationship_manager
+from .utils import process_llm_response
 
 driver = get_driver()
 config = driver.config

src/plugins/chat/message.py

@@ -1,16 +1,12 @@
-from dataclasses import dataclass
-from typing import List, Optional, Dict, Tuple, ForwardRef
 import time
-import jieba.analyse as jieba_analyse
-import os
-from datetime import datetime
-from ...common.database import Database
-from PIL import Image
-from .config import global_config
+from dataclasses import dataclass
+from typing import Dict, ForwardRef, List, Optional
+
 import urllib3
-from .utils_user import get_user_nickname,get_user_cardname,get_groupname
+
+from .cq_code import CQCode, cq_code_tool
 from .utils_cq import parse_cq_code
-from .cq_code import cq_code_tool,CQCode
+from .utils_user import get_groupname, get_user_cardname, get_user_nickname
 
 Message = ForwardRef('Message')  # 添加这行
 # 禁用SSL警告

src/plugins/chat/message_sender.py

@@ -1,15 +1,15 @@
-from typing import Union, List, Optional, Dict
-from collections import deque
-from .message import Message, Message_Thinking, MessageSet, Message_Sending
-import time
 import asyncio
+import time
+from typing import Dict, List, Optional, Union
+
 from nonebot.adapters.onebot.v11 import Bot
-from .config import global_config
-from .storage import MessageStorage
+
 from .cq_code import cq_code_tool
-import random
+from .message import Message, Message_Sending, Message_Thinking, MessageSet
+from .storage import MessageStorage
 from .utils import calculate_typing_time
 
+
 class Message_Sender:
     """发送器"""
     def __init__(self):
@@ -199,7 +199,7 @@ class MessageManager:
                         
                         # 安全地移除消息
                         if not container.remove_message(msg):
-                            print(f"\033[1;33m[警告]\033[0m 尝试删除不存在的消息")
+                            print("\033[1;33m[警告]\033[0m 尝试删除不存在的消息")
                     except Exception as e:
                         print(f"\033[1;31m[错误]\033[0m 处理超时消息时发生错误: {e}")
                         continue

src/plugins/chat/prompt_builder.py

@@ -1,17 +1,14 @@
-import time
 import random
-from ..schedule.schedule_generator import bot_schedule
-import os
-from .utils import get_embedding, combine_messages, get_recent_group_detailed_plain_text
+import time
+from typing import Optional
+
 from ...common.database import Database
-from .config import global_config
-from .topic_identifier import topic_identifier
-from ..memory_system.memory import memory_graph,hippocampus
-from random import choice
-import numpy as np
-import jieba
-from collections import Counter
+from ..memory_system.memory import hippocampus, memory_graph
 from ..moods.moods import MoodManager
+from ..schedule.schedule_generator import bot_schedule
+from .config import global_config
+from .utils import get_embedding, get_recent_group_detailed_plain_text
+
 
 class PromptBuilder:
     def __init__(self):
@@ -25,7 +22,7 @@ class PromptBuilder:
                     message_txt: str, 
                     sender_name: str = "某人",
                     relationship_value: float = 0.0,
-                    group_id: int = None) -> str:
+                    group_id: Optional[int] = None) -> tuple[str, str]:
         """构建prompt
         
         Args:
@@ -101,7 +98,7 @@ class PromptBuilder:
             for memory in relevant_memories:
                 memory_items.append(f"关于「{memory['topic']}」的记忆：{memory['content']}")
             
-            memory_prompt = f"看到这些聊天，你想起来：\n" + "\n".join(memory_items) + "\n"
+            memory_prompt = "看到这些聊天，你想起来：\n" + "\n".join(memory_items) + "\n"
             
             # 打印调试信息
             print("\n\033[1;32m[记忆检索]\033[0m 找到以下相关记忆：")
@@ -203,7 +200,7 @@ class PromptBuilder:
 
         #激活prompt构建
         activate_prompt = ''
-        activate_prompt = f"以上是群里正在进行的聊天。"
+        activate_prompt = "以上是群里正在进行的聊天。"
         personality=global_config.PROMPT_PERSONALITY
         prompt_personality = ''
         personality_choice = random.random()

src/plugins/chat/relationship_manager.py

@@ -1,8 +1,8 @@
-import time
-from ...common.database import Database
-from nonebot.adapters.onebot.v11 import Bot
-from typing import Optional, Tuple
 import asyncio
+from typing import Optional
+
+from ...common.database import Database
+
 
 class Impression:
     traits: str = None
@@ -123,7 +123,7 @@ class RelationshipManager:
         print(f"\033[1;32m[关系管理]\033[0m 已加载 {len(self.relationships)} 条关系记录")
         
         while True:
-            print(f"\033[1;32m[关系管理]\033[0m 正在自动保存关系")
+            print("\033[1;32m[关系管理]\033[0m 正在自动保存关系")
             await asyncio.sleep(300)  # 等待300秒(5分钟)
             await self._save_all_relationships()
     

src/plugins/chat/storage.py

@@ -1,10 +1,8 @@
-from typing import Dict, List, Any, Optional
-import time
-import threading
-from collections import defaultdict
-import asyncio
-from .message import Message
+from typing import Optional
+
 from ...common.database import Database
+from .message import Message
+
 
 class MessageStorage:
     def __init__(self):

src/plugins/chat/topic_identifier.py

@@ -1,10 +1,9 @@
-from typing import Optional, Dict, List
-from openai import OpenAI
-from .message import Message
-import jieba
+from typing import List, Optional
+
 from nonebot import get_driver
-from .config import global_config
+
 from ..models.utils_model import LLM_request
+from .config import global_config
 
 driver = get_driver()
 config = driver.config  
@@ -26,7 +25,7 @@ class TopicIdentifier:
         topic, _ = await self.llm_topic_judge.generate_response(prompt)
         
         if not topic:
-            print(f"\033[1;31m[错误]\033[0m LLM API 返回为空")
+            print("\033[1;31m[错误]\033[0m LLM API 返回为空")
             return None
             
         # 直接在这里处理主题解析

src/plugins/chat/utils.py

@@ -1,19 +1,17 @@
-import time
-import random
-from typing import List
-from .message import Message
-import requests
-import numpy as np
-from .config import global_config
-import re
-from typing import Dict
-from collections import Counter
 import math
-from nonebot import get_driver
-from ..models.utils_model import LLM_request
-import aiohttp
+import random
+import time
+from collections import Counter
+from typing import Dict, List
+
 import jieba
+import numpy as np
+from nonebot import get_driver
+
+from ..models.utils_model import LLM_request
 from ..utils.typo_generator import ChineseTypoGenerator
+from .config import global_config
+from .message import Message
 
 driver = get_driver()
 config = driver.config

src/plugins/chat/utils_image.py

@@ -1,14 +1,15 @@
+import base64
 import io
-from PIL import Image
-import hashlib
-import time
 import os
+import time
+import zlib  # 用于 CRC32
+
+from loguru import logger
+from nonebot import get_driver
+from PIL import Image
+
 from ...common.database import Database
 from ..chat.config import global_config
-import zlib  # 用于 CRC32
-import base64
-from nonebot import get_driver
-from loguru import logger
 
 driver = get_driver()
 config = driver.config
@@ -119,7 +120,7 @@ def storage_compress_image(base64_data: str, max_size: int = 200) -> str:
             
             # 保存记录
             collection.insert_one(image_record)
-            print(f"\033[1;32m[成功]\033[0m 保存图片记录到数据库")
+            print("\033[1;32m[成功]\033[0m 保存图片记录到数据库")
             
         except Exception as db_error:
             print(f"\033[1;31m[错误]\033[0m 数据库操作失败: {str(db_error)}")

src/plugins/chat/utils_user.py

@@ -1,5 +1,6 @@
-from .relationship_manager import relationship_manager
 from .config import global_config
+from .relationship_manager import relationship_manager
+
 
 def get_user_nickname(user_id: int) -> str:
     if int(user_id) == int(global_config.BOT_QQ):

src/plugins/chat/willing_manager.py

@@ -1,5 +1,6 @@
 import asyncio
 
+
 class WillingManager:
     def __init__(self):
         self.group_reply_willing = {}  # 存储每个群的回复意愿

src/plugins/knowledege/knowledge_library.py

@@ -1,8 +1,8 @@
 import os
 import sys
-import numpy as np
-import requests
 import time
+
+import requests
 from dotenv import load_dotenv
 
 # 添加项目根目录到 Python 路径

src/plugins/memory_system/draw_memory.py

@@ -1,19 +1,12 @@
 # -*- coding: utf-8 -*-
 import os
 import sys
-import jieba
-import networkx as nx
-import matplotlib.pyplot as plt
-import math
-from collections import Counter
-import datetime
-import random
 import time
+
+import jieba
+import matplotlib.pyplot as plt
+import networkx as nx
 from dotenv import load_dotenv
-import sys
-import asyncio
-import aiohttp
-from typing import Tuple
 
 sys.path.append("C:/GitHub/MaiMBot")  # 添加项目根目录到 Python 路径
 from src.common.database import Database  # 使用正确的导入语法

src/plugins/memory_system/memory.py

@@ -1,20 +1,21 @@
 # -*- coding: utf-8 -*-
-import os
-import jieba
-import networkx as nx
-import matplotlib.pyplot as plt
-from collections import Counter
 import datetime
+import math
 import random
 import time
+
+import jieba
+import networkx as nx
+
+from ...common.database import Database  # 使用正确的导入语法
 from ..chat.config import global_config
-from ...common.database import Database # 使用正确的导入语法
+from ..chat.utils import (
+    calculate_information_content,
+    cosine_similarity,
+    get_cloest_chat_from_db,
+    text_to_vector,
+)
 from ..models.utils_model import LLM_request
-import math
-from ..chat.utils import calculate_information_content, get_cloest_chat_from_db ,text_to_vector,cosine_similarity
-
-
-
 
 
 class Memory_graph:
@@ -696,6 +697,7 @@ def segment_text(text):
 
 
 from nonebot import get_driver
+
 driver = get_driver()
 config = driver.config
 

src/plugins/memory_system/memory_manual_build.py

@@ -1,21 +1,22 @@
 # -*- coding: utf-8 -*-
-import sys
-import jieba
-import networkx as nx
-import matplotlib.pyplot as plt
-import math
-from collections import Counter
 import datetime
-import random
-import time
+import math
 import os
-from dotenv import load_dotenv
-import pymongo
-from loguru import logger
+import random
+import sys
+import time
+from collections import Counter
 from pathlib import Path
+
+import matplotlib.pyplot as plt
+import networkx as nx
+import pymongo
+from dotenv import load_dotenv
+from loguru import logger
+
 # from chat.config import global_config
 sys.path.append("C:/GitHub/MaiMBot")  # 添加项目根目录到 Python 路径
-from src.common.database import Database  
+from src.common.database import Database
 from src.plugins.memory_system.offline_llm import LLMModel
 
 # 获取当前文件的目录
@@ -102,7 +103,7 @@ def get_cloest_chat_from_db(db, length: int, timestamp: str):
             # 检查当前记录的memorized值
             current_memorized = record.get('memorized', 0)
             if current_memorized  > 3:
-                print(f"消息已读取3次，跳过")
+                print("消息已读取3次，跳过")
                 return ''
                 
             # 更新memorized值
@@ -114,7 +115,7 @@ def get_cloest_chat_from_db(db, length: int, timestamp: str):
             chat_text += record["detailed_plain_text"]
             
         return chat_text
-    print(f"消息已读取3次，跳过")
+    print("消息已读取3次，跳过")
     return ''
 
 class Memory_graph:

src/plugins/memory_system/offline_llm.py

@@ -1,11 +1,13 @@
-import os
-import requests
-from typing import Tuple, Union
-import time
-import aiohttp
 import asyncio
+import os
+import time
+from typing import Tuple, Union
+
+import aiohttp
+import requests
 from loguru import logger
 
+
 class LLMModel:
     def __init__(self, model_name="deepseek-ai/DeepSeek-V3", **kwargs):
         self.model_name = model_name

src/plugins/models/utils_model.py

@@ -1,16 +1,16 @@
-import aiohttp
 import asyncio
 import json
-import requests
-import time
 import re
+from datetime import datetime
 from typing import Tuple, Union
-from nonebot import get_driver
+
+import aiohttp
 from loguru import logger
+from nonebot import get_driver
+
+from ...common.database import Database
 from ..chat.config import global_config
 from ..chat.utils_image import compress_base64_image_by_scale
-from datetime import datetime
-from ...common.database import Database
 
 driver = get_driver()
 config = driver.config

src/plugins/moods/moods.py

@@ -1,10 +1,11 @@
 import math
-import time
 import threading
-from typing import Dict, Tuple, Optional
+import time
 from dataclasses import dataclass
+
 from ..chat.config import global_config
 
+
 @dataclass
 class MoodState:
     valence: float  # 愉悦度 (-1 到 1)

src/plugins/schedule/schedule_generator.py

@@ -1,12 +1,15 @@
 import datetime
-import os
-from typing import List, Dict, Union
-from ...common.database import Database  # 使用正确的导入语法
-from src.plugins.chat.config import global_config
-from nonebot import get_driver
-from ..models.utils_model import LLM_request
-from loguru import logger
 import json
+from typing import Dict, Union
+
+from loguru import logger
+from nonebot import get_driver
+
+from src.plugins.chat.config import global_config
+
+from ...common.database import Database  # 使用正确的导入语法
+from ..models.utils_model import LLM_request
+
 driver = get_driver()
 config = driver.config
 

src/plugins/utils/statistic.py

@@ -1,11 +1,12 @@
-from typing import Dict, List, Any
-import time
 import threading
-import json
-from datetime import datetime, timedelta
+import time
 from collections import defaultdict
+from datetime import datetime, timedelta
+from typing import Any, Dict
+
 from ...common.database import Database
 
+
 class LLMStatistics:
     def __init__(self, output_file: str = "llm_statistics.txt"):
         """初始化LLM统计类

src/plugins/utils/typo_generator.py

@@ -2,15 +2,17 @@
 错别字生成器 - 基于拼音和字频的中文错别字生成工具
 """
 
-from pypinyin import pinyin, Style
-from collections import defaultdict
 import json
-import os
-import jieba
-from pathlib import Path
-import random
 import math
+import os
+import random
 import time
+from collections import defaultdict
+from pathlib import Path
+
+import jieba
+from pypinyin import Style, pinyin
+
 
 class ChineseTypoGenerator:
     def __init__(self,