Fix/memory bug fix (#111)

* 图谱数据量限制数量去掉

* 图谱数据量限制数量去掉

* 图谱数据量限制数量去掉

* 用户详情优化

* 用户详情优化

* 用户详情优化

* 用户详情优化

* 用户详情优化
This commit is contained in:
lixinyue11
2026-01-14 15:36:26 +08:00
committed by GitHub
parent fee22f83c9
commit 93ff64f130
3 changed files with 60 additions and 7 deletions

View File

@@ -1,9 +1,51 @@
"""
情景记忆的请求和响应模型
"""
from abc import ABC
from pydantic import BaseModel, Field
from typing import Optional
type_mapping = {
"Person": "人物实体节点",
"Organization": "组织实体节点",
"ORG": "组织实体节点",
"Location": "地点实体节点",
"LOC": "地点实体节点",
"Event": "事件实体节点",
"Concept": "概念实体节点",
"Time": "时间实体节点",
"Position": "职位实体节点",
"WorkRole": "职业实体节点",
"System": "系统实体节点",
"Policy": "政策实体节点",
"HistoricalPeriod": "历史时期实体节点",
"HistoricalState": "历史国家实体节点",
"HistoricalEvent": "历史事件实体节点",
"EconomicFactor": "经济因素实体节点",
"Condition": "条件实体节点",
"Numeric": "数值实体节点"
}
class EmotionType(ABC):
JOY_TYPE = "joy"
SURPRISE_TYPE = "surprise"
SANDROWNESS_TYPE = "sadness"
FEAR_TYPE = "fear"
ANGET_TYPE="anger"
NEUTRAL_TYPE="neutral"
EMOTION_MAPPING={
"joy":"愉快",
"surprise":"惊喜",
"sadness":"悲伤",
"fear":"恐惧",
"anger":"生气",
"neutral":"中性"
}
class EmotionSubject(ABC):
SUBJECT_MAPPING={
"self":"自己",
"other":"别人",
"object":"事物对象"
}
class EpisodicMemoryOverviewRequest(BaseModel):
"""情景记忆总览查询请求"""