Migrate agent-coordinator app to centralized aitbc package utilities

- Migrate 13 files from logging to aitbc.get_logger
- main.py, monitoring/prometheus_metrics.py, monitoring/alerting.py
- auth/jwt_handler.py, auth/permissions.py, auth/middleware.py
- consensus/distributed_consensus.py
- ai/realtime_learning.py, ai/advanced_ai.py
- protocols/communication.py, protocols/message_types.py
- routing/load_balancer.py, routing/agent_discovery.py
- Remove logging.basicConfig() from main.py
This commit is contained in:
aitbc
2026-04-25 07:06:44 +02:00
parent 119d0f42c0
commit 16ae53db4f
13 changed files with 36 additions and 30 deletions

View File

@@ -4,7 +4,6 @@ Implements machine learning models, neural networks, and intelligent decision ma
"""
import asyncio
import logging
import numpy as np
from datetime import datetime, timedelta
from typing import Dict, List, Any, Optional, Tuple
@@ -14,7 +13,9 @@ import json
import uuid
import statistics
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
@dataclass
class MLModel:

View File

@@ -4,7 +4,6 @@ Implements adaptive learning, predictive analytics, and intelligent optimization
"""
import asyncio
import logging
from datetime import datetime, timedelta
from typing import Dict, List, Any, Optional, Tuple
from dataclasses import dataclass, field
@@ -13,7 +12,9 @@ import json
import statistics
import uuid
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
@dataclass
class LearningExperience:

View File

@@ -8,9 +8,10 @@ import bcrypt
from datetime import datetime, timedelta
from typing import Dict, Any, Optional, List
import secrets
import logging
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
class JWTHandler:
"""JWT token management and validation"""

View File

@@ -6,12 +6,12 @@ Implements JWT and API key authentication middleware
from fastapi import HTTPException, Depends, status
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from typing import Dict, Any, List, Optional
import logging
from functools import wraps
from aitbc import get_logger
from .jwt_handler import jwt_handler, api_key_manager
logger = logging.getLogger(__name__)
logger = get_logger(__name__)
# Security schemes
security = HTTPBearer(auto_error=False)

View File

@@ -6,9 +6,10 @@ Implements RBAC with roles, permissions, and access control
from enum import Enum
from typing import Dict, List, Set, Any
from dataclasses import dataclass
import logging
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
class Permission(Enum):
"""System permissions enumeration"""

View File

@@ -4,7 +4,6 @@ Implements various consensus algorithms for distributed decision making
"""
import asyncio
import logging
from datetime import datetime, timedelta
from typing import Dict, List, Any, Optional, Set, Tuple
from dataclasses import dataclass, field
@@ -14,7 +13,9 @@ import uuid
import hashlib
import statistics
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
@dataclass
class ConsensusProposal:

View File

@@ -3,12 +3,13 @@ Main FastAPI Application for AITBC Agent Coordinator
"""
import asyncio
import logging
from contextlib import asynccontextmanager
from datetime import datetime
from typing import Dict, List, Optional, Any
import uuid
from aitbc import get_logger
from fastapi import FastAPI, HTTPException, BackgroundTasks, Depends, status, Query
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse, Response
@@ -30,11 +31,7 @@ from .monitoring.prometheus_metrics import metrics_registry, performance_monitor
from .monitoring.alerting import alert_manager, SLAMonitor
# Configure logging
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger(__name__)
logger = get_logger(__name__)
# Global variables
agent_registry: Optional[AgentRegistry] = None

View File

@@ -4,7 +4,6 @@ Implements comprehensive alerting with multiple channels and SLA monitoring
"""
import asyncio
import logging
import smtplib
from datetime import datetime, timedelta
from typing import Dict, List, Any, Optional, Callable
@@ -12,6 +11,8 @@ from dataclasses import dataclass, field
from enum import Enum
import json
from aitbc import get_logger
# Try to import email modules, handle gracefully if not available
try:
from email.mime.text import MimeText
@@ -24,7 +25,7 @@ except ImportError:
import requests
logger = logging.getLogger(__name__)
logger = get_logger(__name__)
class AlertSeverity(Enum):
"""Alert severity levels"""

View File

@@ -8,11 +8,12 @@ import threading
from datetime import datetime, timedelta
from typing import Dict, Any, List, Optional
from collections import defaultdict, deque
import logging
from dataclasses import dataclass, field
import json
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
@dataclass
class MetricValue:

View File

@@ -4,7 +4,6 @@ Multi-Agent Communication Protocols for AITBC Agent Coordination
import asyncio
import json
import logging
from enum import Enum
from typing import Dict, List, Optional, Any, Callable
from dataclasses import dataclass, field
@@ -13,7 +12,9 @@ import uuid
import websockets
from pydantic import BaseModel, Field
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
class MessageType(str, Enum):
"""Message types for agent communication"""

View File

@@ -4,7 +4,6 @@ Message Types and Routing System for AITBC Agent Coordination
import asyncio
import json
import logging
from enum import Enum
from typing import Dict, List, Optional, Any, Callable, Union
from dataclasses import dataclass, field
@@ -14,7 +13,9 @@ import hashlib
from pydantic import BaseModel, Field, validator
from .communication import AgentMessage, MessageType, Priority
logger = logging.getLogger(__name__)
from aitbc import get_logger
logger = get_logger(__name__)
class MessageStatus(str, Enum):
"""Message processing status"""

View File

@@ -4,7 +4,6 @@ Agent Discovery and Registration System for AITBC Agent Coordination
import asyncio
import json
import logging
from typing import Dict, List, Optional, Set, Callable, Any
from dataclasses import dataclass, field
from datetime import datetime, timedelta
@@ -14,10 +13,11 @@ from enum import Enum
import redis.asyncio as redis
from pydantic import BaseModel, Field
from aitbc import get_logger
from ..protocols.message_types import DiscoveryMessage, create_discovery_message
from ..protocols.communication import AgentMessage, MessageType
logger = logging.getLogger(__name__)
logger = get_logger(__name__)
class AgentStatus(str, Enum):
"""Agent status enumeration"""

View File

@@ -4,7 +4,6 @@ Load Balancer for Agent Distribution and Task Assignment
import asyncio
import json
import logging
from typing import Dict, List, Optional, Tuple, Any, Callable
from dataclasses import dataclass, field
from datetime import datetime, timedelta
@@ -13,11 +12,12 @@ import statistics
import uuid
from collections import defaultdict, deque
from aitbc import get_logger
from .agent_discovery import AgentRegistry, AgentInfo, AgentStatus, AgentType
from ..protocols.message_types import TaskMessage, create_task_message
from ..protocols.communication import AgentMessage, MessageType, Priority
logger = logging.getLogger(__name__)
logger = get_logger(__name__)
class LoadBalancingStrategy(str, Enum):
"""Load balancing strategies"""