用户详情优化

api/app/services/memory_entity_relationship_service.py

@@ -15,6 +15,8 @@ from neo4j.time import DateTime as Neo4jDateTime
 import json
 from datetime import datetime
 
+from app.schemas.memory_episodic_schema import EmotionType
+
 logger = logging.getLogger(__name__)
 
 class MemoryEntityService:
@@ -496,11 +498,11 @@ class MemoryEmotion:
                 length_data.append(emotion_intensity)
             if emotion_type is not None and emotion_intensity is not None and formatted_created_at is not None:
                 # 使用(emotion_type, created_at)作为分组键
-                if emotion_type in {"joy", "surprise"}:
+                if emotion_type in {EmotionType.JOY_TYPE, EmotionType.SURPRISE_TYPE}:
                     emotion_type='positive'
-                elif emotion_type in {"sadness", "fear", "anger"}:
+                elif emotion_type in {EmotionType.SANDROWNESS_TYPE, EmotionType.FEAR_TYPE, EmotionType.ANGET_TYPE}:
                     emotion_type='negative'
-                elif emotion_type=='neutral':
+                elif emotion_type==EmotionType.NEUTRAL_TYPE:
                     emotion_type='neutral'
                 group_key = (emotion_type, formatted_created_at)
                 # 累加emotion_intensity