isage-middleware 0.1.4__py3-none-any.whl → 0.1.5.1__py3-none-any.whl

sage/middleware/components/neuromem/storage_engine/__init__.py

@@ -2,36 +2,11 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从项目根目录动态加载版本信息"""
-    from pathlib import Path
-    
-    # 获取项目根目录
-    current_file = Path(__file__).resolve()
-    root_dir = current_file.parent.parent.parent.parent.parent.parent.parent.parent
-    version_file = root_dir / "_version.py"
-    
-    # 加载版本信息
-    if version_file.exists():
-        version_globals = {}
-        with open(version_file, 'r', encoding='utf-8') as f:
-            exec(f.read(), version_globals)
-        return {
-            'version': version_globals.get('__version__', '0.1.4'),
-            'author': version_globals.get('__author__', 'SAGE Team'),
-            'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-        }
-    
-    # 默认值
-    return {
-        'version': '0.1.4',
-        'author': 'SAGE Team', 
-        'email': 'shuhao_zhang@hust.edu.cn'
-    }
-
-# 加载信息
-_info = _load_version()
-__version__ = _info['version']
-__author__ = _info['author']
-__email__ = _info['email']
+# 直接从本包的_version模块加载版本信息
+try:
+    from sage.middleware._version import __author__, __email__, __version__
+except ImportError:
+    # 备用硬编码版本
+    __version__ = "0.1.4"
+    __author__ = "IntelliStream Team"
+    __email__ = "shuhao_zhang@hust.edu.cn"

sage/middleware/components/neuromem/storage_engine/kv_backend/__init__.py

@@ -2,36 +2,11 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从项目根目录动态加载版本信息"""
-    from pathlib import Path
-    
-    # 获取项目根目录
-    current_file = Path(__file__).resolve()
-    root_dir = current_file.parent.parent.parent.parent.parent.parent.parent.parent.parent
-    version_file = root_dir / "_version.py"
-    
-    # 加载版本信息
-    if version_file.exists():
-        version_globals = {}
-        with open(version_file, 'r', encoding='utf-8') as f:
-            exec(f.read(), version_globals)
-        return {
-            'version': version_globals.get('__version__', '0.1.4'),
-            'author': version_globals.get('__author__', 'SAGE Team'),
-            'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-        }
-    
-    # 默认值
-    return {
-        'version': '0.1.4',
-        'author': 'SAGE Team', 
-        'email': 'shuhao_zhang@hust.edu.cn'
-    }
-
-# 加载信息
-_info = _load_version()
-__version__ = _info['version']
-__author__ = _info['author']
-__email__ = _info['email']
+# 直接从本包的_version模块加载版本信息
+try:
+    from sage.middleware._version import __author__, __email__, __version__
+except ImportError:
+    # 备用硬编码版本
+    __version__ = "0.1.4"
+    __author__ = "IntelliStream Team"
+    __email__ = "shuhao_zhang@hust.edu.cn"

sage/middleware/components/neuromem/storage_engine/kv_backend/base_kv_backend.py

@@ -3,11 +3,13 @@
 from abc import ABC, abstractmethod
 from typing import Any
 
+
 class BaseKVBackend(ABC):
     """
     Abstract base class for key-value backends.
     抽象基类，用于定义 KV 存储后端接口规范。
     """
+
     @abstractmethod
     def get_all_keys(self) -> list[str]:
         pass
@@ -51,15 +53,15 @@ class BaseKVBackend(ABC):
         清空所有键值对。
         """
         pass
-    
+
     @abstractmethod
     def load_data_to_memory(self, path: str):
         pass
-    
+
     @abstractmethod
     def store_data_to_disk(self, path: str):
         pass
 
     @abstractmethod
     def clear_disk_data(self, path: str):
-        pass
+        pass

sage/middleware/components/neuromem/storage_engine/kv_backend/dict_kv_backend.py

@@ -1,9 +1,13 @@
 # file: sage/core/sage.middleware.services.neuromem./storage_engine/kv_backend/dict_kv_backend.py
 
-import os
 import json
+import os
 from typing import Any, Dict
-from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import BaseKVBackend
+
+from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import (
+    BaseKVBackend,
+)
+
 
 class DictKVBackend(BaseKVBackend):
     """
@@ -37,7 +41,7 @@ class DictKVBackend(BaseKVBackend):
 
     def store_data_to_disk(self, path: str):
         """将当前内存数据存储为 JSON 文件"""
-        with open(path, 'w', encoding='utf-8') as f:
+        with open(path, "w", encoding="utf-8") as f:
             # 用 ensure_ascii=False 保证 utf-8 兼容中文，indent=2 可读性高
             json.dump(self._store, f, ensure_ascii=False, indent=2)
 
@@ -45,10 +49,10 @@ class DictKVBackend(BaseKVBackend):
         """从指定 JSON 文件加载数据到内存（覆盖当前 _store）"""
         if not os.path.exists(path):
             raise FileNotFoundError(f"File '{path}' does not exist.")
-        with open(path, 'r', encoding='utf-8') as f:
+        with open(path, "r", encoding="utf-8") as f:
             self._store = json.load(f)
 
     def clear_disk_data(self, path: str):
         """删除指定 JSON 文件"""
         if os.path.exists(path):
-            os.remove(path)
+            os.remove(path)

sage/middleware/components/neuromem/storage_engine/metadata_storage.py

@@ -1,10 +1,14 @@
 # file sage/core/sage.middleware.services.neuromem./storage_engine/metadata_storage.py
 # python -m sage.core.sage.middleware.services.neuromem..storage_engine.metadata_storage
 
-from typing import Dict, Any, Optional, List
+from typing import Any, Dict, List, Optional
 
-from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import BaseKVBackend
-from sage.middleware.components.neuromem.storage_engine.kv_backend.dict_kv_backend import DictKVBackend
+from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import (
+    BaseKVBackend,
+)
+from sage.middleware.components.neuromem.storage_engine.kv_backend.dict_kv_backend import (
+    DictKVBackend,
+)
 
 
 class MetadataStorage:
@@ -26,13 +30,13 @@ class MetadataStorage:
 
     def add_field(self, field_name: str) -> None:
         """
-        Register a new metadata field. If the field is already registered, 
+        Register a new metadata field. If the field is already registered,
         this operation is ignored.
         注册新的元数据字段。如果字段已存在，则忽略此操作。
 
         Args:
             field_name: Name of the field to register
-            
+
         Raises:
             ValueError: If field_name is None or empty
         """
@@ -66,7 +70,7 @@ class MetadataStorage:
         """
         if not isinstance(metadata, dict):
             raise ValueError("Metadata must be a dictionary")
-            
+
         unregistered = set(metadata.keys()) - self.fields
         if unregistered:
             raise ValueError(f"Unregistered metadata fields: {unregistered}")
@@ -92,7 +96,7 @@ class MetadataStorage:
 
         self.validate_fields(metadata)
         self.backend.set(item_id, metadata.copy())
-        
+
     def get_all_ids(self) -> List[str]:
         """
         Get all stored item IDs.
@@ -113,7 +117,7 @@ class MetadataStorage:
 
         Returns:
             Metadata dictionary, empty dict if item doesn't exist
-            
+
         Raises:
             ValueError: If item_id is invalid
         """
@@ -128,10 +132,10 @@ class MetadataStorage:
 
         Args:
             item_id: ID of the item to check
-            
+
         Returns:
             bool: True if item has metadata, False otherwise
-            
+
         Raises:
             ValueError: If item_id is invalid
         """
@@ -146,7 +150,7 @@ class MetadataStorage:
 
         Args:
             item_id: ID of the item to delete
-            
+
         Raises:
             ValueError: If item_id is invalid
         """
@@ -158,7 +162,7 @@ class MetadataStorage:
         """Clear all metadata and registered fields."""
         self.fields.clear()
         self.backend.clear()
-        
+
     def store_to_disk(self, path: str) -> None:
         """
         Store all data to disk as JSON file.
@@ -166,7 +170,7 @@ class MetadataStorage:
 
         Args:
             path: Path to the JSON file
-            
+
         Raises:
             NotImplementedError: If backend doesn't support disk operations
             ValueError: If path is invalid
@@ -184,7 +188,7 @@ class MetadataStorage:
 
         Args:
             path: Path to the JSON file
-            
+
         Raises:
             NotImplementedError: If backend doesn't support disk operations
             ValueError: If path is invalid
@@ -203,7 +207,7 @@ class MetadataStorage:
 
         Args:
             path: Path to the JSON file
-            
+
         Raises:
             NotImplementedError: If backend doesn't support disk operations
             ValueError: If path is invalid
@@ -214,6 +218,7 @@ class MetadataStorage:
             raise NotImplementedError("Backend does not support clear_disk_data")
         self.backend.clear_disk_data(path)
 
+
 if __name__ == "__main__":
     import os
 
@@ -226,10 +231,7 @@ if __name__ == "__main__":
 
     # 构造示例数据
     item_id = "abc123"
-    metadata = {
-        "author": "Alice",
-        "topic": "AI"
-    }
+    metadata = {"author": "Alice", "topic": "AI"}
 
     # 存储元数据
     metadata_store.store(item_id, metadata)
@@ -245,7 +247,7 @@ if __name__ == "__main__":
 
     # 等待用户输入 yes 再加载
     user_input = input("Step 4 | Enter 'yes' to load metadata from disk: ")
-    if user_input.strip().lower() == 'yes':
+    if user_input.strip().lower() == "yes":
         metadata_store.load_from_disk(disk_path)
         print("Step 5 | After reload, retrieved:", metadata_store.get(item_id))
     else:
@@ -257,4 +259,3 @@ if __name__ == "__main__":
 
     # 可选：检查磁盘上确实没这个文件
     print("Step 7 | File exists after deletion?:", os.path.exists(disk_path))
-

sage/middleware/components/neuromem/storage_engine/text_storage.py

@@ -1,10 +1,14 @@
 # file sage/core/sage.middleware.services.neuromem./storage_engine/text_storage.py
 # python -m sage.core.sage.middleware.services.neuromem..storage_engine.text_storage
 
-from typing import Optional, List
+from typing import List, Optional
 
-from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import BaseKVBackend
-from sage.middleware.components.neuromem.storage_engine.kv_backend.dict_kv_backend import DictKVBackend
+from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import (
+    BaseKVBackend,
+)
+from sage.middleware.components.neuromem.storage_engine.kv_backend.dict_kv_backend import (
+    DictKVBackend,
+)
 
 
 class TextStorage:
@@ -17,7 +21,7 @@ class TextStorage:
     def __init__(self, backend: Optional[BaseKVBackend] = None):
         # 支持自定义后端，默认用内存字典
         self.backend = backend or DictKVBackend()
-        
+
     def get_all_ids(self) -> List[str]:
         return self.backend.get_all_keys()
 
@@ -26,7 +30,7 @@ class TextStorage:
 
     def delete(self, item_id: str):
         self.backend.delete(item_id)
-        
+
     def store(self, item_id: str, text: str):
         """
         Store text under a given item_id.
@@ -70,9 +74,10 @@ class TextStorage:
             raise NotImplementedError("Backend does not support clear_disk_data")
         self.backend.clear_disk_data(path)
 
+
 """测试预期输出
 Retrieved: hello world
-After clear: 
+After clear:
 """
 
 if __name__ == "__main__":
@@ -95,7 +100,7 @@ if __name__ == "__main__":
 
     # 3. 等待用户输入 yes 再读取
     user_input = input("Step 3 | Enter 'yes' to load data from disk: ")
-    if user_input.strip().lower() == 'yes':
+    if user_input.strip().lower() == "yes":
         ts.load_from_disk(disk_path)
         print("Step 3 | After reload, Retrieved:", ts.get(text_id))
     else:

sage/middleware/components/neuromem/storage_engine/vector_storage.py

@@ -2,11 +2,15 @@
 # python -m sage.core.sage.middleware.services.neuromem..storage_engine.vector_storage
 
 import json
-import numpy as np
-from typing import Any, Optional, List
+from typing import Any, List, Optional
 
-from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import BaseKVBackend
-from sage.middleware.components.neuromem.storage_engine.kv_backend.dict_kv_backend import DictKVBackend
+import numpy as np
+from sage.middleware.components.neuromem.storage_engine.kv_backend.base_kv_backend import (
+    BaseKVBackend,
+)
+from sage.middleware.components.neuromem.storage_engine.kv_backend.dict_kv_backend import (
+    DictKVBackend,
+)
 
 
 class VectorStorage:
@@ -17,16 +21,16 @@ class VectorStorage:
 
     def __init__(self, backend: Optional[BaseKVBackend] = None):
         self.backend = backend or DictKVBackend()
-        
+
     def get_all_ids(self) -> List[str]:
         return self.backend.get_all_keys()
-    
+
     def has(self, item_id: str) -> bool:
         return self.backend.has(item_id)
-    
+
     def delete(self, item_id: str):
         self.backend.delete(item_id)
-        
+
     def store(self, hash_id: str, vector: Any):
         self.backend.set(hash_id, vector)
 
@@ -54,6 +58,7 @@ class VectorStorage:
             raise NotImplementedError("Backend does not support clear_disk_data")
         self.backend.clear_disk_data(path)
 
+
 if __name__ == "__main__":
     import hashlib
 
@@ -74,7 +79,7 @@ if __name__ == "__main__":
 
     # 等待用户输入 yes 再读取
     user_input = input("Step 4 | Enter 'yes' to load data from disk: ")
-    if user_input.strip().lower() == 'yes':
+    if user_input.strip().lower() == "yes":
         vs.load_from_disk(disk_path)
         print("Step 5 | After reload, Retrieved:", vs.get(vector_id))
     else:

sage/middleware/components/neuromem/utils/__init__.py

@@ -2,36 +2,11 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从项目根目录动态加载版本信息"""
-    from pathlib import Path
-    
-    # 获取项目根目录
-    current_file = Path(__file__).resolve()
-    root_dir = current_file.parent.parent.parent.parent.parent.parent.parent.parent
-    version_file = root_dir / "_version.py"
-    
-    # 加载版本信息
-    if version_file.exists():
-        version_globals = {}
-        with open(version_file, 'r', encoding='utf-8') as f:
-            exec(f.read(), version_globals)
-        return {
-            'version': version_globals.get('__version__', '0.1.4'),
-            'author': version_globals.get('__author__', 'SAGE Team'),
-            'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-        }
-    
-    # 默认值
-    return {
-        'version': '0.1.4',
-        'author': 'SAGE Team', 
-        'email': 'shuhao_zhang@hust.edu.cn'
-    }
-
-# 加载信息
-_info = _load_version()
-__version__ = _info['version']
-__author__ = _info['author']
-__email__ = _info['email']
+# 直接从本包的_version模块加载版本信息
+try:
+    from sage.middleware._version import __author__, __email__, __version__
+except ImportError:
+    # 备用硬编码版本
+    __version__ = "0.1.4"
+    __author__ = "IntelliStream Team"
+    __email__ = "shuhao_zhang@hust.edu.cn"

sage/middleware/components/neuromem/utils/path_utils.py

@@ -14,9 +14,10 @@
 #         if parent == cur_dir:
 #             raise FileNotFoundError("Could not find 'sage' directory in parent folders.")
 #         cur_dir = parent
-        
+
 import os
 
+
 def get_default_data_dir():
     # 获取当前执行程序的工作目录
     cur_dir = os.getcwd()

sage/middleware/components/sage_db/__init__.py

@@ -2,36 +2,11 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从项目根目录动态加载版本信息"""
-    from pathlib import Path
-    
-    # 获取项目根目录
-    current_file = Path(__file__).resolve()
-    root_dir = current_file.parent.parent.parent.parent.parent.parent.parent
-    version_file = root_dir / "_version.py"
-    
-    # 加载版本信息
-    if version_file.exists():
-        version_globals = {}
-        with open(version_file, 'r', encoding='utf-8') as f:
-            exec(f.read(), version_globals)
-        return {
-            'version': version_globals.get('__version__', '0.1.4'),
-            'author': version_globals.get('__author__', 'SAGE Team'),
-            'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-        }
-    
-    # 默认值
-    return {
-        'version': '0.1.4',
-        'author': 'SAGE Team', 
-        'email': 'shuhao_zhang@hust.edu.cn'
-    }
-
-# 加载信息
-_info = _load_version()
-__version__ = _info['version']
-__author__ = _info['author']
-__email__ = _info['email']
+# 直接从本包的_version模块加载版本信息
+try:
+    from sage.middleware._version import __author__, __email__, __version__
+except ImportError:
+    # 备用硬编码版本
+    __version__ = "0.1.4"
+    __author__ = "IntelliStream Team"
+    __email__ = "shuhao_zhang@hust.edu.cn"