aiecs 1.2.0__py3-none-any.whl → 1.2.2__py3-none-any.whl

aiecs/llm/utils/validate_config.py

@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+"""
+Validation script for LLM models configuration.
+
+This script validates the LLM models configuration file and prints
+information about configured providers and models.
+"""
+
+import sys
+import logging
+from pathlib import Path
+
+# Setup logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+
+def main():
+    """Main validation function"""
+    try:
+        from aiecs.llm.config import get_llm_config_loader, validate_llm_config
+        
+        logger.info("=" * 70)
+        logger.info("LLM Models Configuration Validator")
+        logger.info("=" * 70)
+        
+        # Load configuration
+        loader = get_llm_config_loader()
+        config = loader.get_config()
+        
+        logger.info(f"\nConfiguration loaded from: {loader.get_config_path()}")
+        
+        # Validate configuration
+        logger.info("\nValidating configuration...")
+        is_valid, warnings = validate_llm_config(config)
+        
+        if is_valid:
+            logger.info("✓ Configuration is valid!")
+        
+        # Print summary
+        logger.info("\n" + "=" * 70)
+        logger.info("Configuration Summary")
+        logger.info("=" * 70)
+        
+        logger.info(f"\nTotal Providers: {len(config.providers)}")
+        
+        for provider_name, provider_config in config.providers.items():
+            logger.info(f"\n{provider_name}:")
+            logger.info(f"  Default Model: {provider_config.default_model}")
+            logger.info(f"  Total Models: {len(provider_config.models)}")
+            
+            if provider_config.model_mappings:
+                logger.info(f"  Model Aliases: {len(provider_config.model_mappings)}")
+            
+            logger.info("  Available Models:")
+            for model in provider_config.models:
+                cost_str = f"${model.costs.input:.6f} in / ${model.costs.output:.6f} out (per 1K tokens)"
+                logger.info(f"    - {model.name}: {cost_str}")
+                if model.capabilities.vision:
+                    logger.info(f"        Vision: Yes")
+                if model.capabilities.function_calling:
+                    logger.info(f"        Function Calling: Yes")
+        
+        logger.info("\n" + "=" * 70)
+        logger.info("Validation Complete!")
+        logger.info("=" * 70)
+        
+        if warnings:
+            logger.info(f"\nNote: {len(warnings)} warnings were generated during validation.")
+            logger.info("See logs above for details.")
+        
+        return 0
+        
+    except FileNotFoundError as e:
+        logger.error(f"\n✗ Configuration file not found: {e}")
+        logger.error("\nPlease ensure the LLM models configuration file exists at:")
+        logger.error("  - aiecs/config/llm_models.yaml")
+        logger.error("  - Or set LLM_MODELS_CONFIG environment variable")
+        return 1
+        
+    except Exception as e:
+        logger.error(f"\n✗ Validation failed: {e}", exc_info=True)
+        return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())
+

aiecs/main.py

@@ -29,6 +29,10 @@ from aiecs.infrastructure.persistence import (
 )
 from aiecs.infrastructure.messaging.celery_task_manager import CeleryTaskManager
 from aiecs.infrastructure.monitoring.structured_logger import setup_structured_logging
+from aiecs.infrastructure.monitoring import (
+    initialize_global_metrics,
+    close_global_metrics
+)
 
 # Import LLM client factory
 from aiecs.llm.client_factory import LLMClientFactory
@@ -61,6 +65,13 @@ async def lifespan(app: FastAPI):
     # Setup structured logging
     setup_structured_logging()
     
+    # Initialize global metrics (early in startup)
+    try:
+        await initialize_global_metrics()
+        logger.info("Global metrics initialized")
+    except Exception as e:
+        logger.warning(f"Global metrics initialization failed (continuing without it): {e}")
+    
     # Initialize database connection
     try:
         db_manager = DatabaseManager()
@@ -117,6 +128,13 @@ async def lifespan(app: FastAPI):
     await LLMClientFactory.close_all()
     logger.info("LLM clients closed")
     
+    # Close global metrics
+    try:
+        await close_global_metrics()
+        logger.info("Global metrics closed")
+    except Exception as e:
+        logger.warning(f"Error closing global metrics: {e}")
+    
     logger.info("AIECS shutdown complete")
 
 
