auto modification by running ruff

Author: tangg555

scripts/check_dependencies.py

@@ -11,7 +11,7 @@
 
 def extract_top_level_modules(tree: ast.Module) -> set[str]:
     """
-    Extract all top-level imported modules (excluding relative imports).
+    Extract all top-level imported general_modules (excluding relative imports).
     """
     modules = set()
     for node in tree.body:
@@ -27,12 +27,12 @@ def extract_top_level_modules(tree: ast.Module) -> set[str]:
 def check_importable(modules: set[str], filename: str) -> list[str]:
     """
     Attempt to import each module in the current environment.
-    Return a list of modules that fail to import.
+    Return a list of general_modules that fail to import.
     """
     failed = []
     for mod in sorted(modules):
         if mod in EXCLUDE_MODULES:
-            # Skip excluded modules such as your own package
+            # Skip excluded general_modules such as your own package
             continue
         try:
             importlib.import_module(mod)
@@ -70,7 +70,7 @@ def main():
 
     if has_error:
         print(
-            "\n💥 Top-level imports failed. These modules may not be main dependencies."
+            "\n💥 Top-level imports failed. These general_modules may not be main dependencies."
             " Try moving the imports to a function or class scope, and decorate it with @require_python_package."
         )
         sys.exit(1)

tests/mem_reader/test_memory.py

@@ -7,7 +7,7 @@ def test_memory_initialization():
     """Test initialization of Memory class."""
     user_id = "user123"
     session_id = "session456"
-    created_at = datetime.now()
+    created_at = datetime.utcnow()
 
     memory = Memory(user_id=user_id, session_id=session_id, created_at=created_at)
 

tests/mem_scheduler/test_retriever.py

@@ -37,7 +37,7 @@ def setUp(self):
         self.mem_cube.text_mem = self.tree_text_memory
         self.mem_cube.act_mem = MagicMock()
 
-        # Initialize modules with mock LLM
+        # Initialize general_modules with mock LLM
         self.scheduler.initialize_modules(chat_llm=self.llm, process_llm=self.llm)
         self.scheduler.mem_cube = self.mem_cube
 
@@ -47,7 +47,7 @@ def setUp(self):
         self.logging_warning_patch = patch("logging.warning")
         self.mock_logging_warning = self.logging_warning_patch.start()
 
-        self.logger_info_patch = patch("memos.mem_scheduler.modules.retriever.logger.info")
+        self.logger_info_patch = patch("memos.mem_scheduler.general_modules.retriever.logger.info")
         self.mock_logger_info = self.logger_info_patch.start()
 
     def tearDown(self):

tests/mem_scheduler/test_scheduler.py

@@ -8,8 +8,8 @@
 from memos.configs.mem_scheduler import SchedulerConfigFactory
 from memos.llms.base import BaseLLM
 from memos.mem_cube.general import GeneralMemCube
-from memos.mem_scheduler.modules.monitor import SchedulerMonitor
-from memos.mem_scheduler.modules.retriever import SchedulerRetriever
+from memos.mem_scheduler.general_modules.retriever import SchedulerRetriever
+from memos.mem_scheduler.monitors.general_monitor import SchedulerGeneralMonitor
 from memos.mem_scheduler.scheduler_factory import SchedulerFactory
 from memos.mem_scheduler.schemas.general_schemas import (
     ANSWER_LABEL,
@@ -43,7 +43,7 @@ def setUp(self):
         self.mem_cube.text_mem = self.tree_text_memory
         self.mem_cube.act_mem = MagicMock()
 
-        # Initialize modules with mock LLM
+        # Initialize general_modules with mock LLM
         self.scheduler.initialize_modules(chat_llm=self.llm, process_llm=self.llm)
         self.scheduler.mem_cube = self.mem_cube
 
@@ -60,7 +60,7 @@ def test_initialization(self):
     def test_initialize_modules(self):
         """Test module initialization with proper component assignments."""
         self.assertEqual(self.scheduler.chat_llm, self.llm)
-        self.assertIsInstance(self.scheduler.monitor, SchedulerMonitor)
+        self.assertIsInstance(self.scheduler.monitor, SchedulerGeneralMonitor)
         self.assertIsInstance(self.scheduler.retriever, SchedulerRetriever)
 
     def test_submit_web_logs(self):