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

sage/__init__.py

@@ -0,0 +1,4 @@
+# This is a namespace package
+# 下面这段话可以让 pylance 检查到 middleware，删除该文件将会导致 pylance 无法提示补全
+
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)

sage/middleware/__init__.py

@@ -2,55 +2,11 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从 sage-common 包加载版本信息"""
-    try:
-        # 优先从 sage-common 包加载版本
-        from sage.common._version import __version__
-        return {
-            'version': __version__,
-            'author': 'SAGE Team',
-            'email': 'shuhao_zhang@hust.edu.cn'
-        }
-    except ImportError:
-        # 如果 sage-common 不可用，从项目根目录加载（开发环境）
-        try:
-            from pathlib import Path
-            current_file = Path(__file__).resolve()
-            # 根据当前文件位置计算到项目根目录的层数
-            parts = current_file.parts
-            sage_index = -1
-            for i, part in enumerate(parts):
-                if part == 'SAGE':
-                    sage_index = i
-                    break
-            
-            if sage_index >= 0:
-                root_dir = Path(*parts[:sage_index+1])
-                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.3'),
-                        'author': version_globals.get('__author__', 'SAGE Team'),
-                        'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-                    }
-        except Exception:
-            pass
-    
-    # 最后的默认值
-    return {
-        'version': '0.1.3',
-        '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/_version.py

@@ -1,38 +1,6 @@
 """Version information for sage-middleware package."""
 
-import os
-from pathlib import Path
-
-def _get_version():
-    """智能版本获取 - 开发环境从根目录读取，构建环境使用内嵌版本"""
-    try:
-        # 尝试从项目根目录读取版本（开发环境）
-        current_file = Path(__file__).resolve()
-        
-        # 从当前文件的上级目录开始查找（跳过自己所在的目录）
-        search_path = current_file.parent.parent  # 直接从上级目录开始
-        for _ in range(10):  # 最多向上查找10层
-            root_version_file = search_path / "_version.py"
-            if root_version_file.exists():
-                # 确保不是自己，并且检查是否是项目根目录的版本文件
-                if root_version_file != current_file:
-                    try:
-                        with open(root_version_file, 'r', encoding='utf-8') as f:
-                            content = f.read()
-                            # 检查是否是根目录版本文件（以注释开头，包含项目信息）
-                            if content.startswith('"""') and 'SAGE Version Information' in content and '__version__' in content:
-                                version_globals = {}
-                                exec(content, version_globals)
-                                return version_globals.get('__version__', '0.1.3')
-                    except Exception:
-                        pass
-            search_path = search_path.parent
-            
-        # 如果找不到根目录版本文件，使用内嵌的备份版本（构建环境）
-        return "0.1.3"
-        
-    except Exception:
-        # 最终备份版本
-        return "0.1.3"
-
-__version__ = _get_version()
+# 独立硬编码版本
+__version__ = "0.1.5.1"
+__author__ = "IntelliStream Team"
+__email__ = "shuhao_zhang@hust.edu.cn"

sage/middleware/components/cmake/pybind11_dependency.cmake

@@ -0,0 +1,34 @@
+# Shared pybind11 dependency resolver for SAGE C++ extensions
+#
+# Usage:
+#   include("${CMAKE_CURRENT_SOURCE_DIR}/../cmake/pybind11_dependency.cmake")
+#   # afterwards the function `pybind11_add_module` and targets `pybind11::module`
+#   # will be available.
+#
+# Behaviour:
+#   1. Prefer an existing pybind11 installation discoverable via find_package.
+#   2. Fallback to FetchContent on a pinned, vetted version (v2.13.0).
+#   3. Guarded so the FetchContent path only runs once per configure step.
+
+if(NOT TARGET pybind11::module)
+    find_package(pybind11 CONFIG QUIET)
+    if(pybind11_FOUND)
+        message(STATUS "Using system pybind11 ${pybind11_VERSION} from ${pybind11_CONFIG}")
+    else()
+        include(FetchContent)
+        if(NOT DEFINED pybind11_POPULATED)
+            set(_sage_pybind11_git_tag "v2.13.0" CACHE STRING "Pinned pybind11 version for SAGE extensions")
+            FetchContent_Declare(
+                pybind11
+                GIT_REPOSITORY https://github.com/pybind/pybind11.git
+                GIT_TAG ${_sage_pybind11_git_tag}
+            )
+        endif()
+        FetchContent_MakeAvailable(pybind11)
+        message(STATUS "Fetched pybind11 ${_sage_pybind11_git_tag} for SAGE extensions")
+    endif()
+endif()
+
+if(NOT COMMAND pybind11_add_module)
+    message(FATAL_ERROR "pybind11_add_module is unavailable even after resolving pybind11 dependency")
+endif()

sage/middleware/components/cmake/sage_shared_dependencies.cmake