@@ -124,7 +142,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.2.0",    
+    version="1.2.2",    
     lifespan=lifespan
 )
 
@@ -149,7 +167,19 @@ async def health_check():
     return {
         "status": "healthy",
         "service": "aiecs",
-        "version": "1.2.0"
+        "version": "1.2.2"
+    }
+
+
+# Metrics health check endpoint
+@app.get("/metrics/health")
+async def metrics_health():
+    """Check global metrics health"""
+    from aiecs.infrastructure.monitoring import is_metrics_initialized, get_metrics_summary
+    
+    return {
+        "initialized": is_metrics_initialized(),
+        "summary": get_metrics_summary()
     }
 
 

aiecs/tools/tool_executor/__init__.py

@@ -8,7 +8,7 @@ from .tool_executor import (
     SecurityError,
     TimeoutError,
     ExecutorConfig,
-    ExecutorMetrics,
+    ToolExecutorStats,
     get_executor,
     validate_input,
     cache_result,
@@ -25,7 +25,7 @@ __all__ = [
     'SecurityError',
     'TimeoutError',
     'ExecutorConfig',
-    'ExecutorMetrics',
+    'ToolExecutorStats',
     'get_executor',
     'validate_input',
     'cache_result',

aiecs/tools/tool_executor/tool_executor.py

@@ -76,9 +76,9 @@ class ExecutorConfig(BaseModel):
     model_config = ConfigDict(env_prefix="TOOL_EXECUTOR_")
 
 # Metrics counter
-class ExecutorMetrics:
+class ToolExecutorStats:
     """
-    Tracks executor performance metrics.
+    Tracks tool executor performance statistics.
     """
     def __init__(self):
         self.requests: int = 0
@@ -257,7 +257,7 @@ class ToolExecutor:
         )
         self._thread_pool = ThreadPoolExecutor(max_workers=max(os.cpu_count() or 4, self.config.max_workers))
         self._locks: Dict[str, threading.Lock] = {}
-        self._metrics = ExecutorMetrics()
+        self._metrics = ToolExecutorStats()
         self.execution_utils = ExecutionUtils(
             cache_size=self.config.cache_size,
             cache_ttl=self.config.cache_ttl,

{aiecs-1.2.0.dist-info → aiecs-1.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiecs
-Version: 1.2.0
+Version: 1.2.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.2.0.dist-info → aiecs-1.2.2.dist-info}/RECORD

@@ -1,12 +1,12 @@
-aiecs/__init__.py,sha256=NjZbBge1ngw1XEAwUSHR0zmuJuW6-EmE5ecGuDgFzb4,1859
+aiecs/__init__.py,sha256=j7nUkV4w38idj8EIJXaTfkFWstU-oYs-WfNnRjT6-Xo,1859
 aiecs/__main__.py,sha256=AfQpzy3SgwWuP4DuymYcm4MISMuzqwhxxGSYo53PBvY,1035
-aiecs/aiecs_client.py,sha256=2wzc_XVZvBaTABL2INLFiU_a1uf1xJ9NrL1GTcHOv7Y,17268
-aiecs/main.py,sha256=J9UGndJ1hv9KvfdARvrcAhLG_u-gNY-1XDPySyThzc4,9911
+aiecs/aiecs_client.py,sha256=gIqecRBBH_bYIWhqiHCemdVgmGb9Jqdxf1b6RoqXWlQ,17276
+aiecs/main.py,sha256=jgn6K3H28tKK5z1CPGh-GcswcISh0bhtRaL1yAhtmlg,10837
 aiecs/application/__init__.py,sha256=NkmrUH1DqxJ3vaVC8QwscNdlWqHfC7ZagL4k3nZ_qz4,192
 aiecs/application/executors/__init__.py,sha256=WIl7L9HBsEhNfbNtJdvBvFUJXzESvNZVaiAA6tdtJcs,191
 aiecs/application/executors/operation_executor.py,sha256=-7mFo1hUnWdehVPg0fnSiRhW3LACpIiyLSH-iu7bX4U,13818
 aiecs/config/__init__.py,sha256=HykU6FgZrUx0w8V1_kAjP9NpXZTddZ9M3xo0fmBwMU8,336
-aiecs/config/config.py,sha256=8Xm2x4Jgolx48tuGljzafQ3zXlxYPXT6-8M054V_UaQ,5345
+aiecs/config/config.py,sha256=qjuvD7ew0RKtTftTL4PORwgmridBwS4tAep5llSNUSU,6528
 aiecs/config/registry.py,sha256=5CPJcjeMu3FLc_keuCtJT60DtUxF6w-I68uIoxpcdq8,637
 aiecs/core/__init__.py,sha256=H0ZIk96q0KHKivcobnUCVJdJZmewucVJ9MKhRgUxmk0,1037
 aiecs/core/interface/__init__.py,sha256=soI7zdoN3eQynVb9uiwmgXkM5E75JYffTILktHb48x8,688
@@ -40,23 +40,32 @@ aiecs/infrastructure/__init__.py,sha256=dE-T4IQ0sQTekkMJGqX3HkaaWKJ4gSbq7IN1o8PM
 aiecs/infrastructure/messaging/__init__.py,sha256=KOEywSnktNWEN4O8_GE4KSopjMNEyfYhfUaTOkMxLUE,299
 aiecs/infrastructure/messaging/celery_task_manager.py,sha256=yEKJdO_N9hYb3wlOnoVkBzWahvRj637tOkn4geIjPP0,12984
 aiecs/infrastructure/messaging/websocket_manager.py,sha256=HhhLQw2hRV5Scc5HNMMZbAQGQp9QZBYPJQHulKwaeiI,11280
-aiecs/infrastructure/monitoring/__init__.py,sha256=fQ13Q1MTIJTNlh35BSCqXpayCTM_kYvvPTMRzQfPymA,256
+aiecs/infrastructure/monitoring/__init__.py,sha256=AwMER9ozL1Vvp-Ol8GKIBqMEO-eDQ38Z6Bh6Vnr_aHk,825
 aiecs/infrastructure/monitoring/executor_metrics.py,sha256=z8KJpq6tfCOEArfR-YJ4UClTsef2mNMFuSDHrP51Aww,6040
+aiecs/infrastructure/monitoring/global_metrics_manager.py,sha256=coIBS6v2MmtlQdzAFZTy9XRQvTYzspiJzYTIciU__pw,6549
 aiecs/infrastructure/monitoring/structured_logger.py,sha256=iI895YHmPoaLdXjxHxd952PeTfGw6sh-yUDCnF8R7NY,1657
 aiecs/infrastructure/monitoring/tracing_manager.py,sha256=g4u6paNCZCYdGDEMZiv4bYv_GTG0s8oug-BJgFmkDp0,13449
 aiecs/infrastructure/persistence/__init__.py,sha256=yoog7fEHmhgY50vGdgDNqiZCPUUL2-xnJrdhT5PrPWU,570
 aiecs/infrastructure/persistence/context_engine_client.py,sha256=KIzreimtg6WbuBYI4U0JTiRmNddpdpKHnGvuHVh86Hs,6051
 aiecs/infrastructure/persistence/database_manager.py,sha256=MRkMTALeeybzAfnfuJrOXbEchBCrMAgsz8YYyEUVMjI,12592
-aiecs/infrastructure/persistence/file_storage.py,sha256=d3tcV7Wg_-TGsbw3PY9ttNANntR5rIo7mBgE0CGXKZQ,23321
+aiecs/infrastructure/persistence/file_storage.py,sha256=Z8EozljYPBUuBTzDn2cM2uBEPp2uUW_eHLzwNxbKcOk,23889
 aiecs/infrastructure/persistence/redis_client.py,sha256=CqPtYFP8-KHl3cJG9VHun9YFFSp3kCc3ZaZbW7GlqUU,5791
-aiecs/llm/__init__.py,sha256=INfMR-xeQ0K53iXf_LRMVdo-hDXooIoAcmXRemLWmjQ,1150
-aiecs/llm/base_client.py,sha256=R4B-KBJ0aCEuvni5XwHd6sR3XNYQF3q1ZycgQvCJUfk,3275
-aiecs/llm/client_factory.py,sha256=qvQXo0kIcGC6SzIE7xfCN0f7CcGNKPwauFq2mTUP0SA,13718
-aiecs/llm/custom_callbacks.py,sha256=qwNwWqBl8EYj2aY16zXChZKLpxZahU9QeY4p57uZ5YU,9933
-aiecs/llm/googleai_client.py,sha256=4cJuhdzzn-svM7NLnYASYsnV2P758EeD5_ljVZu_XFc,6213
-aiecs/llm/openai_client.py,sha256=T3-LMDV-bzv0fwyDCw6h9D2XbNbWd0dt-QmjKg-wkd0,4397
-aiecs/llm/vertex_client.py,sha256=DSKyTwlhpOVTXfCTgTD7NholVFtVm7F5qmHk8p8X1Po,10823
-aiecs/llm/xai_client.py,sha256=VYfBGD8ns6NHscT68DDmAmvwBMEVykLQufH8kFNf_L8,6809
+aiecs/llm/__init__.py,sha256=QlpjRn6WK5rCC0Aswx4AeFmSAIxJQ0XyxjimrIee7jE,1860
+aiecs/llm/client_factory.py,sha256=534lVM4JplAFT-4tMHl4ebN9T9Cfz8WTyJjBGA9ly2s,14383
+aiecs/llm/callbacks/__init__.py,sha256=ewPn_AKobfdCkjxFJ1xlIM3bTWdmM01v4XT9_KUZTbg,192
+aiecs/llm/callbacks/custom_callbacks.py,sha256=V94TSNsorgExgzQkNwvId9lxa11oF7efYOT9ULwsdF0,9941
+aiecs/llm/clients/__init__.py,sha256=uQM004TQappwJMqTxVZNscpVPQtirkvYUPea3QYB7d0,697
+aiecs/llm/clients/base_client.py,sha256=j4NY-oEdG5ALBCSddblPpjttISn5teqLVVUuZyYn7g4,5880
+aiecs/llm/clients/googleai_client.py,sha256=sTgdw4eicxWruNGOMSsuEHbfF9RuDQo8SClqEtu1JOQ,6591
+aiecs/llm/clients/openai_client.py,sha256=x7Y_yTVu0kp-gu5Z-M0Bx-O20D0YDgZoJQxzkjNpr6c,4202
+aiecs/llm/clients/vertex_client.py,sha256=LqCCJSCdsaZ9VuBp8SFDdkqk3MxxHn-0HrtH4_xZG7U,10961
+aiecs/llm/clients/xai_client.py,sha256=XEELb9_qFeeQaansDWvAJRJVpt8CaBRLcYskuv9uDq0,6386
+aiecs/llm/config/__init__.py,sha256=KZbwHoBlbcN7HgNueA5p-0GpyVMJRNG1V5T-tkri8G4,1115
+aiecs/llm/config/config_loader.py,sha256=PTMsZax3CoTrMo6BZlUoI7takII3_DHm3w5xTKgBJpA,8921
+aiecs/llm/config/config_validator.py,sha256=y-wGySLtXfv9ZCOAYY1MhlTkKKUplrSOyK2GlP619Ok,7703
+aiecs/llm/config/model_config.py,sha256=cAEaJ-vtE9vD--1PDkPRUfHQI_roMJmG1RBlaJjapGU,5680
+aiecs/llm/utils/__init__.py,sha256=8Fxcr203TcgNWQ5oalzZJ0P2kFSKbVS231kYYXvGPGs,240
+aiecs/llm/utils/validate_config.py,sha256=oNdU8Zw_rwdJYZN9TO6hq6PNOngChjeIOaA4gV-uDlQ,3114
 aiecs/scripts/__init__.py,sha256=cVxQ5iqym520eDQSpV7B6hWolndCLMOVdvhC_D280PE,66
 aiecs/scripts/aid/VERSION_MANAGEMENT.md,sha256=ElbRZFB9KdnDt9Lc4SQWLNY_3NbnLsl2ZHvLmczh2qw,2497
 aiecs/scripts/aid/__init__.py,sha256=nWdqTPA9pib6NGqckhYHYTwuO70TPkW02eV-e3u5i4w,345
@@ -116,8 +125,8 @@ aiecs/tools/task_tools/research_tool.py,sha256=teF5PxwzmTdoJ4f1EHWeO6PmwI8p7n5Yr
 aiecs/tools/task_tools/scraper_tool.py,sha256=OquSAWrCHauWYJGyFoSD5hJwmWbSeBCYFLbZbEVwC4M,24628
 aiecs/tools/task_tools/search_tool.py,sha256=UJNSZwAw8BXECrH_PmyPYQ_ezv65RcuNxBZrgjp4pIA,37142
 aiecs/tools/task_tools/stats_tool.py,sha256=Z5VtdlToioFpgslaBr9aJO7cZn4pBZ5J9TiX2DmxI9s,24302
-aiecs/tools/tool_executor/__init__.py,sha256=gxuujSlQFj8DQ9ejCisO_cYqk5GQkJnwv0sJERd80CM,711
-aiecs/tools/tool_executor/tool_executor.py,sha256=fuoKQahLBgRT6AQbHuOCLzG53w4UAD-QTraGjraEEsw,19496
+aiecs/tools/tool_executor/__init__.py,sha256=FzQ1paMoGF7gZySTSfg96FwAgV914iMx_rfaFU_UFb8,715
+aiecs/tools/tool_executor/tool_executor.py,sha256=x8RDSdfUaGGSp7ilTsZLFQpRditzfXgqkVbSmzVrXZ8,19508
 aiecs/utils/LLM_output_structor.py,sha256=zKkOhrg6smToD0NTCqj3OepBQDTZXgPK4VsKJEFgxyg,15793
 aiecs/utils/__init__.py,sha256=PkukDzRaeeAJUmfm9vA9ez1l3tjvhDRnWZCqzWI6nNw,562
 aiecs/utils/base_callback.py,sha256=UpNrOZZ1RCmiVPnuhjFNde_m29yWg1ID16vzfzBMk7U,1661
@@ -127,9 +136,9 @@ aiecs/utils/prompt_loader.py,sha256=cBS2bZXpYQOWSiOGkhwIzyy3_bETqwIblRi_9qQT9iQ,
 aiecs/utils/token_usage_repository.py,sha256=1xjenLYwC0YT6lKZFEGO4scRCXLuWdec2MWjzih5SZY,10210
 aiecs/ws/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiecs/ws/socket_server.py,sha256=j_9idVY_rWlTsF51FgmuhWCWFVt7_gAHL8vNg3IxV5g,1476
-aiecs-1.2.0.dist-info/licenses/LICENSE,sha256=_1YRaIS0eZu1pv6xfz245UkU0i1Va2B841hv3OWRwqg,12494
-aiecs-1.2.0.dist-info/METADATA,sha256=XhuEwMPhkgzBUrSsh-jgfeqDySzA1cc9OmJSAr3-uAU,16635
-aiecs-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiecs-1.2.0.dist-info/entry_points.txt,sha256=TfLBuwLOfgQqKvnoF1sgTS19-Hgl0aWvCZjIdblIiig,667
-aiecs-1.2.0.dist-info/top_level.txt,sha256=22IlUlOqh9Ni3jXlQNMNUqzbW8dcxXPeR_EQ-BJVcV8,6
-aiecs-1.2.0.dist-info/RECORD,,
+aiecs-1.2.2.dist-info/licenses/LICENSE,sha256=_1YRaIS0eZu1pv6xfz245UkU0i1Va2B841hv3OWRwqg,12494
+aiecs-1.2.2.dist-info/METADATA,sha256=ibzbDBa2h5RjgtWcUybkj25ex60eXekm92K0tVWCPOs,16635
+aiecs-1.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiecs-1.2.2.dist-info/entry_points.txt,sha256=TfLBuwLOfgQqKvnoF1sgTS19-Hgl0aWvCZjIdblIiig,667
+aiecs-1.2.2.dist-info/top_level.txt,sha256=22IlUlOqh9Ni3jXlQNMNUqzbW8dcxXPeR_EQ-BJVcV8,6
+aiecs-1.2.2.dist-info/RECORD,,