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 asyncio
import logging
import numpy as np import numpy as np
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Dict, List, Any, Optional, Tuple from typing import Dict, List, Any, Optional, Tuple
@@ -14,7 +13,9 @@ import json
import uuid import uuid
import statistics import statistics
logger = logging.getLogger(__name__) from aitbc import get_logger
logger = get_logger(__name__)
@dataclass @dataclass
class MLModel: class MLModel:

View File

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

View File

@@ -8,9 +8,10 @@ import bcrypt
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Dict, Any, Optional, List from typing import Dict, Any, Optional, List
import secrets import secrets
import logging
logger = logging.getLogger(__name__) from aitbc import get_logger
logger = get_logger(__name__)
class JWTHandler: class JWTHandler:
"""JWT token management and validation""" """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 import HTTPException, Depends, status
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from typing import Dict, Any, List, Optional from typing import Dict, Any, List, Optional
import logging
from functools import wraps from functools import wraps
from aitbc import get_logger
from .jwt_handler import jwt_handler, api_key_manager from .jwt_handler import jwt_handler, api_key_manager
logger = logging.getLogger(__name__) logger = get_logger(__name__)
# Security schemes # Security schemes
security = HTTPBearer(auto_error=False) security = HTTPBearer(auto_error=False)

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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