ReMe/memory_scope/scheme/memory_node.py
2024-06-25 19:49:48 +08:00

26 lines
822 B
Python

from typing import Dict, List
from pydantic import Field, BaseModel
class MemoryNode(BaseModel):
user_id: str = Field("", description="unique memory id for user")
memory_id: str = Field("", description="unique id for memory item")
content: str = Field("", description="memory content")
score_similar: float = Field(0, description="es similar score")
score_rank: float = Field(0, description="rank model score")
score_rerank: float = Field(0, description="rerank score")
memory_type: str = Field("", description="conversation/observation/insight...")
meta_data: Dict[str, str] = Field({}, description="other data infos")
status: str = Field("active", description="active or expired")
vector: List[float] = Field([], description="content embedding result, return empty")