We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 07a47b7 commit ffb64b7Copy full SHA for ffb64b7
src/memos/mem_scheduler/utils/status_tracker.py
@@ -1,10 +1,13 @@
1
# src/memos/mem_scheduler/utils/status_tracker.py
2
import json
3
-import redis
4
from datetime import datetime, timezone
5
+from memos.dependency import require_python_package
6
+
7
8
class TaskStatusTracker:
- def __init__(self, redis_client: redis.Redis):
9
+ @require_python_package(import_name="redis", install_command="pip install redis")
10
+ def __init__(self, redis_client: "redis.Redis"):
11
self.redis = redis_client
12
13
def _get_key(self, user_id: str) -> str:
0 commit comments