feat: 更新数据库模型,添加个人信息存储模型
This commit is contained in:
@@ -6,14 +6,14 @@ from src.chat.utils.chat_message_builder import build_readable_messages, get_raw
|
|||||||
from src.chat.person_info.relationship_manager import relationship_manager
|
from src.chat.person_info.relationship_manager import relationship_manager
|
||||||
from src.chat.utils.utils import get_embedding
|
from src.chat.utils.utils import get_embedding
|
||||||
import time
|
import time
|
||||||
from typing import Union, Optional, Dict, Any
|
from typing import Union, Optional
|
||||||
from common.database.database import db
|
from common.database.database import db
|
||||||
from src.chat.utils.utils import get_recent_group_speaker
|
from src.chat.utils.utils import get_recent_group_speaker
|
||||||
from src.manager.mood_manager import mood_manager
|
from src.manager.mood_manager import mood_manager
|
||||||
from src.chat.memory_system.Hippocampus import HippocampusManager
|
from src.chat.memory_system.Hippocampus import HippocampusManager
|
||||||
from src.chat.knowledge.knowledge_lib import qa_manager
|
from src.chat.knowledge.knowledge_lib import qa_manager
|
||||||
from src.chat.focus_chat.expressors.exprssion_learner import expression_learner
|
from src.chat.focus_chat.expressors.exprssion_learner import expression_learner
|
||||||
import traceback
|
# import traceback
|
||||||
import random
|
import random
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -182,3 +182,23 @@ class OnlineTime(BaseModel):
|
|||||||
# database = db # 继承自 BaseModel
|
# database = db # 继承自 BaseModel
|
||||||
table_name = 'online_time'
|
table_name = 'online_time'
|
||||||
|
|
||||||
|
class PersonInfo(BaseModel):
|
||||||
|
"""
|
||||||
|
用于存储个人信息数据的模型。
|
||||||
|
"""
|
||||||
|
person_id = TextField(unique=True, index=True) # 个人唯一ID
|
||||||
|
person_name = TextField() # 个人名称
|
||||||
|
name_reason = TextField(null=True) # 名称设定的原因
|
||||||
|
platform = TextField() # 平台
|
||||||
|
user_id = TextField(index=True) # 用户ID
|
||||||
|
nickname = TextField() # 用户昵称
|
||||||
|
relationship_value = IntegerField(default=0) # 关系值
|
||||||
|
konw_time = FloatField() # 认识时间 (时间戳)
|
||||||
|
msg_interval = IntegerField() # 消息间隔
|
||||||
|
# msg_interval_list: 存储为 JSON 字符串的列表
|
||||||
|
msg_interval_list = TextField(null=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
# database = db # 继承自 BaseModel
|
||||||
|
table_name = 'person_info'
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user