This commit is contained in:
SengokuCola
2025-04-25 18:12:14 +08:00
5 changed files with 97 additions and 62 deletions

View File

@@ -24,6 +24,9 @@ jobs:
- name: Clone maim_message - name: Clone maim_message
run: git clone https://github.com/MaiM-with-u/maim_message maim_message run: git clone https://github.com/MaiM-with-u/maim_message maim_message
- name: Clone lpmm
run: git clone https://github.com/MaiM-with-u/MaiMBot-LPMM.git MaiMBot-LPMM
- name: Set up Docker Buildx - name: Set up Docker Buildx
uses: docker/setup-buildx-action@v3 uses: docker/setup-buildx-action@v3

View File

@@ -6,15 +6,22 @@ WORKDIR /MaiMBot
# 复制依赖列表 # 复制依赖列表
COPY requirements.txt . COPY requirements.txt .
# 同级目录下需要有 maim_message # 同级目录下需要有 maim_message MaiMBot-LPMM
COPY maim_message /maim_message #COPY maim_message /maim_message
COPY MaiMBot-LPMM /MaiMBot-LPMM
# 编译器 # 编译器
RUN apt-get update && apt-get install -y g++ RUN apt-get update && apt-get install -y build-essential
# lpmm编译安装
RUN cd /MaiMBot-LPMM && uv pip install --system -r requirements.txt
RUN uv pip install --system Cython py-cpuinfo setuptools
RUN cd /MaiMBot-LPMM/lib/quick_algo && python build_lib.py --cleanup --cythonize --install
# 安装依赖 # 安装依赖
RUN uv pip install --system --upgrade pip RUN uv pip install --system --upgrade pip
RUN uv pip install --system -e /maim_message #RUN uv pip install --system -e /maim_message
RUN uv pip install --system -r requirements.txt RUN uv pip install --system -r requirements.txt
# 复制项目代码 # 复制项目代码

View File

@@ -4,7 +4,7 @@
# 适用于Arch/Ubuntu 24.10/Debian 12/CentOS 9 # 适用于Arch/Ubuntu 24.10/Debian 12/CentOS 9
# 请小心使用任何一键脚本! # 请小心使用任何一键脚本!
INSTALLER_VERSION="0.0.3-refactor" INSTALLER_VERSION="0.0.4-refactor"
LANG=C.UTF-8 LANG=C.UTF-8
# 如无法访问GitHub请修改此处镜像地址 # 如无法访问GitHub请修改此处镜像地址
@@ -19,10 +19,10 @@ RESET="\e[0m"
declare -A REQUIRED_PACKAGES=( declare -A REQUIRED_PACKAGES=(
["common"]="git sudo python3 curl gnupg" ["common"]="git sudo python3 curl gnupg"
["debian"]="python3-venv python3-pip" ["debian"]="python3-venv python3-pip build-essential"
["ubuntu"]="python3-venv python3-pip" ["ubuntu"]="python3-venv python3-pip build-essential"
["centos"]="python3-pip" ["centos"]="epel-release python3-pip python3-devel gcc gcc-c++ make"
["arch"]="python-virtualenv python-pip" ["arch"]="python-virtualenv python-pip base-devel"
) )
# 默认项目目录 # 默认项目目录

View File

