Skip to content

Commit ffb64b7

Browse files
fix(ci): Resolve top-level redis import error in TaskStatusTracker
1 parent 07a47b7 commit ffb64b7

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/memos/mem_scheduler/utils/status_tracker.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,13 @@
11
# src/memos/mem_scheduler/utils/status_tracker.py
22
import json
3-
import redis
43
from datetime import datetime, timezone
54

5+
from memos.dependency import require_python_package
6+
7+
68
class TaskStatusTracker:
7-
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"):
811
self.redis = redis_client
912

1013
def _get_key(self, user_id: str) -> str:

0 commit comments

Comments
 (0)