isage-middleware 0.1.3__py3-none-any.whl → 0.1.3.1__py3-none-any.whl

sage/middleware/components/sage_db/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/components/sage_db/python/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/components/sage_db/sage_db.py

@@ -1,218 +1,10 @@
 """
-SAGE DB - 高性能向量数据库
+SAGE DB - High-performance vector database with FAISS backend
 
-直接导出 C++ 扩展中的类和函数
+This module provides a Python interface to the SAGE DB vector database,
+which supports efficient similarity search with metadata filtering.
 """
 
-# 直接从 C++ 扩展导入所有必要的类
-try:
-    from ._sage_db import (
-        SageDB,
-        DatabaseConfig,
-        IndexType,
-        DistanceMetric,
-        QueryResult,
-        SearchParams,
-        SageDBException,
-        create_database,
-        index_type_to_string,
-        string_to_index_type,
-        distance_metric_to_string,
-        string_to_distance_metric
-    )
-    print("✓ SAGE DB C++ extension loaded successfully")
-except ImportError as e:
-    raise ImportError(f"Failed to import SAGE DB C++ extension: {e}")
-
-# 导出所有API
-__all__ = [
-    'SageDB',
-    'DatabaseConfig', 
-    'IndexType',
-    'DistanceMetric',
-    'QueryResult',
-    'SearchParams',
-    'SageDBException',
-    'create_database',
-    'index_type_to_string',
-    'string_to_index_type',
-    'distance_metric_to_string',
-    'string_to_distance_metric'
-]
-
-import numpy as np
-from typing import List, Dict, Any, Optional, Callable, Tuple, Union
-
-# 尝试导入C++扩展
-try:
-    from . import _sage_db
-    _cpp_available = True
-except ImportError:
-    try:
-        import _sage_db
-        _cpp_available = True
-    except ImportError:
-        _cpp_available = False
-        raise ImportError("SAGE DB C++ extension not available")
-
-# 重新导出C++类和枚举
-IndexType = _sage_db.IndexType
-DistanceMetric = _sage_db.DistanceMetric
-QueryResult = _sage_db.QueryResult
-SearchParams = _sage_db.SearchParams
-DatabaseConfig = _sage_db.DatabaseConfig
-SageDBException = _sage_db.SageDBException
-
-class SageDB:
-    """
-    High-performance vector database with FAISS backend.
-    
-    Supports efficient similarity search, metadata filtering, and hybrid search.
-    """
-    
-    def __init__(self, dimension: int, 
-                 index_type: IndexType = IndexType.AUTO,
-                 metric: DistanceMetric = DistanceMetric.L2):
-        """
-        Initialize a new SageDB instance.
-        
-        Args:
-            dimension: Vector dimension
-            index_type: Index type for similarity search
-            metric: Distance metric
-        """
-        self._db = _sage_db.create_database(dimension, index_type, metric)
-    
-    @classmethod
-    def from_config(cls, config: DatabaseConfig):
-        """Create SageDB from configuration object."""
-        instance = cls.__new__(cls)
-        instance._db = _sage_db.create_database(config)
-        return instance
-    
-    def add(self, vector: Union[List[float], np.ndarray], 
-            metadata: Optional[Dict[str, str]] = None) -> int:
-        """Add a single vector with optional metadata."""
-        if isinstance(vector, np.ndarray):
-            vector = vector.tolist()
-        return self._db.add(vector, metadata or {})
-    
-    def add_batch(self, vectors: Union[List[List[float]], np.ndarray],
-                  metadata: Optional[List[Dict[str, str]]] = None) -> List[int]:
-        """Add multiple vectors with optional metadata."""
-        if isinstance(vectors, np.ndarray):
-            if len(vectors.shape) != 2:
-                raise ValueError("Vectors array must be 2-dimensional")
-            return _sage_db.add_numpy(self._db, vectors, metadata or [])
-        else:
-            return self._db.add_batch(vectors, metadata or [])
-    
-    def search(self, query: Union[List[float], np.ndarray], 
-               k: int = 10, include_metadata: bool = True) -> List[QueryResult]:
-        """Search for similar vectors."""
-        if isinstance(query, np.ndarray):
-            return _sage_db.search_numpy(self._db, query, SearchParams(k))
-        return self._db.search(query, k, include_metadata)
-    
-    def search_with_params(self, query: Union[List[float], np.ndarray],
-                          params: SearchParams) -> List[QueryResult]:
-        """Search with custom parameters."""
-        if isinstance(query, np.ndarray):
-            return _sage_db.search_numpy(self._db, query, params)
-        return self._db.search(query, params)
-    
-    def filtered_search(self, query: Union[List[float], np.ndarray],
-                       params: SearchParams,
-                       filter_fn: Callable[[Dict[str, str]], bool]) -> List[QueryResult]:
-        """Search with metadata filtering."""
-        if isinstance(query, np.ndarray):
-            query = query.tolist()
-        return self._db.filtered_search(query, params, filter_fn)
-    
-    def remove(self, vector_id: int) -> bool:
-        """Remove a vector by ID."""
-        return self._db.remove(vector_id)
-    
-    def update(self, vector_id: int, vector: Union[List[float], np.ndarray],
-               metadata: Optional[Dict[str, str]] = None) -> bool:
-        """Update a vector and its metadata."""
-        if isinstance(vector, np.ndarray):
-            vector = vector.tolist()
-        return self._db.update(vector_id, vector, metadata or {})
-    
-    def build_index(self):
-        """Build the search index."""
-        self._db.build_index()
-    
-    def train_index(self, training_data: Optional[List[List[float]]] = None):
-        """Train the index with training data."""
-        self._db.train_index(training_data or [])
-    
-    def is_trained(self) -> bool:
-        """Check if the index is trained."""
-        return self._db.is_trained()
-    
-    def set_metadata(self, vector_id: int, metadata: Dict[str, str]) -> bool:
-        """Set metadata for a vector."""
-        return self._db.set_metadata(vector_id, metadata)
-    
-    def get_metadata(self, vector_id: int) -> Optional[Dict[str, str]]:
-        """Get metadata for a vector."""
-        result = self._db.get_metadata(vector_id)
-        return result if result is not None else None
-    
-    def find_by_metadata(self, key: str, value: str) -> List[int]:
-        """Find vectors by metadata key-value pair."""
-        return self._db.find_by_metadata(key, value)
-    
-    def save(self, filepath: str):
-        """Save database to file."""
-        self._db.save(filepath)
-    
-    def load(self, filepath: str):
-        """Load database from file."""
-        self._db.load(filepath)
-    
-    def size(self) -> int:
-        """Get number of vectors in database."""
-        return self._db.size()
-    
-    def dimension(self) -> int:
-        """Get vector dimension."""
-        return self._db.dimension()
-    
-    def index_type(self) -> IndexType:
-        """Get index type."""
-        return self._db.index_type()
-    
-    def config(self) -> DatabaseConfig:
-        """Get database configuration."""
-        return self._db.config()
-    
-    # 高级功能访问
-    def query_engine(self):
-        """Get query engine for advanced search operations."""
-        return self._db.query_engine()
-    
-    def vector_store(self):
-        """Get vector store for low-level operations."""
-        return self._db.vector_store()
-    
-    def metadata_store(self):
-        """Get metadata store for metadata operations."""
-        return self._db.metadata_store()
-
-# 导出的API
-__all__ = [
-    'SageDB',
-    'IndexType',
-    'DistanceMetric', 
-    'QueryResult',
-    'SearchParams',
-    'DatabaseConfig',
-    'SageDBException'
-]
-
 import numpy as np
 from typing import List, Dict, Any, Optional, Callable, Tuple, Union
 try:

sage/middleware/services/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/graph/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/graph/search_engine/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/kv/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/kv/search_engine/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/memory/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/memory/utils/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/vdb/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/services/vdb/search_engine/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }

sage/middleware/utils/__init__.py

@@ -35,7 +35,7 @@ def _load_version():
                     with open(version_file, 'r', encoding='utf-8') as f:
                         exec(f.read(), version_globals)
                     return {
-                        'version': version_globals.get('__version__', '0.1.4'),
+                        'version': version_globals.get('__version__', '0.1.3'),
                         'author': version_globals.get('__author__', 'SAGE Team'),
                         'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
                     }
@@ -44,7 +44,7 @@ def _load_version():
     
     # 最后的默认值
     return {
-        'version': '0.1.4',
+        'version': '0.1.3',
         'author': 'SAGE Team', 
         'email': 'shuhao_zhang@hust.edu.cn'
     }