diff --git a/apps/agent-coordinator/src/app/ai/advanced_ai.py b/apps/agent-coordinator/src/app/ai/advanced_ai.py index b05ca904..1edcb5a5 100644 --- a/apps/agent-coordinator/src/app/ai/advanced_ai.py +++ b/apps/agent-coordinator/src/app/ai/advanced_ai.py @@ -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: diff --git a/apps/agent-coordinator/src/app/ai/realtime_learning.py b/apps/agent-coordinator/src/app/ai/realtime_learning.py index 233af641..d466cd84 100644 --- a/apps/agent-coordinator/src/app/ai/realtime_learning.py +++ b/apps/agent-coordinator/src/app/ai/realtime_learning.py @@ -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: diff --git a/apps/agent-coordinator/src/app/auth/jwt_handler.py b/apps/agent-coordinator/src/app/auth/jwt_handler.py index 4cd54073..0cd00f17 100644 --- a/apps/agent-coordinator/src/app/auth/jwt_handler.py +++ b/apps/agent-coordinator/src/app/auth/jwt_handler.py @@ -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""" diff --git a/apps/agent-coordinator/src/app/auth/middleware.py b/apps/agent-coordinator/src/app/auth/middleware.py index a85d54a7..f61d5e18 100644 --- a/apps/agent-coordinator/src/app/auth/middleware.py +++ b/apps/agent-coordinator/src/app/auth/middleware.py @@ -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) diff --git a/apps/agent-coordinator/src/app/auth/permissions.py b/apps/agent-coordinator/src/app/auth/permissions.py index f5cdd7bc..f82aa33c 100644 --- a/apps/agent-coordinator/src/app/auth/permissions.py +++ b/apps/agent-coordinator/src/app/auth/permissions.py @@ -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""" diff --git a/apps/agent-coordinator/src/app/consensus/distributed_consensus.py b/apps/agent-coordinator/src/app/consensus/distributed_consensus.py index cfc8cb9d..4cf82fea 100644 --- a/apps/agent-coordinator/src/app/consensus/distributed_consensus.py +++ b/apps/agent-coordinator/src/app/consensus/distributed_consensus.py @@ -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: diff --git a/apps/agent-coordinator/src/app/main.py b/apps/agent-coordinator/src/app/main.py index 358cf058..43885172 100644 --- a/apps/agent-coordinator/src/app/main.py +++ b/apps/agent-coordinator/src/app/main.py @@ -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 diff --git a/apps/agent-coordinator/src/app/monitoring/alerting.py b/apps/agent-coordinator/src/app/monitoring/alerting.py index ae45d808..c9229e63 100644 --- a/apps/agent-coordinator/src/app/monitoring/alerting.py +++ b/apps/agent-coordinator/src/app/monitoring/alerting.py @@ -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""" diff --git a/apps/agent-coordinator/src/app/monitoring/prometheus_metrics.py b/apps/agent-coordinator/src/app/monitoring/prometheus_metrics.py index 9673fe5b..9c38c5d6 100644 --- a/apps/agent-coordinator/src/app/monitoring/prometheus_metrics.py +++ b/apps/agent-coordinator/src/app/monitoring/prometheus_metrics.py @@ -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: diff --git a/apps/agent-coordinator/src/app/protocols/communication.py b/apps/agent-coordinator/src/app/protocols/communication.py index 9cbde448..433f3c93 100644 --- a/apps/agent-coordinator/src/app/protocols/communication.py +++ b/apps/agent-coordinator/src/app/protocols/communication.py @@ -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""" diff --git a/apps/agent-coordinator/src/app/protocols/message_types.py b/apps/agent-coordinator/src/app/protocols/message_types.py index ed107694..689abb1b 100644 --- a/apps/agent-coordinator/src/app/protocols/message_types.py +++ b/apps/agent-coordinator/src/app/protocols/message_types.py @@ -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""" diff --git a/apps/agent-coordinator/src/app/routing/agent_discovery.py b/apps/agent-coordinator/src/app/routing/agent_discovery.py index bb6031f2..3a82b741 100644 --- a/apps/agent-coordinator/src/app/routing/agent_discovery.py +++ b/apps/agent-coordinator/src/app/routing/agent_discovery.py @@ -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""" diff --git a/apps/agent-coordinator/src/app/routing/load_balancer.py b/apps/agent-coordinator/src/app/routing/load_balancer.py index 89ffd654..a938c269 100644 --- a/apps/agent-coordinator/src/app/routing/load_balancer.py +++ b/apps/agent-coordinator/src/app/routing/load_balancer.py @@ -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"""