@@ -0,0 +1,64 @@
+# Centralized dependency setup for SAGE C++ extensions when built via the Python superbuild.
+# This file is optional: individual C++ projects can still build stand-alone by
+# falling back to their local dependency scripts when SAGE_COMMON_DEPS_FILE isn't set.
+
+cmake_minimum_required(VERSION 3.20)
+
+# --- Configuration knobs ---------------------------------------------------
+set(SAGE_PYBIND11_VERSION "2.13.0" CACHE STRING "Pinned pybind11 version for all SAGE extensions")
+option(SAGE_ENABLE_GPERFTOOLS "Build extensions with gperftools/tcmalloc" OFF)
+# Allow callers to provide a pre-installed gperftools root
+set(SAGE_GPERFTOOLS_ROOT "" CACHE PATH "Optional root path for gperftools installation")
+
+# --- pybind11 ---------------------------------------------------------------
+find_package(pybind11 ${SAGE_PYBIND11_VERSION} EXACT CONFIG QUIET)
+if(NOT pybind11_FOUND)
+    include(FetchContent)
+    if(NOT DEFINED pybind11_POPULATED)
+        FetchContent_Declare(
+            pybind11
+            GIT_REPOSITORY https://github.com/pybind/pybind11.git
+            GIT_TAG v${SAGE_PYBIND11_VERSION}
+        )
+    endif()
+    FetchContent_MakeAvailable(pybind11)
+endif()
+
+# --- gperftools -------------------------------------------------------------
+set(SAGE_GPERFTOOLS_LIBS "")
+set(SAGE_GPERFTOOLS_INCLUDE "")
+if(SAGE_ENABLE_GPERFTOOLS)
+    if(SAGE_GPERFTOOLS_ROOT)
+        list(APPEND CMAKE_PREFIX_PATH ${SAGE_GPERFTOOLS_ROOT})
+    endif()
+    find_package(gperftools QUIET)
+    if(gperftools_FOUND)
+        set(SAGE_GPERFTOOLS_LIBS gperftools::profiler gperftools::tcmalloc)
+        if(TARGET gperftools::profiler)
+            get_target_property(_tmp_include gperftools::profiler INTERFACE_INCLUDE_DIRECTORIES)
+            set(SAGE_GPERFTOOLS_INCLUDE ${_tmp_include})
+        endif()
+    else()
+        include(FetchContent)
+        if(NOT TARGET gperftools::profiler)
+            FetchContent_Declare(
+                gperftools
+                GIT_REPOSITORY https://github.com/gperftools/gperftools.git
+                GIT_TAG gperftools-2.15
+            )
+            FetchContent_MakeAvailable(gperftools)
+        endif()
+        set(SAGE_GPERFTOOLS_LIBS gperftools::profiler gperftools::tcmalloc)
+    endif()
+endif()
+
+# --- Common compile flags / definitions ------------------------------------
+set(SAGE_COMMON_COMPILE_DEFINITIONS
+    PYBIND11_INTERNALS_ID="sage_pybind11_shared"
+    _GLIBCXX_USE_CXX11_ABI=1
+    CACHE INTERNAL "")
+
+set(SAGE_COMMON_COMPILE_OPTIONS
+    -fvisibility=hidden
+    -fvisibility-inlines-hidden
+    CACHE INTERNAL "")

sage/middleware/components/extensions_compat.py

