From 9fb96ae15eecda09a4d35bf0e492b9ed2e5ea7e6 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Fri, 21 Jun 2024 11:33:49 +0800 Subject: [PATCH] [dev] add scores to memory node --- memory_scope/definition/memory_node.py | 6 ++++++ memory_scope/definition/memory_wrap_node.py | 15 --------------- 2 files changed, 6 insertions(+), 15 deletions(-) delete mode 100644 memory_scope/definition/memory_wrap_node.py 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 核心，返回给上游的结构")