aiecs 1.0.0__tar.gz → 1.0.2__tar.gz

{aiecs-1.0.0/aiecs.egg-info → aiecs-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiecs
-Version: 1.0.0
+Version: 1.0.2
 Summary: AI Execute Services - A middleware framework for AI-powered task execution and tool orchestration
 Author-email: AIECS Team <iretbl@gmail.com>
 License-Expression: MIT

{aiecs-1.0.0 → aiecs-1.0.2}/aiecs/__init__.py

@@ -5,7 +5,7 @@ A powerful Python middleware framework for building AI-powered applications
 with tool orchestration, task execution, and multi-provider LLM support.
 """
 
-__version__ = "1.0.0"
+__version__ = "1.0.2"
 __author__ = "AIECS Team"
 __email__ = "iretbl@gmail.com"
 

{aiecs-1.0.0 → aiecs-1.0.2}/aiecs/infrastructure/persistence/database_manager.py

@@ -23,6 +23,21 @@ class DatabaseManager:
         self.connection_pool = None
         self._initialized = False
 
+    async def connect(self, min_size: int = 10, max_size: int = 20):
+        """Connect to database and initialize connection pool"""
+        if self._initialized:
+            logger.info("Database already connected")
+            return
+
+        try:
+            await self.init_connection_pool(min_size, max_size)
+            await self.init_database_schema()
+            self._initialized = True
+            logger.info("Database connection and schema initialization completed")
+        except Exception as e:
+            logger.error(f"Failed to connect to database: {e}")
+            raise
+
     async def init_connection_pool(self, min_size: int = 10, max_size: int = 20):
         """Initialize database connection pool"""
         try:
@@ -279,6 +294,12 @@ class DatabaseManager:
             logger.error(f"Database error during cleanup: {e}")
             return False
 
+    async def disconnect(self):
+        """Disconnect from database and close connection pool"""
+        await self.close()
+        self._initialized = False
+        logger.info("Database disconnected")
+
     async def close(self):
         """Close database connection pool"""
         if self.connection_pool:

{aiecs-1.0.0 → aiecs-1.0.2}/aiecs/main.py

@@ -106,7 +106,7 @@ async def lifespan(app: FastAPI):
 app = FastAPI(
     title="AIECS - AI Execute Services",
     description="Middleware service for AI-powered task execution and tool orchestration",
-    version="1.0.0",
+    version="1.0.2",
     lifespan=lifespan
 )
 

{aiecs-1.0.0 → aiecs-1.0.2}/aiecs/scripts/dependency_checker.py

@@ -298,16 +298,24 @@ class DependencyChecker:
             is_critical=False
         ))
         
-        # Python packages
-        python_packages = ["tika", "python-docx", "python-pptx", "openpyxl", "pdfplumber", "pytesseract", "PIL"]
-        for pkg in python_packages:
-            status = self.check_python_package(pkg)
+        # Python packages (package_name: import_name)
+        python_packages = {
+            "tika": "tika",
+            "python-docx": "docx",  # Package name vs import name
+            "python-pptx": "pptx",  # Package name vs import name
+            "openpyxl": "openpyxl",
+            "pdfplumber": "pdfplumber",
+            "pytesseract": "pytesseract",
+            "PIL": "PIL"
+        }
+        for pkg_name, import_name in python_packages.items():
+            status = self.check_python_package(import_name)
             python_deps.append(DependencyInfo(
-                name=pkg,
+                name=pkg_name,
                 status=status,
-                description=f"Python package: {pkg}",
-                install_command=f"pip install {pkg}",
-                impact=f"{pkg} functionality will be unavailable",
+                description=f"Python package: {pkg_name}",
+                install_command=f"pip install {pkg_name}",
+                impact=f"{pkg_name} functionality will be unavailable",
                 is_critical=True
             ))
         
@@ -348,16 +356,24 @@ class DependencyChecker:
             is_critical=False
         ))
         
-        # Python packages
-        python_packages = ["pandas", "numpy", "scipy", "scikit-learn", "statsmodels", "pyreadstat", "openpyxl"]
-        for pkg in python_packages:
-            status = self.check_python_package(pkg)
+        # Python packages (package_name: import_name)
+        python_packages = {
+            "pandas": "pandas",
+            "numpy": "numpy",
+            "scipy": "scipy",
+            "scikit-learn": "sklearn",  # Package name vs import name
+            "statsmodels": "statsmodels",
+            "pyreadstat": "pyreadstat",
+            "openpyxl": "openpyxl"
+        }
+        for pkg_name, import_name in python_packages.items():
+            status = self.check_python_package(import_name)
             python_deps.append(DependencyInfo(
-                name=pkg,
+                name=pkg_name,
                 status=status,
-                description=f"Python package: {pkg}",
-                install_command=f"pip install {pkg}",
-                impact=f"{pkg} functionality will be unavailable",
+                description=f"Python package: {pkg_name}",
+                install_command=f"pip install {pkg_name}",
+                impact=f"{pkg_name} functionality will be unavailable",
                 is_critical=True
             ))
         
@@ -398,16 +414,26 @@ class DependencyChecker:
             is_critical=False
         ))
         
-        # Python packages
-        python_packages = ["jinja2", "matplotlib", "weasyprint", "bleach", "markdown", "pandas", "openpyxl", "python-docx", "python-pptx"]
-        for pkg in python_packages:
-            status = self.check_python_package(pkg)
+        # Python packages (package_name: import_name)
+        python_packages = {
+            "jinja2": "jinja2",
+            "matplotlib": "matplotlib",
+            "weasyprint": "weasyprint",
+            "bleach": "bleach",
+            "markdown": "markdown",
+            "pandas": "pandas",
+            "openpyxl": "openpyxl",
+            "python-docx": "docx",  # Package name vs import name
+            "python-pptx": "pptx"   # Package name vs import name
+        }
+        for pkg_name, import_name in python_packages.items():
+            status = self.check_python_package(import_name)
             python_deps.append(DependencyInfo(
-                name=pkg,
+                name=pkg_name,
                 status=status,
-                description=f"Python package: {pkg}",
-                install_command=f"pip install {pkg}",
-                impact=f"{pkg} functionality will be unavailable",
+                description=f"Python package: {pkg_name}",
+                install_command=f"pip install {pkg_name}",
+                impact=f"{pkg_name} functionality will be unavailable",
                 is_critical=True
             ))
         
@@ -448,16 +474,22 @@ class DependencyChecker:
             is_critical=False
         ))
         
-        # Python packages
-        python_packages = ["playwright", "scrapy", "httpx", "beautifulsoup4", "lxml"]
-        for pkg in python_packages:
-            status = self.check_python_package(pkg)
+        # Python packages (package_name: import_name)
+        python_packages = {
+            "playwright": "playwright",
+            "scrapy": "scrapy",
+            "httpx": "httpx",
+            "beautifulsoup4": "bs4",  # Package name vs import name
+            "lxml": "lxml"
+        }
+        for pkg_name, import_name in python_packages.items():
+            status = self.check_python_package(import_name)
             python_deps.append(DependencyInfo(
-                name=pkg,
+                name=pkg_name,
                 status=status,
-                description=f"Python package: {pkg}",
-                install_command=f"pip install {pkg}",
-                impact=f"{pkg} functionality will be unavailable",
+                description=f"Python package: {pkg_name}",
+                install_command=f"pip install {pkg_name}",
+                impact=f"{pkg_name} functionality will be unavailable",
                 is_critical=True
             ))
         

{aiecs-1.0.0 → aiecs-1.0.2/aiecs.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiecs
-Version: 1.0.0
+Version: 1.0.2
 Summary: AI Execute Services - A middleware framework for AI-powered task execution and tool orchestration
 Author-email: AIECS Team <iretbl@gmail.com>
 License-Expression: MIT

{aiecs-1.0.0 → aiecs-1.0.2}/pyproject.toml

@@ -21,7 +21,7 @@ exclude = '''
 
 [project]
 name = "aiecs"
-version = "1.0.0"
+version = "1.0.2"
 description = "AI Execute Services - A middleware framework for AI-powered task execution and tool orchestration"
 authors = [
     {name = "AIECS Team", email = "iretbl@gmail.com"}
@@ -143,7 +143,7 @@ dependencies = [
     "psutil (>=7.0.0,<8.0.0)",                      # System and process monitoring utilities
 ]
 
-# 开发依赖项 - 移动到 optional-dependencies
+# 开发依赖项 - 移动到 optional-dependencies，use `pip install aiecs[dev]`
 [project.optional-dependencies]
 dev = [
     "pytest>=8.3.5",                               # Testing framework
@@ -184,3 +184,30 @@ py-modules = []
 [build-system]
 requires = ["setuptools>=68.0", "wheel"]
 build-backend = "setuptools.build_meta"
+
+# Pytest configuration
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+testpaths = ["test"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = [
+    "-v",
+    "--tb=short",
+    "--strict-markers",
+    "--disable-warnings"
+]
+markers = [
+    "asyncio: marks tests as async",
+    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
+    "integration: marks tests as integration tests",
+    "security: marks tests as security tests",
+    "performance: marks tests as performance tests"
+]
+filterwarnings = [
+    "ignore::DeprecationWarning",
+    "ignore::UserWarning",
+    "ignore::pytest.PytestUnknownMarkWarning"
+]