diff --git a/.dockerignore b/.dockerignore
index 6c2d07736..fac1bf99a 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -1,6 +1,5 @@
.git
__pycache__
-*.pyc
*.pyo
*.pyd
.DS_Store
diff --git a/CONTRIBUTE.md b/CONTRIBUTE.md
index 440a8202d..c372aebe8 100644
--- a/CONTRIBUTE.md
+++ b/CONTRIBUTE.md
@@ -19,7 +19,6 @@
● [我有问题](#我有问题)
● [我想做贡献](#我想做贡献)
-● [我想报告BUG](#报告BUG)
● [我想提出建议](#提出建议)
## 我有问题
diff --git a/README.md b/README.md
index 325e3ad22..e5a3d1306 100644
--- a/README.md
+++ b/README.md
@@ -113,17 +113,17 @@
## 🎯 功能介绍
-| 模块 | 主要功能 | 特点 |
-|------|---------|------|
-| 💬 聊天系统 | • 心流/推理聊天
• 关键词主动发言
• 多模型支持
• 动态prompt构建
• 私聊功能(PFC) | 拟人化交互 |
-| 🧠 心流系统 | • 实时思考生成
• 自动启停机制
• 日程系统联动
• 工具调用能力 | 智能化决策 |
-| 🧠 记忆系统 | • 优化记忆抽取
• 海马体记忆机制
• 聊天记录概括 | 持久化记忆 |
-| 😊 表情系统 | • 情绪匹配发送
• GIF支持
• 自动收集与审查 | 丰富表达 |
-| 📅 日程系统 | • 动态日程生成
• 自定义想象力
• 思维流联动 | 智能规划 |
-| 👥 关系系统 | • 关系管理优化
• 丰富接口支持
• 个性化交互 | 深度社交 |
-| 📊 统计系统 | • 使用数据统计
• LLM调用记录
• 实时控制台显示 | 数据可视 |
-| 🔧 系统功能 | • 优雅关闭机制
• 自动数据保存
• 异常处理完善 | 稳定可靠 |
-| 🛠️ 工具系统 | • 知识获取工具
• 自动注册机制
• 多工具支持 | 扩展功能 |
+| 模块 | 主要功能 | 特点 |
+|----------|------------------------------------------------------------------|-------|
+| 💬 聊天系统 | • 心流/推理聊天
• 关键词主动发言
• 多模型支持
• 动态prompt构建
• 私聊功能(PFC) | 拟人化交互 |
+| 🧠 心流系统 | • 实时思考生成
• 自动启停机制
• 日程系统联动
• 工具调用能力 | 智能化决策 |
+| 🧠 记忆系统 | • 优化记忆抽取
• 海马体记忆机制
• 聊天记录概括 | 持久化记忆 |
+| 😊 表情系统 | • 情绪匹配发送
• GIF支持
• 自动收集与审查 | 丰富表达 |
+| 📅 日程系统 | • 动态日程生成
• 自定义想象力
• 思维流联动 | 智能规划 |
+| 👥 关系系统 | • 关系管理优化
• 丰富接口支持
• 个性化交互 | 深度社交 |
+| 📊 统计系统 | • 使用数据统计
• LLM调用记录
• 实时控制台显示 | 数据可视 |
+| 🔧 系统功能 | • 优雅关闭机制
• 自动数据保存
• 异常处理完善 | 稳定可靠 |
+| 🛠️ 工具系统 | • 知识获取工具
• 自动注册机制
• 多工具支持 | 扩展功能 |
## 📐 项目架构
diff --git a/src/plugins/message/test.py b/src/plugins/message/test.py
deleted file mode 100644
index abb4c03b5..000000000
--- a/src/plugins/message/test.py
+++ /dev/null
@@ -1,95 +0,0 @@
-import unittest
-import asyncio
-import aiohttp
-from api import BaseMessageAPI
-from message_base import (
- BaseMessageInfo,
- UserInfo,
- GroupInfo,
- FormatInfo,
- MessageBase,
- Seg,
-)
-
-
-send_url = "http://localhost"
-receive_port = 18002 # 接收消息的端口
-send_port = 18000 # 发送消息的端口
-test_endpoint = "/api/message"
-
-# 创建并启动API实例
-api = BaseMessageAPI(host="0.0.0.0", port=receive_port)
-
-
-class TestLiveAPI(unittest.IsolatedAsyncioTestCase):
- async def asyncSetUp(self):
- """测试前的设置"""
- self.received_messages = []
-
- async def message_handler(message):
- self.received_messages.append(message)
-
- self.api = api
- self.api.register_message_handler(message_handler)
- self.server_task = asyncio.create_task(self.api.run())
- try:
- await asyncio.wait_for(asyncio.sleep(1), timeout=5)
- except asyncio.TimeoutError:
- self.skipTest("服务器启动超时")
-
- async def asyncTearDown(self):
- """测试后的清理"""
- if hasattr(self, "server_task"):
- await self.api.stop() # 先调用正常的停止流程
- if not self.server_task.done():
- self.server_task.cancel()
- try:
- await asyncio.wait_for(self.server_task, timeout=100)
- except (asyncio.CancelledError, asyncio.TimeoutError):
- pass
-
- async def test_send_and_receive_message(self):
- """测试向运行中的API发送消息并接收响应"""
- # 准备测试消息
- user_info = UserInfo(user_id=12345678, user_nickname="测试用户", platform="qq")
- group_info = GroupInfo(group_id=12345678, group_name="测试群", platform="qq")
- format_info = FormatInfo(content_format=["text"], accept_format=["text", "emoji", "reply"])
- template_info = None
- message_info = BaseMessageInfo(
- platform="qq",
- message_id=12345678,
- time=12345678,
- group_info=group_info,
- user_info=user_info,
- format_info=format_info,
- template_info=template_info,
- )
- message = MessageBase(
- message_info=message_info,
- raw_message="测试消息",
- message_segment=Seg(type="text", data="测试消息"),
- )
- test_message = message.to_dict()
-
- # 发送测试消息到发送端口
- async with aiohttp.ClientSession() as session:
- async with session.post(
- f"{send_url}:{send_port}{test_endpoint}",
- json=test_message,
- ) as response:
- response_data = await response.json()
- self.assertEqual(response.status, 200)
- self.assertEqual(response_data["status"], "success")
- try:
- async with asyncio.timeout(5): # 设置5秒超时
- while len(self.received_messages) == 0:
- await asyncio.sleep(0.1)
- received_message = self.received_messages[0]
- print(received_message)
- self.received_messages.clear()
- except asyncio.TimeoutError:
- self.fail("等待接收消息超时")
-
-
-if __name__ == "__main__":
- unittest.main()
diff --git a/src/plugins/topic_identify/topic_identifier.py b/src/plugins/topic_identify/topic_identifier.py
index bc81874c1..25c290a3a 100644
--- a/src/plugins/topic_identify/topic_identifier.py
+++ b/src/plugins/topic_identify/topic_identifier.py
@@ -28,7 +28,7 @@ class TopicIdentifier:
消息内容：{text}"""
- # 使用 LLM_request 类进行请求
+ # 使用 LLMRequest 类进行请求
try:
topic, _, _ = await self.llm_topic_judge.generate_response(prompt)
except Exception as e:
diff --git a/src/plugins/willing/mode_llmcheck.py b/src/plugins/willing/mode_llmcheck.py
index d6ac3dc6b..ec1cde295 100644
--- a/src/plugins/willing/mode_llmcheck.py
+++ b/src/plugins/willing/mode_llmcheck.py
@@ -13,7 +13,7 @@ llmcheck 模式：
import time
from loguru import logger
-from ..models.utils_model import LLM_request
+from ..models.utils_model import LLMRequest
from ...config.config import global_config
# from ..chat.chat_stream import ChatStream
@@ -61,7 +61,7 @@ def llmcheck_decorator(trigger_condition_func):
class LlmcheckWillingManager(MxpWillingManager):
def __init__(self):
super().__init__()
- self.model_v3 = LLM_request(model=global_config.llm_normal, temperature=0.3)
+ self.model_v3 = LLMRequest(model=global_config.llm_normal, temperature=0.3)
async def get_llmreply_probability(self, message_id: str):
message_info = self.ongoing_messages[message_id]