impressmem 0.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- impressmem/__init__.py +23 -0
- impressmem/config.py +17 -0
- impressmem/impression_manager.py +704 -0
- impressmem/tools/__init__.py +9 -0
- impressmem/tools/base.py +38 -0
- impressmem/tools/organize_impressions.py +186 -0
- impressmem/tools/recall_impressions.py +190 -0
- impressmem/tools/save_impression.py +140 -0
- impressmem/utils.py +34 -0
- impressmem-0.1.0.dist-info/METADATA +190 -0
- impressmem-0.1.0.dist-info/RECORD +14 -0
- impressmem-0.1.0.dist-info/WHEEL +5 -0
- impressmem-0.1.0.dist-info/licenses/LICENSE +21 -0
- impressmem-0.1.0.dist-info/top_level.txt +1 -0
impressmem/__init__.py
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
"""
|
|
2
|
+
ImpressMem - Impression-based Memory Management Library
|
|
3
|
+
"""
|
|
4
|
+
|
|
5
|
+
__version__ = "0.1.0"
|
|
6
|
+
|
|
7
|
+
from impressmem.config import Config
|
|
8
|
+
from impressmem.impression_manager import ImpressionManager
|
|
9
|
+
from impressmem.tools import (
|
|
10
|
+
SaveImpressionTool,
|
|
11
|
+
OrganizeImpressionsTool,
|
|
12
|
+
RecallImpressionsTool,
|
|
13
|
+
)
|
|
14
|
+
from impressmem.utils import slice_new_turn_messages
|
|
15
|
+
|
|
16
|
+
__all__ = [
|
|
17
|
+
"Config",
|
|
18
|
+
"ImpressionManager",
|
|
19
|
+
"SaveImpressionTool",
|
|
20
|
+
"OrganizeImpressionsTool",
|
|
21
|
+
"RecallImpressionsTool",
|
|
22
|
+
"slice_new_turn_messages",
|
|
23
|
+
]
|
impressmem/config.py
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
"""
|
|
2
|
+
Configuration for ImpressMem
|
|
3
|
+
"""
|
|
4
|
+
from dataclasses import dataclass, field
|
|
5
|
+
from typing import Dict, Any
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
@dataclass
|
|
9
|
+
class Config:
|
|
10
|
+
"""Configuration class for ImpressMem
|
|
11
|
+
|
|
12
|
+
Args:
|
|
13
|
+
bot_name: Name of the bot/assistant (default: "Bot")
|
|
14
|
+
redis_config: Redis configuration dictionary (default: {"host": "localhost", "port": 6379, "db": 0})
|
|
15
|
+
"""
|
|
16
|
+
bot_name: str = "Bot"
|
|
17
|
+
redis_config: Dict[str, Any] = field(default_factory=lambda: {"host": "localhost", "port": 6379, "db": 0})
|