@@ -248,54 +248,72 @@ class Conversation:
# --- 根据不同的 action 执行 --- # --- 根据不同的 action 执行 ---
if action == "direct_reply": if action == "direct_reply":
# --- 这个 if 块内部的所有代码都需要正确缩进 --- max_reply_attempts = 3 # 设置最大尝试次数(与 reply_checker.py 中的 max_retries 保持一致或稍大)
self.waiter.wait_accumulated_time = 0 # 重置等待时间 reply_attempt_count = 0
is_suitable = False
self.state = ConversationState.GENERATING
# 生成回复
self.generated_reply = await self.reply_generator.generate(observation_info, conversation_info)
logger.info(f"生成回复: {self.generated_reply}") # 使用 logger
# --- 调用 ReplyChecker 检查回复 ---
is_suitable = False # 先假定不合适,检查通过再改为 True
check_reason = "检查未执行" # 用不同的变量名存储检查原因
need_replan = False need_replan = False
try: check_reason = "未进行尝试"
# 尝试获取当前主要目标 final_reply_to_send = ""
current_goal_str = conversation_info.goal_list[0][0] if conversation_info.goal_list else ""
# 调用检查器 while reply_attempt_count < max_reply_attempts and not is_suitable:
reply_attempt_count += 1
logger.info(f"尝试生成回复 (第 {reply_attempt_count}/{max_reply_attempts} 次)...")
self.state = ConversationState.GENERATING
# 1. 生成回复
self.generated_reply = await self.reply_generator.generate(observation_info, conversation_info)
logger.info(f"{reply_attempt_count} 次生成的回复: {self.generated_reply}")
# 2. 检查回复
self.state = ConversationState.CHECKING
try:
current_goal_str = conversation_info.goal_list[0][0] if conversation_info.goal_list else ""
# 注意:这里传递的是 reply_attempt_count - 1 作为 retry_count 给 checker
is_suitable, check_reason, need_replan = await self.reply_generator.check_reply( is_suitable, check_reason, need_replan = await self.reply_generator.check_reply(
reply=self.generated_reply, reply=self.generated_reply,
goal=current_goal_str, goal=current_goal_str,
chat_history=observation_info.chat_history, # 传入最新的历史记录! chat_history=observation_info.chat_history,
retry_count=0, retry_count=reply_attempt_count - 1, # 传递当前尝试次数从0开始计数
)
logger.info(
f"{reply_attempt_count} 次检查结果: 合适={is_suitable}, 原因='{check_reason}', 需重新规划={need_replan}"
) )
logger.info(f"回复检查结果: 合适={is_suitable}, 原因='{check_reason}', 需重新规划={need_replan}")
except Exception as check_err:
logger.error(f"调用 ReplyChecker 时出错: {check_err}")
check_reason = f"检查过程出错: {check_err}" # 记录错误原因
# is_suitable 保持 False
# --- 处理检查结果 ---
if is_suitable: if is_suitable:
# 回复合适,继续执行 final_reply_to_send = self.generated_reply # 保存合适的回复
# 检查是否有新消息进来 break # 回复合适,跳出循环
elif need_replan:
logger.warning(f"{reply_attempt_count} 次检查建议重新规划,停止尝试。原因: {check_reason}")
break # 如果检查器建议重新规划,也停止尝试
# 如果不合适但不需要重新规划,循环会继续进行下一次尝试
except Exception as check_err:
logger.error(f"{reply_attempt_count} 次调用 ReplyChecker 时出错: {check_err}")
check_reason = f"{reply_attempt_count} 次检查过程出错: {check_err}"
# 如果检查本身出错,可以选择跳出循环或继续尝试
# 这里选择跳出循环,避免无限循环在检查错误上
break
# 循环结束,处理最终结果
if is_suitable:
# 回复合适且已保存在 final_reply_to_send 中
# 检查是否有新消息进来 (在所有尝试结束后再检查一次)
if self._check_new_messages_after_planning(): if self._check_new_messages_after_planning():
logger.info("检查到新消息,取消发送已生成的回复,重新规划行动") logger.info("生成回复期间收到新消息,取消发送,重新规划行动")
# 更新 action 状态为 recall
conversation_info.done_action[action_index].update( conversation_info.done_action[action_index].update(
{ {
"status": "recall", "status": "recall",
"reason": f"有新消息,取消发送: {self.generated_reply}", # 更新原因 "final_reason": f"有新消息,取消发送: {final_reply_to_send}",
"time": datetime.datetime.now().strftime("%H:%M:%S"), "time": datetime.datetime.now().strftime("%H:%M:%S"),
} }
) )
return None # 退出 _handle_action # 这里直接返回不执行后续发送和wait
return
# 发送回复 # 发送合适的回复
await self._send_reply() # 这个函数内部会处理自己的错误 self.generated_reply = final_reply_to_send # 确保 self.generated_reply 是最终要发送的内容
await self._send_reply()
# 更新 action 历史状态为 done # 更新 action 历史状态为 done
conversation_info.done_action[action_index].update( conversation_info.done_action[action_index].update(
@@ -306,26 +324,30 @@ class Conversation:
) )
else: else:
# 回复不合适 # 循环结束但没有找到合适的回复(达到最大次数或检查出错/建议重规划)
logger.warning(f"生成的回复被 ReplyChecker 拒绝: '{self.generated_reply}'. 原因: {check_reason}") logger.warning(f"经过 {reply_attempt_count} 次尝试,未能生成合适的回复。最终原因: {check_reason}")
# 更新 action 状态为 recall (因为没执行发送)
conversation_info.done_action[action_index].update( conversation_info.done_action[action_index].update(
{ {
"status": "recall", "status": "recall", # 标记为 recall 因为没有成功发送
"final_reason": check_reason, "final_reason": f"尝试{reply_attempt_count}次后失败: {check_reason}",
"time": datetime.datetime.now().strftime("%H:%M:%S"), "time": datetime.datetime.now().strftime("%H:%M:%S"),
} }
) )
# 如果检查器建议重新规划 # 执行 Wait 操作
if need_replan: logger.info("由于无法生成合适回复,执行 'wait' 操作...")
logger.info("ReplyChecker 建议重新规划目标。") self.state = ConversationState.WAITING
# 可选:在此处清空目标列表以强制重新规划 # 直接调用 wait 方法
# conversation_info.goal_list = [] await self.waiter.wait(self.conversation_info)
# 可以选择添加一条新的 action 记录来表示这个 wait
# 注意:不发送消息,也不执行后面的代码 wait_action_record = {
"action": "wait",
# --- 之前重复的代码块已被删除 --- "plan_reason": "因 direct_reply 多次尝试失败而执行的后备等待",
"status": "done", # wait 完成后可以认为是 done
"time": datetime.datetime.now().strftime("%H:%M:%S"),
"final_reason": None,
}
conversation_info.done_action.append(wait_action_record)
elif action == "fetch_knowledge": elif action == "fetch_knowledge":
self.waiter.wait_accumulated_time = 0 self.waiter.wait_accumulated_time = 0

