diff --git a/reme/agent/memory/personal/personal_v1_retriever.py b/reme/agent/memory/personal/personal_v1_retriever.py index 1c3ba9ef..525d597a 100644 --- a/reme/agent/memory/personal/personal_v1_retriever.py +++ b/reme/agent/memory/personal/personal_v1_retriever.py @@ -97,7 +97,7 @@ class PersonalV1Retriever(BaseMemoryAgent): include_memory_id=False, include_when_to_use=False, include_content=True, - include_message_time=False, + include_message_time=True, ref_memory_id_key="", ) for n in self.retrieved_nodes diff --git a/reme/core/memory_storage/__init__.py b/reme/core/memory_storage/__init__.py deleted file mode 100644 index 371d4e0f..00000000 --- a/reme/core/memory_storage/__init__.py +++ /dev/null @@ -1,16 +0,0 @@ -"""Memory storage module for persistent memory management. - -This module provides storage backends for memory chunks and file metadata, -including SQLite-based implementations with vector and full-text search. -""" - -from .base_memory_store import BaseMemoryStore -from .sqlite_memory_store import SqliteMemoryStore -from ..context import R - -__all__ = [ - "BaseMemoryStore", - "SqliteMemoryStore", -] - -R.memory_store.register("sqlite")(SqliteMemoryStore) diff --git a/reme/core/memory_store/sqlite_memory_store.py b/reme/core/memory_store/sqlite_memory_store.py index e5a32bb6..911d4a00 100644 --- a/reme/core/memory_store/sqlite_memory_store.py +++ b/reme/core/memory_store/sqlite_memory_store.py @@ -548,6 +548,7 @@ class SqliteMemoryStore(BaseMemoryStore): score=score, snippet=snippet, source=MemorySource(src), + distance=dist, ), ) diff --git a/reme/core/schema/memory_search_result.py b/reme/core/schema/memory_search_result.py index 784fe7c7..24b2a789 100644 --- a/reme/core/schema/memory_search_result.py +++ b/reme/core/schema/memory_search_result.py @@ -14,6 +14,7 @@ class MemorySearchResult(BaseModel): score: float = Field(..., description="Relevance score of the search result") snippet: str = Field(..., description="Text snippet from the matched content") source: MemorySource = Field(..., description="Source of the memory data") + distance: float | None = Field(None, description="Original distance value from vector search") metadata: dict = Field(default_factory=dict, description="Additional metadata") @property diff --git a/reme/reme_fs.py b/reme/reme_fs.py index c28d9c32..4ef7c149 100644 --- a/reme/reme_fs.py +++ b/reme/reme_fs.py @@ -26,6 +26,8 @@ from .tool.fs import ( ReadTool, WriteTool, ) +from .tool.gallery import ExecuteCode +from .tool.search import DashscopeSearch class ReMeFs(Application): @@ -100,6 +102,8 @@ class ReMeFs(Application): LsTool(cwd=self.working_dir), ReadTool(cwd=self.working_dir), WriteTool(cwd=self.working_dir), + ExecuteCode(), + DashscopeSearch(), ] # Commands