aiecs 1.0.1__py3-none-any.whl → 1.7.6__py3-none-any.whl

aiecs/utils/execution_utils.py

@@ -2,22 +2,30 @@ import json
 import asyncio
 import threading
 import time
-from typing import Any, Callable, Dict, List, Optional, Tuple
+from typing import Any, Callable, Dict, Optional, MutableMapping
 from cachetools import LRUCache
 from contextlib import contextmanager
 import logging
-from tenacity import retry, stop_after_attempt, wait_exponential, after_log
+from tenacity import retry, stop_after_attempt, wait_exponential
 
 logger = logging.getLogger(__name__)
 
+
 class ExecutionUtils:
     """
     Provides common utility set for execution layer, including caching and retry logic.
     """
-    def __init__(self, cache_size: int = 100, cache_ttl: int = 3600, retry_attempts: int = 3, retry_backoff: float = 1.0):
+
+    def __init__(
+        self,
+        cache_size: int = 100,
+        cache_ttl: int = 3600,
+        retry_attempts: int = 3,
+        retry_backoff: float = 1.0,
+    ):
         """
         Initialize execution utility class.
-        
+
         Args:
             cache_size (int): Maximum number of cache entries
             cache_ttl (int): Cache time-to-live (seconds)
@@ -28,30 +36,37 @@ class ExecutionUtils:
         self.cache_ttl = cache_ttl
         self.retry_attempts = retry_attempts
         self.retry_backoff = retry_backoff
-        self._cache = LRUCache(maxsize=self.cache_size) if cache_size > 0 else None
+        self._cache: Optional[MutableMapping[str, Any]] = LRUCache(maxsize=self.cache_size) if cache_size > 0 else None
         self._cache_lock = threading.Lock()
         self._cache_ttl_dict: Dict[str, float] = {}
-        
-    def generate_cache_key(self, func_name: str, user_id: str, task_id: str, args: tuple, kwargs: Dict[str, Any]) -> str:
+
+    def generate_cache_key(
+        self,
+        func_name: str,
+        user_id: str,
+        task_id: str,
+        args: tuple,
+        kwargs: Dict[str, Any],
+    ) -> str:
         """
         Generate context-based cache key including user ID, task ID, function name and parameters.
-        
+
         Args:
             func_name (str): Function name
             user_id (str): User ID
             task_id (str): Task ID
             args (tuple): Positional arguments
             kwargs (Dict[str, Any]): Keyword arguments
-            
+
         Returns:
             str: Cache key
         """
         key_dict = {
-            'func': func_name,
-            'user_id': user_id,
-            'task_id': task_id,
-            'args': args,
-            'kwargs': {k: v for k, v in kwargs.items() if k != 'self'}
+            "func": func_name,
+            "user_id": user_id,
+            "task_id": task_id,
+            "args": args,
+            "kwargs": {k: v for k, v in kwargs.items() if k != "self"},
         }
         try:
             key_str = json.dumps(key_dict, sort_keys=True)
@@ -62,14 +77,14 @@ class ExecutionUtils:
     def get_from_cache(self, cache_key: str) -> Optional[Any]:
         """
         Get result from cache if it exists and is not expired.
-        
+
         Args:
             cache_key (str): Cache key
-            
+
         Returns:
             Optional[Any]: Cached result or None
         """
-        if not self._cache:
+        if self._cache is None:
             return None
         with self._cache_lock:
             if cache_key in self._cache:
@@ -83,13 +98,13 @@ class ExecutionUtils:
     def add_to_cache(self, cache_key: str, result: Any, ttl: Optional[int] = None) -> None:
         """
         Add result to cache with optional time-to-live setting.
-        
+
         Args:
             cache_key (str): Cache key
             result (Any): Cached result
             ttl (Optional[int]): Time-to-live (seconds)
         """
-        if not self._cache:
+        if self._cache is None:
             return
         with self._cache_lock:
             self._cache[cache_key] = result
@@ -100,36 +115,40 @@ class ExecutionUtils:
     def create_retry_strategy(self, metric_name: Optional[str] = None) -> Callable:
         """
         Create retry strategy for execution operations.
