impressmem 0.1.0__tar.gz

impressmem-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Pacer
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

impressmem-0.1.0/PKG-INFO

@@ -0,0 +1,190 @@
+Metadata-Version: 2.4
+Name: impressmem
+Version: 0.1.0
+Summary: Impression-based hierarchical memory management library for AI applications with Redis storage
+Author-email: Weichuan Chen <weichuan.chen@foxmail.com>
+Project-URL: Homepage, https://github.com/chenweichuan/ImpressMem
+Project-URL: Issues, https://github.com/chenweichuan/ImpressMem/issues
+Project-URL: Documentation, https://github.com/chenweichuan/ImpressMem#readme
+Project-URL: Repository, https://github.com/chenweichuan/ImpressMem.git
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: redis>=5.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Provides-Extra: llm
+Requires-Dist: openai>=1.0.0; extra == "llm"
+Dynamic: license-file
+
+# ImpressMem
+
+ImpressMem is an impression-based memory management library for AI applications.
+
+## Installation
+
+Requires Redis server running on localhost:6379.
+
+```bash
+pip install impressmem
+```
+
+## Quick Start
+
+```python
+import asyncio
+from impressmem import Config, ImpressionManager
+
+async def main():
+    # Initialize configuration
+    config = Config(
+        bot_name="Bot",
+        redis_config={
+            "host": "localhost",
+            "port": 6379,
+            "db": 0
+        }
+    )
+    
+    # Initialize ImpressionManager
+    manager = await ImpressionManager.initialize(config)
+    
+    # Save an impression
+    await manager.save_impression(
+        clue="USER-NAME",
+        content="John Doe",
+        category="PersonalInfo",
+        labels=["UserProfile", "Identity"],
+        pin=False
+    )
+    
+    # Get and build memory context
+    mixed_impressions = await manager.get_mixed_impressions()
+    mixed_labels = await manager.get_mixed_labels()
+    recent_categories = await manager.get_recent_categories()
+    
+    memory_context = await manager.build_context()
+    
+    print(memory_context)
+    await manager.close()
+
+asyncio.run(main())
+```
+
+## Features
+
+- Category, label, clue-based hierarchical memory organization
+- Time-based rolling memory with limits
+- Pinned critical impressions
+- Memory merge operations for organization
+- Redis-based storage for efficiency
+- Text unit-based memory limits for LLM context management
+- **OpenAI-compatible tool definitions for function calling**
+
+## Configuration
+
+### Config
+```python
+Config(
+    bot_name: str,
+    redis_config: Dict[str, Any] = ...
+)
+```
+
+### Memory Limits
+Memory limits can be adjusted through the ImpressionManager class constants:
+- `CATEGORIES_PER_SET`: Maximum categories to keep (default: 500)
+- `LABELS_PER_SET`: Maximum labels to keep (default: 1500)
+- `CLUES_PER_SET`: Maximum clues to keep (default: 500)
+- `IMPRESSION_TEXT_UNITS_PER_SET`: Maximum text units for impressions (default: 15000)
+
+## API Reference
+
+### ImpressionManager
+Core memory management class.
+
+```python
+# Initialize
+manager = await ImpressionManager.initialize(config)
+
+# Save
+await manager.save_impression(clue, content, category, labels, pin=False)
+
+# Get
+mixed_impressions = await manager.get_mixed_impressions()
+mixed_labels = await manager.get_mixed_labels()
+recent_categories = await manager.get_recent_categories()
+recent_labels = await manager.get_recent_labels()
+recent_clues = await manager.get_recent_clues()
+pinned_clues = await manager.get_pinned_clues()
+category_labels = await manager.get_category_labels(category)
+category_clues = await manager.get_category_clues(category)
+label_clues = await manager.get_label_clues(label)
+impressions = await manager.get_impressions_by_clues(clues)
+
+# Build context
+memory_context = await manager.build_context()
+
+# Merge
+await manager.merge_categories(old, new)
+await manager.merge_labels(sources, target)
+await manager.merge_clues(sources, target, new_content=None)
+
+# Close
+await manager.close()
+```
+
+### Tools
+OpenAI-compatible tool classes for function calling.
+
+```python
+from impressmem import SaveImpressionTool, OrganizeImpressionsTool, RecallImpressionsTool
+
+# Initialize tools
+save_tool = SaveImpressionTool()
+organize_tool = OrganizeImpressionsTool()
+recall_tool = RecallImpressionsTool()
+
+# Get tool definitions (for OpenAI function calling)
+save_def = await save_tool.get_definition()
+organize_def = await organize_tool.get_definition()
+recall_def = await recall_tool.get_definition()
+
+# Execute tools
+full_result, summary = await save_tool.execute(json.dumps(args))
+```
+
+### Utility Functions
+
+```python
+from impressmem import slice_new_turn_messages
+
+# Slice messages for incremental memory processing
+sliced = slice_new_turn_messages(messages)
+```
+
+## Examples
+
+See the `examples/` directory for more usage examples:
+- `context_example.py` - Build memory context
+- `tools_example.py` - Using the tool classes
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+MIT License

impressmem-0.1.0/README.md

