diff --git a/memory_scope/definition/memory_node.py b/memory_scope/definition/memory_node.py index 3ca84a86..70eb9410 100644 --- a/memory_scope/definition/memory_node.py +++ b/memory_scope/definition/memory_node.py @@ -10,6 +10,12 @@ class MemoryNode(BaseModel): memoryId: str = Field("", description="unique memory id") + 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") + memoryType: str = Field("", description="conversation/observation/insight...") metaData: Dict[str, str] = Field({}, description="other data infos") diff --git a/memory_scope/definition/memory_wrap_node.py b/memory_scope/definition/memory_wrap_node.py deleted file mode 100644 index 8533e31f..00000000 --- a/memory_scope/definition/memory_wrap_node.py +++ /dev/null @@ -1,15 +0,0 @@ -from pydantic import Field, BaseModel - -from memory_scope.definition.memory_node import MemoryNode - - -class MemoryWrapNode(BaseModel): - id: str = Field("", description="uuid64") - - score_similar: float = Field(0, description="相似度打分") - - score_rank: float = Field(0, description="排序打分") - - score_rerank: float = Field(0, description="重排打分") - - memory_node: MemoryNode = Field(None, description="memory node 核心,返回给上游的结构")