-        
+
         Args:
             metric_name (Optional[str]): Metric name for logging
-            
+
         Returns:
             Callable: Retry decorator
         """
+
         def after_retry(retry_state):
             logger.warning(f"Retry {retry_state.attempt_number}/{self.retry_attempts} for {metric_name or 'operation'} after {retry_state.idle_for}s: {retry_state.outcome.exception()}")
-        
+
         return retry(
             stop=stop_after_attempt(self.retry_attempts),
             wait=wait_exponential(multiplier=self.retry_backoff, min=1, max=10),
-            after=after_retry
+            after=after_retry,
         )
 
     @contextmanager
     def timeout_context(self, seconds: int):
         """
         Context manager for enforcing operation timeout.
-        
+
         Args:
             seconds (int): Timeout duration (seconds)
-            
+
         Raises:
             TimeoutError: If operation exceeds timeout duration
         """
         loop = asyncio.get_event_loop()
-        future = asyncio.Future()
-        handle = loop.call_later(seconds, lambda: future.set_exception(TimeoutError(f"Operation timed out after {seconds}s")))
+        future: asyncio.Future[None] = asyncio.Future()
+        handle = loop.call_later(
+            seconds,
+            lambda: future.set_exception(TimeoutError(f"Operation timed out after {seconds}s")),
+        )
         try:
             yield future
         finally:
@@ -138,16 +157,16 @@ class ExecutionUtils:
     async def execute_with_retry_and_timeout(self, func: Callable, timeout: int, *args, **kwargs) -> Any:
         """
         Execute operation with retry and timeout mechanism.
-        
+
         Args:
             func (Callable): Function to execute
             timeout (int): Timeout duration (seconds)
             *args: Positional arguments
             **kwargs: Keyword arguments
-            
+
         Returns:
             Any: Operation result
-            
+
         Raises:
             OperationError: If all retry attempts fail
         """
@@ -155,4 +174,4 @@ class ExecutionUtils:
         try:
             return await asyncio.wait_for(retry_strategy(func)(*args, **kwargs), timeout=timeout)
         except asyncio.TimeoutError:
-            raise TimeoutError(f"Operation timed out after {timeout}s")
+            raise TimeoutError(f"Operation timed out after {timeout}s")

aiecs/utils/prompt_loader.py

@@ -1,6 +1,7 @@
 import yaml
 import os
 
+
 def get_prompt(mode: str, service: str) -> str:
     """
     Load the prompt for the specified service from services/{mode}/prompts.yaml.

aiecs/utils/token_usage_repository.py

@@ -1,10 +1,11 @@
 from datetime import datetime
 from typing import Optional, Dict, Any
 import logging
-from ..infrastructure.persistence.redis_client import get_redis_client
+from aiecs.infrastructure.persistence.redis_client import get_redis_client
 
 logger = logging.getLogger(__name__)
 
+
 class TokenUsageRepository:
     """Encapsulates all Redis operations related to user token usage"""
 
@@ -28,7 +29,12 @@ class TokenUsageRepository:
 
         return f"token_usage:{user_id}:{period}"
 
-    async def increment_prompt_tokens(self, user_id: str, prompt_tokens: int, cycle_start_date: Optional[str] = None):
+    async def increment_prompt_tokens(
+        self,
+        user_id: str,
+        prompt_tokens: int,
+        cycle_start_date: Optional[str] = None,
+    ):
         """
         Increment prompt token usage for specified user
 
@@ -51,7 +57,12 @@ class TokenUsageRepository:
             logger.error(f"Failed to increment prompt tokens for user {user_id}: {e}")
             raise
 
-    async def increment_completion_tokens(self, user_id: str, completion_tokens: int, cycle_start_date: Optional[str] = None):
+    async def increment_completion_tokens(
+        self,
+        user_id: str,
+        completion_tokens: int,
+        cycle_start_date: Optional[str] = None,
+    ):
         """
         Increment completion token usage for specified user
 
