better:优化工作记忆,给reply action加入额外信息

This commit is contained in:
SengokuCola
2025-06-02 21:47:21 +08:00
parent 476743427a
commit dd4a327b18
3 changed files with 67 additions and 47 deletions

View File

@@ -137,7 +137,8 @@ class ActionPlanner(BasePlanner):
self_info = info.get_processed_info()
elif isinstance(info, StructuredInfo):
structured_info = info.get_processed_info()
# print(f"structured_info: {structured_info}")
else:
extra_info.append(info.get_processed_info())
# elif not isinstance(info, ActionInfo): # 跳过已处理的ActionInfo
# extra_info.append(info.get_processed_info())
@@ -218,6 +219,15 @@ class ActionPlanner(BasePlanner):
action_data[key] = value
action_data["identity"] = self_info
extra_info_block = "\n".join(extra_info)
extra_info_block += f"\n{structured_info}"
if extra_info or structured_info:
extra_info_block = f"以下是一些额外的信息,现在请你阅读以下内容,进行决策\n{extra_info_block}\n以上是一些额外的信息,现在请你阅读以下内容,进行决策"
else:
extra_info_block = ""
action_data["extra_info_block"] = extra_info_block
# 对于reply动作不需要额外处理因为相关字段已经在上面的循环中添加到action_data
@@ -257,9 +267,13 @@ class ActionPlanner(BasePlanner):
)
action_result = {"action_type": action, "action_data": action_data, "reasoning": reasoning}
plan_result = {
"action_result": action_result,
# "extra_info_block": extra_info_block,
"current_mind": current_mind,
"observed_messages": observed_messages,
"action_prompt": prompt,
@@ -288,7 +302,7 @@ class ActionPlanner(BasePlanner):
if running_memorys:
memory_str = "以下是当前在聊天中,你回忆起的记忆:\n"
for running_memory in running_memorys:
memory_str += f"{running_memory['topic']}: {running_memory['content']}\n"
memory_str += f"{running_memory['content']}\n"
chat_context_description = "你现在正在一个群聊中"
chat_target_name = None # Only relevant for private