Skip to content

Commit 7a2a2b2

Browse files
justin-cechmanekjustin-cechmanek
committed
wraps message history classes for backward compatibility. improves warnings
1 parent a41e1df commit 7a2a2b2

File tree

7 files changed

+54
-33
lines changed

7 files changed

+54
-33
lines changed

‎redisvl/extensions/message_history/__init__.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
from redisvl.extensions.message_history.base_history import BaseMessageHistory
22
from redisvl.extensions.message_history.message_history import MessageHistory
3-
from redisvl.extensions.message_history.semantic_message_history import (
4-
SemanticMessageHistory,
5-
)
3+
from redisvl.extensions.message_history.semantic_history import SemanticMessageHistory
64

75
__all__ = ["BaseMessageHistory", "MessageHistory", "SemanticMessageHistory"]

redisvl/extensions/message_history/semantic_message_history.py renamed to redisvl/extensions/message_history/semantic_history.py

File renamed without changes.

‎redisvl/extensions/session_manager/__init__.py‎

Lines changed: 7 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -6,27 +6,18 @@
66

77
import warnings
88

9-
from redisvl.extensions.message_history.message_history import MessageHistory
10-
from redisvl.extensions.message_history.schema import (
11-
ChatMessage,
12-
MessageHistorySchema,
13-
SemanticMessageHistorySchema,
14-
)
15-
from redisvl.extensions.message_history.semantic_message_history import (
16-
SemanticMessageHistory,
17-
)
9+
from redisvl.extensions.session_manager.base_session import BaseSessionManager
10+
from redisvl.extensions.session_manager.semantic_session import SemanticSessionManager
11+
from redisvl.extensions.session_manager.standard_session import StandardSessionManager
1812

1913
warnings.warn(
2014
"Importing from redisvl.extensions.session_manager is deprecated. "
15+
"StandardSessionManager has been renamed to MessageHistory. "
16+
"SemanticSessionManager has been renamed to SemanticMessageHistory. "
2117
"Please import from redisvl.extensions.message_history instead.",
2218
DeprecationWarning,
2319
stacklevel=2,
2420
)
2521

26-
__all__ = [
27-
"MessageHistory",
28-
"SemanticMessageHistory",
29-
"ChatMessage",
30-
"MessageHistorySchema",
31-
"SemanticMessageHistorySchema",
32-
]
22+
23+
__all__ = ["BaseSessionManager", "StandardSessionManager", "SemanticSessionManager"]

‎redisvl/extensions/session_manager/base_session.py‎

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,25 @@
11
"""
22
RedisVL Standard Session Manager (Deprecated Path)
33
4-
This module is kept for backward compatibility. Please use `redisvl.extensions.standard_history` instead.
4+
This module is kept for backward compatibility. Please use `redisvl.extensions.base_history` instead.
55
"""
66

77
import warnings
88

9-
from redisvl.extensions.message_history.standard_history import StandardHistory
9+
from redisvl.extensions.message_history.base_history import BaseMessageHistory
1010

1111
warnings.warn(
12-
"Importing from redisvl.extensions.session_manager.standard_session is deprecated. "
13-
"Please import from redisvl.extensions.message_history instead.",
12+
"Importing from redisvl.extensions.session_manager.base_session is deprecated. "
13+
"BaseSessionManager has been renamed to BaseMessageHistory. "
14+
"Please import BaseMessageHistory from redisvl.extensions.base_history instead.",
1415
DeprecationWarning,
1516
stacklevel=2,
1617
)
1718

18-
__all__ = ["StandardHistory"]
19+
20+
class BaseSessionManager(BaseMessageHistory):
21+
# keep for backward compatibility
22+
pass
23+
24+
25+
__all__ = ["BaseSessionManager"]

‎redisvl/extensions/session_manager/schema.py‎

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88

99
from redisvl.extensions.message_history.schema import (
1010
ChatMessage,
11-
SemanticMessageHistorySchema,
1211
MessageHistorySchema,
12+
SemanticMessageHistorySchema,
1313
)
1414

1515
warnings.warn(
@@ -19,8 +19,19 @@
1919
stacklevel=2,
2020
)
2121

22+
23+
class StandardSessionIndexSchema(MessageHistorySchema):
24+
# keep for backward compatibility
25+
pass
26+
27+
28+
class SemanticSessionIndexSchema(SemanticMessageHistorySchema):
29+
# keep for backward compatibility
30+
pass
31+
32+
2233
__all__ = [
2334
"ChatMessage",
24-
"MessageHistory",
25-
"SemanticMessageHistory",
35+
"StandardSessionIndexSchema",
36+
"SemanticSessionIndexSchema",
2637
]

‎redisvl/extensions/session_manager/semantic_session.py‎

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,20 @@
66

77
import warnings
88

9-
from redisvl.extensions.message_history.semantic_history import SemanticHistory
9+
from redisvl.extensions.message_history.semantic_history import SemanticMessageHistory
1010

1111
warnings.warn(
1212
"Importing from redisvl.extensions.session_manger.semantic_session is deprecated. "
13-
"Please import from redisvl.extensions.semantic_history instead.",
13+
"SemanticSessionManager has been renamed to SemanticMessageHistory. "
14+
"Please import SemanticMessageHistory from redisvl.extensions.semantic_history instead.",
1415
DeprecationWarning,
1516
stacklevel=2,
1617
)
1718

18-
__all__ = ["SemanticHistory"]
19+
20+
class SemanticSessionManager(SemanticMessageHistory):
21+
# keep for backwards compatibility
22+
pass
23+
24+
25+
__all__ = ["SemanticSessionManager"]

‎redisvl/extensions/session_manager/standard_session.py‎

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,20 @@
66

77
import warnings
88

9-
from redisvl.extensions.message_history.standard_history import MessageHistory
9+
from redisvl.extensions.message_history.message_history import MessageHistory
1010

1111
warnings.warn(
1212
"Importing from redisvl.extensions.session_manger.standard_session is deprecated. "
13-
"Please import from redisvl.extensions.message_history instead.",
13+
"StandardSessionManager has been renamed to MessageHistory. "
14+
"Please import MessageHistory from redisvl.extensions.message_history instead.",
1415
DeprecationWarning,
1516
stacklevel=2,
1617
)
1718

18-
__all__ = ["MessageHistory"]
19+
20+
class StandardSessionManager(MessageHistory):
21+
# keep for backward compatibility
22+
pass
23+
24+
25+
__all__ = ["StandardSessionManager"]

0 commit comments

Comments
 (0)