mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-14 23:21:04 +00:00
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""memory tools"""
|
|
|
|
from .base_memory_tool import BaseMemoryTool
|
|
from .delegate_task import DelegateTask
|
|
from .history.add_history import AddHistory
|
|
from .history.read_history import ReadHistory
|
|
from .profiles.add_draft_and_read_all_profiles import AddDraftAndReadAllProfiles
|
|
from .profiles.add_profile import AddProfile
|
|
from .profiles.delete_profile import DeleteProfile
|
|
from .profiles.profile_handler import ProfileHandler
|
|
from .profiles.read_all_profiles import ReadAllProfiles
|
|
from .profiles.update_profile import UpdateProfile
|
|
from .profiles.update_profiles_v1 import UpdateProfilesV1
|
|
from .vector.add_and_retrieve_similar_memory import AddAndRetrieveSimilarMemory
|
|
from .vector.add_draft_and_retrieve_similar_memory import AddDraftAndRetrieveSimilarMemory
|
|
from .vector.add_memory import AddMemory
|
|
from .vector.delete_memory import DeleteMemory
|
|
from .vector.memory_handler import MemoryHandler
|
|
from .vector.retrieve_memory import RetrieveMemory
|
|
from .vector.retrieve_recent_memory import RetrieveRecentMemory
|
|
from .vector.update_memory import UpdateMemory
|
|
from .vector.update_memory_v1 import UpdateMemoryV1
|
|
from .vector.update_memory_v2 import UpdateMemoryV2
|
|
from ...core import R
|
|
|
|
__all__ = [
|
|
# Base
|
|
"BaseMemoryTool",
|
|
"DelegateTask",
|
|
# History
|
|
"AddHistory",
|
|
"ReadHistory",
|
|
# Profiles
|
|
"AddDraftAndReadAllProfiles",
|
|
"AddProfile",
|
|
"ProfileHandler",
|
|
"ReadAllProfiles",
|
|
"UpdateProfile",
|
|
"DeleteProfile",
|
|
"UpdateProfilesV1",
|
|
# Vector
|
|
"AddAndRetrieveSimilarMemory",
|
|
"AddDraftAndRetrieveSimilarMemory",
|
|
"AddMemory",
|
|
"DeleteMemory",
|
|
"MemoryHandler",
|
|
"RetrieveMemory",
|
|
"RetrieveRecentMemory",
|
|
"UpdateMemory",
|
|
"UpdateMemoryV1",
|
|
"UpdateMemoryV2",
|
|
]
|
|
|
|
for name in __all__:
|
|
tool_class = globals()[name]
|
|
if isinstance(tool_class, type) and issubclass(tool_class, BaseMemoryTool) and tool_class is not BaseMemoryTool:
|
|
R.op.register(tool_class)
|