refactor: streamline import statements across memory management tools

This commit refactors the import statements in several memory management scripts, consolidating imports for better readability and maintainability. The changes include grouping related imports together and removing unnecessary line breaks, enhancing the overall structure of the code without altering functionality.

Author: 龚震宇

tools/delete_memory.py

@@ -9,17 +9,15 @@
 from dify_plugin import Tool
 
 from utils.config_builder import is_async_mode
-from utils.constants import DELETE_ACCEPT_RESULT, MAX_PENDING_TASKS_MULTIPLIER, WRITE_OPERATION_TIMEOUT
+from utils.constants import (
+    DELETE_ACCEPT_RESULT,
+    MAX_PENDING_TASKS_MULTIPLIER,
+    WRITE_OPERATION_TIMEOUT,
+)
 from utils.helpers import parse_timeout
 from utils.logger import get_logger
-from utils.mem0_client import (
-    get_async_client,
-    get_sync_client,
-)
-from utils.memory_tool_helpers import (
-    init_request_context,
-    yield_error,
-)
+from utils.mem0_client import get_async_client, get_sync_client
+from utils.memory_tool_helpers import init_request_context, yield_error
 
 if TYPE_CHECKING:
     from collections.abc import Generator

tools/extract_long_term_memory.py

@@ -56,13 +56,13 @@
     get_time_range_from_days,
     update_conv_checkpoint,
 )
-from utils.helpers import _parse_user_ids, dedup_keep_order
-from utils.helpers import parse_positive_int
-from utils.logger import get_logger
-from utils.mem0_client import (
-    AsyncMem0Client,
-    SyncMem0Client,
+from utils.helpers import (
+    _parse_user_ids,
+    dedup_keep_order,
+    parse_positive_int,
 )
+from utils.logger import get_logger
+from utils.mem0_client import AsyncMem0Client, SyncMem0Client
 from utils.mem0_extraction import (
     AsyncMemoryClassificationManager,
     AsyncMemoryWriter,

tools/update_memory.py

@@ -10,17 +10,15 @@
 from dify_plugin import Tool
 
 from utils.config_builder import is_async_mode
-from utils.constants import MAX_PENDING_TASKS_MULTIPLIER, UPDATE_ACCEPT_RESULT, WRITE_OPERATION_TIMEOUT
+from utils.constants import (
+    MAX_PENDING_TASKS_MULTIPLIER,
+    UPDATE_ACCEPT_RESULT,
+    WRITE_OPERATION_TIMEOUT,
+)
 from utils.helpers import parse_timeout
 from utils.logger import get_logger
-from utils.mem0_client import (
-    get_async_client,
-    get_sync_client,
-)
-from utils.memory_tool_helpers import (
-    init_request_context,
-    yield_error,
-)
+from utils.mem0_client import get_async_client, get_sync_client
+from utils.memory_tool_helpers import init_request_context, yield_error
 
 if TYPE_CHECKING:
     from collections.abc import Generator

utils/memory_tool_helpers.py

@@ -12,13 +12,9 @@
 from concurrent.futures import TimeoutError as FuturesTimeoutError
 from typing import TYPE_CHECKING, Any
 
-from utils.logger import get_logger
 from utils.constants import MAX_PENDING_TASKS_MULTIPLIER
-from utils.mem0_client import (
-    QueueOverloadError,
-    get_async_client,
-    get_sync_client,
-)
+from utils.logger import get_logger
+from utils.mem0_client import QueueOverloadError, get_async_client, get_sync_client
 
 if TYPE_CHECKING:
     from collections.abc import Generator