@@ -0,0 +1,159 @@
+# ImpressMem
+
+ImpressMem is an impression-based memory management library for AI applications.
+
+## Installation
+
+Requires Redis server running on localhost:6379.
+
+```bash
+pip install impressmem
+```
+
+## Quick Start
+
+```python
+import asyncio
+from impressmem import Config, ImpressionManager
+
+async def main():
+    # Initialize configuration
+    config = Config(
+        bot_name="Bot",
+        redis_config={
+            "host": "localhost",
+            "port": 6379,
+            "db": 0
+        }
+    )
+    
+    # Initialize ImpressionManager
+    manager = await ImpressionManager.initialize(config)
+    
+    # Save an impression
+    await manager.save_impression(
+        clue="USER-NAME",
+        content="John Doe",
+        category="PersonalInfo",
+        labels=["UserProfile", "Identity"],
+        pin=False
+    )
+    
+    # Get and build memory context
+    mixed_impressions = await manager.get_mixed_impressions()
+    mixed_labels = await manager.get_mixed_labels()
+    recent_categories = await manager.get_recent_categories()
+    
+    memory_context = await manager.build_context()
+    
+    print(memory_context)
+    await manager.close()
+
+asyncio.run(main())
+```
+
+## Features
+
+- Category, label, clue-based hierarchical memory organization
+- Time-based rolling memory with limits
+- Pinned critical impressions
+- Memory merge operations for organization
+- Redis-based storage for efficiency
+- Text unit-based memory limits for LLM context management
+- **OpenAI-compatible tool definitions for function calling**
+
+## Configuration
+
+### Config
+```python
+Config(
+    bot_name: str,
+    redis_config: Dict[str, Any] = ...
+)
+```
+
+### Memory Limits
+Memory limits can be adjusted through the ImpressionManager class constants:
+- `CATEGORIES_PER_SET`: Maximum categories to keep (default: 500)
+- `LABELS_PER_SET`: Maximum labels to keep (default: 1500)
+- `CLUES_PER_SET`: Maximum clues to keep (default: 500)
+- `IMPRESSION_TEXT_UNITS_PER_SET`: Maximum text units for impressions (default: 15000)
+
+## API Reference
+
+### ImpressionManager
+Core memory management class.
+
+```python
+# Initialize
+manager = await ImpressionManager.initialize(config)
+
+# Save
+await manager.save_impression(clue, content, category, labels, pin=False)
+
+# Get
+mixed_impressions = await manager.get_mixed_impressions()
+mixed_labels = await manager.get_mixed_labels()
+recent_categories = await manager.get_recent_categories()
+recent_labels = await manager.get_recent_labels()
+recent_clues = await manager.get_recent_clues()
+pinned_clues = await manager.get_pinned_clues()
+category_labels = await manager.get_category_labels(category)
+category_clues = await manager.get_category_clues(category)
+label_clues = await manager.get_label_clues(label)
+impressions = await manager.get_impressions_by_clues(clues)
+
+# Build context
+memory_context = await manager.build_context()
+
+# Merge
+await manager.merge_categories(old, new)
+await manager.merge_labels(sources, target)
+await manager.merge_clues(sources, target, new_content=None)
+
+# Close
+await manager.close()
+```
+
+### Tools
+OpenAI-compatible tool classes for function calling.
+
+```python
+from impressmem import SaveImpressionTool, OrganizeImpressionsTool, RecallImpressionsTool
+
+# Initialize tools
+save_tool = SaveImpressionTool()
+organize_tool = OrganizeImpressionsTool()
+recall_tool = RecallImpressionsTool()
+
+# Get tool definitions (for OpenAI function calling)
+save_def = await save_tool.get_definition()
+organize_def = await organize_tool.get_definition()
+recall_def = await recall_tool.get_definition()
+
+# Execute tools
+full_result, summary = await save_tool.execute(json.dumps(args))
+```
+
+### Utility Functions
+
+```python
+from impressmem import slice_new_turn_messages
+
+# Slice messages for incremental memory processing
+sliced = slice_new_turn_messages(messages)
+```
+
+## Examples
+
+See the `examples/` directory for more usage examples:
+- `context_example.py` - Build memory context
+- `tools_example.py` - Using the tool classes
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+MIT License

impressmem-0.1.0/impressmem/__init__.py

@@ -0,0 +1,23 @@
+"""
+ImpressMem - Impression-based Memory Management Library
+"""
+
+__version__ = "0.1.0"
+
+from impressmem.config import Config
+from impressmem.impression_manager import ImpressionManager
+from impressmem.tools import (
+    SaveImpressionTool,
+    OrganizeImpressionsTool,
+    RecallImpressionsTool,
+)
+from impressmem.utils import slice_new_turn_messages
+
+__all__ = [
+    "Config",
+    "ImpressionManager",
+    "SaveImpressionTool",
+    "OrganizeImpressionsTool",
+    "RecallImpressionsTool",
+    "slice_new_turn_messages",
+]

impressmem-0.1.0/impressmem/config.py

@@ -0,0 +1,17 @@
+"""
+Configuration for ImpressMem
+"""
+from dataclasses import dataclass, field
+from typing import Dict, Any
+
+
+@dataclass
+class Config:
+    """Configuration class for ImpressMem
+    
+    Args:
+        bot_name: Name of the bot/assistant (default: "Bot")
+        redis_config: Redis configuration dictionary (default: {"host": "localhost", "port": 6379, "db": 0})
+    """
+    bot_name: str = "Bot"
+    redis_config: Dict[str, Any] = field(default_factory=lambda: {"host": "localhost", "port": 6379, "db": 0})