@@ -0,0 +1,102 @@
+"""
+SAGE Middleware Components - 运行时兼容性检测
+"""
+
+import warnings
+
+# 尝试导入C++扩展，失败时使用纯Python实现
+_SAGE_DB_AVAILABLE = False
+_SAGE_FLOW_AVAILABLE = False
+
+try:
+    from sage.middleware.components.sage_db.python import _sage_db
+
+    _SAGE_DB_AVAILABLE = True
+except ImportError as e:
+    _sage_db = None
+    warnings.warn(
+        f"SAGE DB C++扩展不可用，某些高性能功能将受限。错误: {e}\n"
+        "安装完整版本：pip install --force-reinstall isage-middleware",
+        UserWarning,
+    )
+
+try:
+    from sage.middleware.components.sage_flow.python import _sage_flow
+    from sage.middleware.components.sage_flow.python.sage_flow import *  # noqa: F401, F403
+
+    _SAGE_FLOW_AVAILABLE = True
+except ImportError as e:
+    _sage_flow = None
+    warnings.warn(
+        f"SAGE Flow C++扩展不可用，流处理功能将受限。错误: {e}\n"
+        "安装完整版本：pip install --force-reinstall isage-middleware",
+        UserWarning,
+    )
+
+
+def is_sage_db_available() -> bool:
+    """检查SAGE DB扩展是否可用"""
+    return _SAGE_DB_AVAILABLE
+
+
+def is_sage_flow_available() -> bool:
+    """检查SAGE Flow扩展是否可用"""
+    return _SAGE_FLOW_AVAILABLE
+
+
+def get_extension_status() -> dict:
+    """获取所有扩展的状态"""
+    return {
+        "sage_db": _SAGE_DB_AVAILABLE,
+        "sage_flow": _SAGE_FLOW_AVAILABLE,
+        "total_available": sum([_SAGE_DB_AVAILABLE, _SAGE_FLOW_AVAILABLE]),
+        "total_extensions": 2,
+    }
+
+
+def check_extensions_availability() -> dict:
+    """检查扩展可用性，返回兼容格式用于CI"""
+    return {
+        "sage_db": _SAGE_DB_AVAILABLE,
+        "sage_flow": _SAGE_FLOW_AVAILABLE,
+    }
+
+
+def require_sage_db():
+    """要求SAGE DB扩展可用，否则抛出异常"""
+    if not _SAGE_DB_AVAILABLE:
+        raise ImportError(
+            "此功能需要SAGE DB C++扩展。请安装完整版本：\n"
+            "pip install --force-reinstall isage-middleware\n"
+            "或安装构建依赖后重新安装：\n"
+            "Ubuntu/Debian: sudo apt-get install build-essential cmake\n"
+            "macOS: brew install cmake"
+        )
+    return _sage_db
+
+
+def require_sage_flow():
+    """要求SAGE Flow扩展可用，否则抛出异常"""
+    if not _SAGE_FLOW_AVAILABLE:
+        raise ImportError(
+            "此功能需要SAGE Flow C++扩展。请安装完整版本：\n"
+            "pip install --force-reinstall isage-middleware\n"
+            "或安装构建依赖后重新安装：\n"
+            "Ubuntu/Debian: sudo apt-get install build-essential cmake\n"
+            "macOS: brew install cmake"
+        )
+    return _sage_flow
+
+
+# 在模块导入时显示状态
+if __name__ != "__main__":
+    status = get_extension_status()
+    if status["total_available"] < status["total_extensions"]:
+        print(
+            f"ℹ️  SAGE扩展状态: {status['total_available']}/{status['total_extensions']} 可用"
+        )
+        if not _SAGE_DB_AVAILABLE:
+            print("  ❌ SAGE DB: C++扩展不可用")
+        if not _SAGE_FLOW_AVAILABLE:
+            print("  ❌ SAGE Flow: C++扩展不可用")
+        print("  💡 提示: 安装构建依赖后重新安装可启用完整功能")

sage/middleware/components/neuromem/__init__.py

@@ -2,55 +2,16 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从 sage-common 包加载版本信息"""
-    try:
-        # 优先从 sage-common 包加载版本
-        from sage.common._version import __version__
-        return {
-            'version': __version__,
-            'author': 'SAGE Team',
-            'email': 'shuhao_zhang@hust.edu.cn'
-        }
-    except ImportError:
-        # 如果 sage-common 不可用，从项目根目录加载（开发环境）
-        try:
-            from pathlib import Path
-            current_file = Path(__file__).resolve()
-            # 根据当前文件位置计算到项目根目录的层数
-            parts = current_file.parts
-            sage_index = -1
-            for i, part in enumerate(parts):
-                if part == 'SAGE':
-                    sage_index = i
-                    break
-            
-            if sage_index >= 0:
-                root_dir = Path(*parts[:sage_index+1])
-                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.3'),
-                        'author': version_globals.get('__author__', 'SAGE Team'),
-                        'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-                    }
-        except Exception:
-            pass
-    
-    # 最后的默认值
-    return {
-        'version': '0.1.3',
-        'author': 'SAGE Team', 
-        'email': 'shuhao_zhang@hust.edu.cn'
-    }
+# 直接从本包的_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"
 
-# 加载信息
-_info = _load_version()
-__version__ = _info['version']
-__author__ = _info['author']
-__email__ = _info['email']
+# 导入主要组件
+from sage.middleware.components.neuromem.micro_service.neuromem_vdb import NeuroMemVDB
+
+__all__ = ["NeuroMemVDB"]

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

@@ -2,55 +2,11 @@
 SAGE - Streaming-Augmented Generative Execution
 """
 
-# 动态版本加载
-def _load_version():
-    """从 sage-common 包加载版本信息"""
-    try:
-        # 优先从 sage-common 包加载版本
-        from sage.common._version import __version__
-        return {
-            'version': __version__,
-            'author': 'SAGE Team',
-            'email': 'shuhao_zhang@hust.edu.cn'
-        }
-    except ImportError:
-        # 如果 sage-common 不可用，从项目根目录加载（开发环境）
-        try:
-            from pathlib import Path
-            current_file = Path(__file__).resolve()
-            # 根据当前文件位置计算到项目根目录的层数
-            parts = current_file.parts
-            sage_index = -1
-            for i, part in enumerate(parts):
-                if part == 'SAGE':
-                    sage_index = i
-                    break
-            
-            if sage_index >= 0:
-                root_dir = Path(*parts[:sage_index+1])
-                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.3'),
-                        'author': version_globals.get('__author__', 'SAGE Team'),
-                        'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-                    }
-        except Exception:
-            pass
-    
-    # 最后的默认值
-    return {
-        'version': '0.1.3',
-        '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"