adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

用户详情优化

Browse Source
This commit is contained in:
lixinyue
2026-01-14 15:34:45 +08:00
parent 6db37d35ed
commit 2c9e5df27d
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/app/schemas/memory_episodic_schema.py
View File

@@ -40,6 +40,12 @@ class EmotionType(ABC):
"anger":"生气", "anger":"生气",
"neutral":"中性" "neutral":"中性"
} }
class EmotionSubject(ABC):
SUBJECT_MAPPING={
"self":"自己",
"other":"别人",
"object":"事物对象"
}
class EpisodicMemoryOverviewRequest(BaseModel): class EpisodicMemoryOverviewRequest(BaseModel):
"""情景记忆总览查询请求""" """情景记忆总览查询请求"""

4
api/app/services/user_memory_service.py
View File

@@ -15,7 +15,7 @@ from app.core.memory.utils.llm.llm_utils import MemoryClientFactory
from app.db import get_db_context from app.db import get_db_context
from app.repositories.end_user_repository import EndUserRepository from app.repositories.end_user_repository import EndUserRepository
from app.repositories.neo4j.neo4j_connector import Neo4jConnector from app.repositories.neo4j.neo4j_connector import Neo4jConnector
from app.schemas.memory_episodic_schema import type_mapping, EmotionType from app.schemas.memory_episodic_schema import type_mapping, EmotionType, EmotionSubject
from app.services.memory_base_service import MemoryBaseService from app.services.memory_base_service import MemoryBaseService
from app.services.memory_config_service import MemoryConfigService from app.services.memory_config_service import MemoryConfigService
@@ -1584,6 +1584,8 @@ async def _extract_node_properties(label: str, properties: Dict[str, Any],node_
value=type_mapping.get(value,'') value=type_mapping.get(value,'')
if str(field)=="emotion_type": if str(field)=="emotion_type":
value=EmotionType.EMOTION_MAPPING.get(value) value=EmotionType.EMOTION_MAPPING.get(value)
if str(field)=="emotion_subject":
value=EmotionSubject.SUBJECT_MAPPING.get(value)
# 清理 Neo4j 特殊类型 # 清理 Neo4j 特殊类型
filtered_props[field] = _clean_neo4j_value(value) filtered_props[field] = _clean_neo4j_value(value)
filtered_props['associative_memory']=[i['rel_count'] for i in node_results][0] filtered_props['associative_memory']=[i['rel_count'] for i in node_results][0]