From 423cf2f2625344d96f2177aeab62834142b997c3 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Wed, 27 Aug 2025 11:07:02 +0800 Subject: [PATCH] refactor(import): update import statements and rename classes across multiple modules - Update import statements to use relative imports where appropriate - Rename classes to match file names for better consistency - Add new imports for recently added modules --- reme_ai/react/__init__.py | 1 + reme_ai/retrieve/__init__.py | 4 ++-- reme_ai/retrieve/task/__init__.py | 6 ++++-- reme_ai/summary/__init__.py | 4 ++-- reme_ai/summary/task/__init__.py | 11 ++++++++++- reme_ai/summary/task/memory_deduplication_op.py | 2 +- 6 files changed, 20 insertions(+), 8 deletions(-) diff --git a/reme_ai/react/__init__.py b/reme_ai/react/__init__.py index e69de29b..f49ce6d5 100644 --- a/reme_ai/react/__init__.py +++ b/reme_ai/react/__init__.py @@ -0,0 +1 @@ +from .react_v1_op import ReactV1Op diff --git a/reme_ai/retrieve/__init__.py b/reme_ai/retrieve/__init__.py index 53a36c1f..0d054f82 100644 --- a/reme_ai/retrieve/__init__.py +++ b/reme_ai/retrieve/__init__.py @@ -1,2 +1,2 @@ -from reme_ai.retrieve import task -from reme_ai.retrieve import personal +from . import task +from . import personal diff --git a/reme_ai/retrieve/task/__init__.py b/reme_ai/retrieve/task/__init__.py index 88fed8b8..cf6c1ded 100644 --- a/reme_ai/retrieve/task/__init__.py +++ b/reme_ai/retrieve/task/__init__.py @@ -1,2 +1,4 @@ -from reme_ai.retrieve.task.build_query_op import BuildQueryOp -from reme_ai.retrieve.task.merge_memory_op import MergeMemoryOp \ No newline at end of file +from .build_query_op import BuildQueryOp +from .merge_memory_op import MergeMemoryOp +from .rerank_memory_op import RerankMemoryOp +from .rewrite_memory_op import RewriteMemoryOp diff --git a/reme_ai/summary/__init__.py b/reme_ai/summary/__init__.py index 1f47c200..dec8dbf7 100644 --- a/reme_ai/summary/__init__.py +++ b/reme_ai/summary/__init__.py @@ -1,2 +1,2 @@ -from reme_ai.summary import task -from reme_ai.summary import personal \ No newline at end of file +from . import task +from . import personal \ No newline at end of file diff --git a/reme_ai/summary/task/__init__.py b/reme_ai/summary/task/__init__.py index c6b143af..169f8958 100644 --- a/reme_ai/summary/task/__init__.py +++ b/reme_ai/summary/task/__init__.py @@ -1 +1,10 @@ -from reme_ai.summary.task.simple_summary_op import SimpleSummaryOp \ No newline at end of file +from .comparative_extraction_op import ComparativeExtractionOp +from .failure_extraction_op import FailureExtractionOp +from .memory_deduplication_op import MemoryDeduplicationOp +from .memory_validation_op import MemoryValidationOp +from .pdf_preprocess_op_wrapper import PDFPreprocessOp +from .simple_comparative_summary_op import SimpleComparativeSummaryOp +from .simple_summary_op import SimpleSummaryOp +from .success_extraction_op import SuccessExtractionOp +from .trajectory_preprocess_op import TrajectoryPreprocessOp +from .trajectory_segmentation_op import TrajectorySegmentationOp diff --git a/reme_ai/summary/task/memory_deduplication_op.py b/reme_ai/summary/task/memory_deduplication_op.py index 8982a576..a8d33266 100644 --- a/reme_ai/summary/task/memory_deduplication_op.py +++ b/reme_ai/summary/task/memory_deduplication_op.py @@ -6,7 +6,7 @@ from reme_ai.schema.memory import BaseMemory @C.register_op() -class TaskMemoryDeduplicationOp(BaseOp): +class MemoryDeduplicationOp(BaseOp): file_path: str = __file__ def execute(self):