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:
@@ -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:
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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"""
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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"""
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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"""
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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"""
|
||||
|
||||
@@ -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"""
|
||||
|
||||
@@ -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"""
|
||||
|
||||
@@ -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"""
|
||||
|
||||
Reference in New Issue
Block a user