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