@@ -74,7 +85,12 @@ class TokenUsageRepository:
             logger.error(f"Failed to increment completion tokens for user {user_id}: {e}")
             raise
 
-    async def increment_total_usage(self, user_id: str, total_tokens: int, cycle_start_date: Optional[str] = None):
+    async def increment_total_usage(
+        self,
+        user_id: str,
+        total_tokens: int,
+        cycle_start_date: Optional[str] = None,
+    ):
         """
         Increment total token usage for specified user
 
@@ -102,7 +118,7 @@ class TokenUsageRepository:
         user_id: str,
         prompt_tokens: int,
         completion_tokens: int,
-        cycle_start_date: Optional[str] = None
+        cycle_start_date: Optional[str] = None,
     ):
         """
         Increment both prompt and completion token usage for specified user
@@ -187,7 +203,7 @@ class TokenUsageRepository:
             return {
                 "prompt_tokens": 0,
                 "completion_tokens": 0,
-                "total_tokens": 0
+                "total_tokens": 0,
             }
 
     async def reset_usage(self, user_id: str, cycle_start_date: Optional[str] = None):
@@ -228,7 +244,7 @@ class TokenUsageRepository:
 
         try:
             client = await get_redis_client()
-            await client.hset(redis_key, {"usage_limit": str(limit)})
+            await client.hset(redis_key, key="usage_limit", value=str(limit))
             logger.info(f"[Repository] Set usage limit {limit} for user '{user_id}' in key '{redis_key}'.")
         except Exception as e:
             logger.error(f"Failed to set usage limit for user {user_id}: {e}")
