From d62c6a22a5f8444fc57a0cb5ccfbfcb0d2fd5ef1 Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Sat, 7 Mar 2026 15:30:45 +0800
Subject: [PATCH] refactor(memory): move file utility functions to shared
 module

---
 reme/__init__.py                                            | 2 +-
 reme/memory/file_based/tools/file_io.py                     | 2 +-
 reme/memory/file_based/tools/shell.py                       | 2 +-
 reme/memory/file_based/utils/__init__.py                    | 6 ++++++
 .../file_based/{tools/utils.py => utils/file_utils.py}      | 0
 5 files changed, 9 insertions(+), 3 deletions(-)
 rename reme/memory/file_based/{tools/utils.py => utils/file_utils.py} (100%)

diff --git a/reme/__init__.py b/reme/__init__.py
index 9842ce1a..c75dfbcc 100644
--- a/reme/__init__.py
+++ b/reme/__init__.py
@@ -6,7 +6,7 @@ from . import extension
 from . import memory
 from .reme import ReMe
 
-__version__ = "0.3.0.6b1"
+__version__ = "0.3.0.6b2"
 
 __all__ = [
     "config",
diff --git a/reme/memory/file_based/tools/file_io.py b/reme/memory/file_based/tools/file_io.py
index 58e9a5f7..2b792475 100644
--- a/reme/memory/file_based/tools/file_io.py
+++ b/reme/memory/file_based/tools/file_io.py
@@ -7,7 +7,7 @@ from typing import Optional
 from agentscope.message import TextBlock
 from agentscope.tool import ToolResponse
 
-from .utils import DEFAULT_MAX_BYTES, read_file_safe, truncate_output
+from ..utils import DEFAULT_MAX_BYTES, read_file_safe, truncate_output
 
 
 class FileIO:
diff --git a/reme/memory/file_based/tools/shell.py b/reme/memory/file_based/tools/shell.py
index c2714b87..2bee1bd6 100644
--- a/reme/memory/file_based/tools/shell.py
+++ b/reme/memory/file_based/tools/shell.py
@@ -12,7 +12,7 @@ from pathlib import Path
 from agentscope.message import TextBlock
 from agentscope.tool import ToolResponse
 
-from .utils import truncate_shell_output
+from ..utils import truncate_shell_output
 
 
 def _execute_subprocess_sync(
diff --git a/reme/memory/file_based/utils/__init__.py b/reme/memory/file_based/utils/__init__.py
index 6249ab15..48231232 100644
--- a/reme/memory/file_based/utils/__init__.py
+++ b/reme/memory/file_based/utils/__init__.py
@@ -1,7 +1,13 @@
 """utils"""
 
 from .as_msg_handler import AsMsgHandler
+from .file_utils import truncate_output, truncate_shell_output, read_file_safe, DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES
 
 __all__ = [
     "AsMsgHandler",
+    "truncate_output",
+    "truncate_shell_output",
+    "read_file_safe",
+    "DEFAULT_MAX_BYTES",
+    "DEFAULT_MAX_LINES",
 ]
diff --git a/reme/memory/file_based/tools/utils.py b/reme/memory/file_based/utils/file_utils.py
similarity index 100%
rename from reme/memory/file_based/tools/utils.py
rename to reme/memory/file_based/utils/file_utils.py