scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-05 08:06:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

[dev] add scores to memory node

Browse source
This commit is contained in:
jinli.yl 2024-06-21 11:33:49 +08:00
parent 0d51ff56d3
commit 9fb96ae15e
2 changed files with 6 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
memory_scope/definition/memory_node.py
View file

@ -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")

15
memory_scope/definition/memory_wrap_node.py
View file

@ -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 核心，返回给上游的结构")