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