fix: correct context agent logger name from chef to context-agent

Author: rrbanda

agents/context_agent/context_agent.py

@@ -2,7 +2,7 @@
 import logging
 from llama_stack_client import LlamaStackClient
 from llama_stack_client.types import UserMessage
-from shared.log_utils import create_chef_logger, step_printer
+from shared.log_utils import create_correlation_logger, step_printer
 from datetime import datetime
 
 class ContextAgent:
@@ -18,7 +18,7 @@ def __init__(self, client: LlamaStackClient, agent_id: str, session_id: str, vec
         # Model will be determined from agent configuration, not hard-coded
 
         # Use shared logging utilities with correlation ID
-        self.logger = create_chef_logger("context-agent-init")
+        self.logger = create_correlation_logger("context-agent", "context-agent-init")
 
         self.logger.info("ContextAgent initialized")
         self.logger.info(f"Vector DB: {self.vector_db_id}")
@@ -35,7 +35,7 @@ def create_new_session(self, correlation_id: str) -> str:
             )
             session_id = response.session_id
             # Create logger with correlation ID for this session
-            session_logger = create_chef_logger(correlation_id)
+            session_logger = create_correlation_logger("context-agent", correlation_id)
             session_logger.info(f"Created context session: {session_id} for correlation: {correlation_id}")
             return session_id
         except Exception as e:
@@ -47,7 +47,7 @@ async def query_context(self, code, top_k=5, correlation_id=None):
         correlation_id = correlation_id or str(uuid.uuid4())
 
         # Create logger with correlation ID for this query
-        query_logger = create_chef_logger(correlation_id)
+        query_logger = create_correlation_logger("context-agent", correlation_id)
 
         query_logger.info(f"Sending query to ContextAgent: {repr(code)[:200]}...")
         query_logger.info(f"Using vector DB: {self.vector_db_id}")
@@ -266,7 +266,7 @@ async def query_context_stream(self, code, top_k=5, correlation_id=None):
         correlation_id = correlation_id or str(uuid.uuid4())
 
         # Create logger with correlation ID for this query
-        query_logger = create_chef_logger(correlation_id)
+        query_logger = create_correlation_logger("context-agent", correlation_id)
 
         query_logger.info(f"Starting streaming query: {repr(code)[:200]}...")
 

shared/log_utils.py

@@ -106,14 +106,70 @@ def log_step_completion(self, step_name: str, step_number: int, total_steps: int
         self.info(f"Completed {step_name} ({step_number}/{total_steps} - {progress:.1f}%)")
 
 
-def step_printer(steps: List[Any], logger: Optional[ChefAnalysisLogger] = None):
+class CorrelationLogger:
+    """Generic logger for any agent type with correlation ID tracking."""
+    
+    def __init__(self, agent_name: str, correlation_id: str):
+        self.agent_name = agent_name
+        self.correlation_id = correlation_id
+        self.logger = logging.getLogger(f"{agent_name}_{correlation_id}")
+        self.logger.setLevel(logging.INFO)
+        
+        # Create console handler if not already exists
+        if not self.logger.handlers:
+            console_handler = logging.StreamHandler(sys.stdout)
+            console_handler.setLevel(logging.INFO)
+            
+            # Create formatter
+            formatter = logging.Formatter(
+                '%(asctime)s - %(name)s - %(levelname)s - [%(correlation_id)s] %(message)s'
+            )
+            console_handler.setFormatter(formatter)
+            self.logger.addHandler(console_handler)
+        
+        # Add correlation ID to log records
+        for handler in self.logger.handlers:
+            handler.setFormatter(logging.Formatter(
+                '%(asctime)s - %(name)s - %(levelname)s - [%(correlation_id)s] %(message)s'
+            ))
+    
+    def _log_with_correlation(self, level: str, message: str):
+        """Log message with correlation ID."""
+        extra = {'correlation_id': self.correlation_id}
+        if level == 'info':
+            self.logger.info(message, extra=extra)
+        elif level == 'warning':
+            self.logger.warning(message, extra=extra)
+        elif level == 'error':
+            self.logger.error(message, extra=extra)
+        elif level == 'debug':
+            self.logger.debug(message, extra=extra)
+    
+    def info(self, message: str):
+        """Log info message."""
+        self._log_with_correlation('info', message)
+    
+    def warning(self, message: str):
+        """Log warning message."""
+        self._log_with_correlation('warning', message)
+    
+    def error(self, message: str):
+        """Log error message."""
+        self._log_with_correlation('error', message)
+    
+    def debug(self, message: str):
+        """Log debug message."""
+        self._log_with_correlation('debug', message)
+
+
+def step_printer(steps: List[Any], logger: Optional[Any] = None):
     """
     Print the steps of an agent's response in a formatted way.
     Enhanced version with Chef Analysis specific logging.
     
     Args:
         steps: List of steps from an agent's response
-        logger: Optional ChefAnalysisLogger instance
+        logger: Optional logger instance (ChefAnalysisLogger or CorrelationLogger)
     """
     if not steps:
         if logger:
@@ -188,6 +244,11 @@ def create_chef_logger(correlation_id: str) -> ChefAnalysisLogger:
     return ChefAnalysisLogger(correlation_id)
 
 
+def create_correlation_logger(agent_name: str, correlation_id: str) -> CorrelationLogger:
+    """Create a generic correlation logger for any agent type."""
+    return CorrelationLogger(agent_name, correlation_id)
+
+
 def setup_logging(level: str = "INFO") -> None:
     """Setup basic logging configuration."""
     logging.basicConfig(