ReMe/reme/extension/procedural_memory/__init__.py

47 lines
1.4 KiB
Python

"""Procedural memory workflow."""
from ...core import R
from .dump_memory import DumpMemory
from .load_memory import LoadMemory
from .summary.trajectory_preprocess import TrajectoryPreprocess
from .summary.trajectory_segmentation import TrajectorySegmentation
from .summary.success_extraction import SuccessExtraction
from .summary.failure_extraction import FailureExtraction
from .summary.comparative_extraction import ComparativeExtraction
from .summary.memory_validation import MemoryValidation
from .summary.memory_deduplication import MemoryDeduplication
from .summary.memory_addition import MemoryAddition
from .retrieve.build_query import BuildQuery
from .retrieve.memory_deletion import MemoryDeletion
from .retrieve.memory_retrieval import MemoryRetrieval
from .retrieve.merge_memory import MergeMemory
from .retrieve.rerank_memory import RerankMemory
from .retrieve.rewrite_memory import RewriteMemory
from .retrieve.update_memory_metadata import UpdateMemoryMetadata
__all__ = [
"DumpMemory",
"LoadMemory",
"TrajectoryPreprocess",
"TrajectorySegmentation",
"SuccessExtraction",
"FailureExtraction",
"ComparativeExtraction",
"MemoryValidation",
"MemoryDeduplication",
"MemoryAddition",
"BuildQuery",
"MemoryDeletion",
"MemoryRetrieval",
"MergeMemory",
"RerankMemory",
"RewriteMemory",
"UpdateMemoryMetadata",
]
for name in __all__:
tool_class = globals()[name]
R.ops.register()(tool_class)