Skip to content

fix(memory,persona): resolve AttributeError in graph and updater #44

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
NickCharlie wants to merge 1 commit into from
Closed

fix(memory,persona): resolve AttributeError in graph and updater #44

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions services/memory_graph_manager.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -487,7 +487,7 @@ async def add_memory_from_message(self, message: MessageData, group_id: str):
# 添加记忆节点
await memory_graph.add_memory_node(
concept=concept,
memory=message.content,
memory=message.message,
llm_adapter=self.llm_adapter
)

Expand Down Expand Up @@ -516,7 +516,7 @@ async def _extract_concepts_from_message(self, message: MessageData) -> List[str
try:
from ..statics.prompts import ENTITY_EXTRACTION_PROMPT

prompt = ENTITY_EXTRACTION_PROMPT.format(text=message.content)
prompt = ENTITY_EXTRACTION_PROMPT.format(text=message.message)

response = await self.llm_adapter.generate_response(
prompt,
Expand Down
15 changes: 5 additions & 10 deletions services/persona_updater.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def clone_persona_data(persona_data: Any) -> Dict[str, Any]:
return True

except Exception as e:
self._logger.error(f"人格更新失败 for group {group_id}: {e}")
self._logger.error(f"人格更新失败 for group {group_id}: {e}", exc_info=True)
raise SelfLearningError(f"人格更新失败: {str(e)}")

async def record_persona_update_for_review(self, record: PersonaUpdateRecord) -> int:
Expand Down Expand Up @@ -413,7 +413,7 @@ async def get_current_persona_description(self, group_id: str) -> Optional[str]:
return persona.get('prompt', '')
return None
except Exception as e:
self._logger.error(f"获取当前人格描述失败 for group {group_id}: {e}")
self._logger.error(f"获取当前人格描述失败 for group {group_id}: {e}", exc_info=True)
return None

async def get_current_persona(self, group_id: str) -> Optional[Dict[str, Any]]:
Expand All @@ -430,7 +430,7 @@ async def get_current_persona(self, group_id: str) -> Optional[Dict[str, Any]]:
return None

except Exception as e:
self._logger.error(f"获取当前人格失败 for group {group_id}: {e}")
self._logger.error(f"获取当前人格失败 for group {group_id}: {e}", exc_info=True)
return None

def _merge_prompts(self, original: str, enhancement: str) -> str:
Expand Down Expand Up @@ -580,14 +580,9 @@ async def _update_style_based_features_with_maibot(self, current_persona: Person

# 2. 更新记忆图谱
if hasattr(self, 'memory_graph_manager') and self.memory_graph_manager:
group_id = current_persona.get('group_id', 'default') if isinstance(current_persona, dict) else 'default'
for msg in filtered_messages:
await self.memory_graph_manager.add_memory_node(
memory_id=f"msg_{msg.timestamp}",
content=msg.message,
memory_type="dialogue_style",
tags=['style_learning'],
metadata={'sender': msg.sender_name, 'group_id': msg.group_id}
)
await self.memory_graph_manager.add_memory_from_message(msg, group_id)
self._logger.info(f"向记忆图谱添加了 {len(filtered_messages)} 个风格记忆节点")

# 3. 更新知识图谱
Expand Down