View File

@@ -15,11 +15,11 @@ class ReplyChecker:
def __init__(self, stream_id: str): def __init__(self, stream_id: str):
self.llm = LLMRequest( self.llm = LLMRequest(
model=global_config.llm_PFC_reply_checker, temperature=0.55, max_tokens=1000, request_type="reply_check" model=global_config.llm_PFC_reply_checker, temperature=0.50, max_tokens=1000, request_type="reply_check"
) )
self.name = global_config.BOT_NICKNAME self.name = global_config.BOT_NICKNAME
self.chat_observer = ChatObserver.get_instance(stream_id) self.chat_observer = ChatObserver.get_instance(stream_id)
self.max_retries = 2 # 最大重试次数 self.max_retries = 3 # 最大重试次数
async def check( async def check(
self, reply: str, goal: str, chat_history: List[Dict[str, Any]], retry_count: int = 0 self, reply: str, goal: str, chat_history: List[Dict[str, Any]], retry_count: int = 0
@@ -76,8 +76,11 @@ class ReplyChecker:
False, False,
) )
except Exception as self_check_err: except Exception as e:
logger.error(f"检查自身重复发言时出错: {self_check_err}") import traceback
logger.error(f"检查回复时出错: 类型={type(e)}, 值={e}")
logger.error(traceback.format_exc()) # 打印详细的回溯信息
for msg in chat_history[-20:]: for msg in chat_history[-20:]:
time_str = datetime.datetime.fromtimestamp(msg["time"]).strftime("%H:%M:%S") time_str = datetime.datetime.fromtimestamp(msg["time"]).strftime("%H:%M:%S")