@@ -246,7 +262,12 @@ class TokenUsageRepository:
             Dictionary containing limit check results
         """
         if not user_id:
-            return {"exceeded": False, "current_usage": 0, "limit": 0, "remaining": 0}
+            return {
+                "exceeded": False,
+                "current_usage": 0,
+                "limit": 0,
+                "remaining": 0,
+            }
 
         try:
             stats = await self.get_usage_stats(user_id, cycle_start_date)
@@ -258,13 +279,13 @@ class TokenUsageRepository:
             limit = int(limit_str) if limit_str else 0
 
             exceeded = limit > 0 and current_usage >= limit
-            remaining = max(0, limit - current_usage) if limit > 0 else float('inf')
+            remaining = max(0, limit - current_usage) if limit > 0 else float("inf")
 
             result = {
                 "exceeded": exceeded,
                 "current_usage": current_usage,
                 "limit": limit,
-                "remaining": remaining
+                "remaining": remaining,
             }
 
             logger.debug(f"[Repository] Usage limit check for user '{user_id}': {result}")
@@ -272,7 +293,12 @@ class TokenUsageRepository:
 
         except Exception as e:
             logger.error(f"Failed to check usage limit for user {user_id}: {e}")
-            return {"exceeded": False, "current_usage": 0, "limit": 0, "remaining": 0}
+            return {
+                "exceeded": False,
+                "current_usage": 0,
+                "limit": 0,
+                "remaining": 0,
+            }
 
 
 # Create a singleton for global application use

aiecs/ws/socket_server.py

@@ -1,4 +1,7 @@
-import socketio
+from typing import Any, Dict
+
+import socketio  # type: ignore[import-untyped]
+from typing import Dict, Any
 from aiecs.config.config import get_settings
 
 settings = get_settings()
@@ -9,17 +12,20 @@ allowed_origins = settings.cors_allowed_origins.split(",") if hasattr(settings,
 # Allow all origins for development (more permissive)
 # In production, you should use specific origins
 # Explicitly set async_mode to 'asgi' for compatibility with uvicorn
-sio = socketio.AsyncServer(cors_allowed_origins="*", async_mode='asgi')
+sio = socketio.AsyncServer(cors_allowed_origins="*", async_mode="asgi")
 # We no longer create a FastAPI app or combined ASGI app here
-# The FastAPI app will be created in main.py and the Socket.IO server will be mounted there
+# The FastAPI app will be created in main.py and the Socket.IO server will
+# be mounted there
 
 # Store connected clients by user ID
-connected_clients = {}
+connected_clients: Dict[str, Any] = {}
+
 
 @sio.event
 async def connect(sid, environ, auth=None):
     print(f"Client connected: {sid}")
 
+
 @sio.event
 async def disconnect(sid):
     print(f"Client disconnected: {sid}")
@@ -27,6 +33,7 @@ async def disconnect(sid):
         if socket_id == sid:
             del connected_clients[user]
 
+
 @sio.event
 async def register(sid, data):
     user_id = data.get("user_id")
@@ -34,7 +41,10 @@ async def register(sid, data):
         connected_clients[user_id] = sid
         print(f"Registered user {user_id} on SID {sid}")
 
+
 # Send progress update to user
+
+
 async def push_progress(user_id: str, data: dict):
     sid = connected_clients.get(user_id)
     if sid:

{aiecs-1.0.1.dist-info → aiecs-1.7.6.dist-info}/METADATA

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: aiecs
-Version: 1.0.1
+Version: 1.7.6
 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
-Project-URL: Homepage, https://github.com/aiecs-team/aiecs
-Project-URL: Bug Tracker, https://github.com/aiecs-team/aiecs/issues
+Project-URL: Homepage, https://github.com/Howmany-Zeta/AI-Execute-Services
+Project-URL: Bug Tracker, https://github.com/Howmany-Zeta/AI-Execute-Services/issues
 Project-URL: Documentation, https://aiecs.readthedocs.io
 Keywords: ai,middleware,llm,orchestration,async,tools
 Classifier: Development Status :: 4 - Beta
@@ -18,7 +18,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Framework :: FastAPI
 Classifier: Framework :: Celery
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: <3.13,>=3.10
+Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: fastapi<0.116.0,>=0.115.12
@@ -37,7 +37,12 @@ Requires-Dist: python-engineio<5.0.0,>=4.12.1
 Requires-Dist: tenacity<10.0.0,>=9.1.2
 Requires-Dist: flower<3.0.0,>=2.0.1
 Requires-Dist: openai<1.76.0,>=1.68.2
-Requires-Dist: google-cloud-aiplatform<2.0.0,>=1.71.1
+Requires-Dist: google-cloud-aiplatform<2.0.0,>=1.80.0
+Requires-Dist: google-genai<2.0.0,>=1.0.0
+Requires-Dist: google-api-python-client<3.0.0,>=2.108.0
+Requires-Dist: google-auth<3.0.0,>=2.25.0
+Requires-Dist: google-auth-httplib2<1.0.0,>=0.2.0
+Requires-Dist: google-auth-oauthlib<2.0.0,>=1.2.0
 Requires-Dist: langchain<0.4.0,>=0.3.26
 Requires-Dist: langgraph<0.6.0,>=0.5.3
 Requires-Dist: weasel==0.4.1
@@ -49,13 +54,14 @@ Requires-Dist: scipy<2.0.0,>=1.15.3
 Requires-Dist: scikit-learn<2.0.0,>=1.5.0
 Requires-Dist: statsmodels<0.15.0,>=0.14.4
 Requires-Dist: pyreadstat<2.0.0,>=1.2.9
+Requires-Dist: pyarrow<22.0.0,>=21.0.0
 Requires-Dist: tabulate<0.10.0,>=0.9.0
 Requires-Dist: python-docx<2.0.0,>=1.1.2
 Requires-Dist: python-pptx<2.0.0,>=1.0.2
 Requires-Dist: openpyxl<4.0.0,>=3.1.5
 Requires-Dist: pdfplumber<0.12.0,>=0.11.7
 Requires-Dist: pdfminer-six==20250506
-Requires-Dist: tika<3.0.0,>=2.6.0
+Requires-Dist: tika<4.0.0,>=3.1.0
 Requires-Dist: matplotlib<4.0.0,>=3.10.3
 Requires-Dist: seaborn<0.14.0,>=0.13.2
 Requires-Dist: jinja2<4.0.0,>=3.1.6
@@ -70,20 +76,35 @@ Requires-Dist: markdown<4.0,>=3.8
 Requires-Dist: bleach<7.0.0,>=6.2.0
 Requires-Dist: sqlalchemy<3.0.0,>=2.0.41
 Requires-Dist: asyncpg<1.0.0,>=0.30.0
+Requires-Dist: aiosqlite<1.0.0,>=0.20.0
+Requires-Dist: networkx<4.0.0,>=3.0
+Requires-Dist: lark-parser<0.13.0,>=0.12.0
 Requires-Dist: prometheus-client<1.0.0,>=0.21.1
 Requires-Dist: jaeger-client<5.0.0,>=4.8.0
 Requires-Dist: opentracing<3.0.0,>=2.4.0
 Requires-Dist: psutil<8.0.0,>=7.0.0
 Provides-Extra: dev
-Requires-Dist: pytest>=8.3.5; extra == "dev"
-Requires-Dist: pytest-asyncio>=1.0.0; extra == "dev"
-Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
-Requires-Dist: pytest-xdist>=3.0.0; extra == "dev"
-Requires-Dist: pytest-mock>=3.10.0; extra == "dev"
-Requires-Dist: black>=25.1.0; extra == "dev"
-Requires-Dist: flake8>=7.2.0; extra == "dev"
-Requires-Dist: mypy>=1.15.0; extra == "dev"
-Requires-Dist: bump-pydantic>=0.8.0; extra == "dev"
+Requires-Dist: pytest<9.0.0,>=8.4.2; extra == "dev"
+Requires-Dist: pytest-asyncio<2.0.0,>=1.2.0; extra == "dev"
+Requires-Dist: pytest-cov<8.0.0,>=7.0.0; extra == "dev"
+Requires-Dist: pytest-timeout<3.0.0,>=2.4.0; extra == "dev"
+Requires-Dist: mypy<2.0.0,>=1.18.2; extra == "dev"
+Requires-Dist: flake8<8.0.0,>=7.3.0; extra == "dev"
+Requires-Dist: black<26.0.0,>=25.1.0; extra == "dev"
+Requires-Dist: types-aiofiles>=25.1.0.20251011; extra == "dev"
+Requires-Dist: types-cachetools>=6.2.0.20251022; extra == "dev"
+Requires-Dist: types-colorama>=0.4.15.20250801; extra == "dev"
+Requires-Dist: types-tqdm>=4.67.0.20250809; extra == "dev"
+Requires-Dist: types-pyyaml>=6.0.12.20250915; extra == "dev"
+Requires-Dist: types-opentracing>=2.4.10.20250622; extra == "dev"
+Requires-Dist: types-lxml>=2025.8.25; extra == "dev"
+Requires-Dist: types-bleach>=6.3.0.20251115; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: sphinx<8.0.0,>=7.0.0; extra == "docs"
+Requires-Dist: sphinx-rtd-theme<3.0.0,>=2.0.0; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints<3.0.0,>=2.0.0; extra == "docs"
+Requires-Dist: myst-parser<3.0.0,>=2.0.0; extra == "docs"
+Requires-Dist: sphinx-copybutton<1.0.0,>=0.5.0; extra == "docs"
 Dynamic: license-file
 
 # AIECS - AI Execute Services
@@ -146,6 +167,22 @@ aiecs-fix-deps
 aiecs-patch-weasel
 ```
 
+### Container Deployment
+
+When installing aiecs in a container (e.g., Docker), you may encounter a warning about scripts not being on PATH:
+
+```
+WARNING: The scripts aiecs, aiecs-check-deps, ... are installed in '/tmp/.local/bin' which is not on PATH.
+```
+
+**Quick Fix:** Add the user bin directory to PATH in your Dockerfile:
+
+```dockerfile
+ENV PATH="${PATH}:/root/.local/bin"
+```
+
+For detailed troubleshooting and best practices, see [Deployment Troubleshooting Guide](docs/user/DEPLOYMENT_TROUBLESHOOTING.md).
+
 ## Quick Start
 
 ### Basic Usage