bug fixes

backend/src/api/routes/environments.py

@@ -16,6 +16,7 @@ from backend.src.core.superset_client import SupersetClient
 from superset_tool.models import SupersetConfig
 from pydantic import BaseModel, Field
 from backend.src.core.config_models import Environment as EnvModel
+from backend.src.core.logger import belief_scope
 # [/SECTION]
 
 router = APIRouter()

backend/src/api/routes/plugins.py

@@ -8,6 +8,7 @@ from fastapi import APIRouter, Depends
 
 from ...core.plugin_base import PluginConfig
 from ...dependencies import get_plugin_loader
+from ...core.logger import belief_scope
 
 router = APIRouter()
 

backend/src/core/config_manager.py

@@ -16,7 +16,7 @@ import os
 from pathlib import Path
 from typing import Optional, List
 from .config_models import AppConfig, Environment, GlobalSettings
-from .logger import logger, configure_logger
+from .logger import logger, configure_logger, belief_scope
 # [/SECTION]
 
 # [DEF:ConfigManager:Class]

backend/src/core/database.py

@@ -11,10 +11,11 @@
 # [SECTION: IMPORTS]
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, Session
-from backend.src.models.mapping import Base
+from ..models.mapping import Base
 # Import models to ensure they're registered with Base
-from backend.src.models.task import TaskRecord
-from backend.src.models.connection import ConnectionConfig
+from ..models.task import TaskRecord
+from ..models.connection import ConnectionConfig
+from .logger import belief_scope
 import os
 # [/SECTION]
 

backend/src/core/logger.py

@@ -142,7 +142,6 @@ class WebSocketLogHandler(logging.Handler):
     # @POST: Instance initialized with empty deque.
     # @PARAM: capacity (int) - Maximum number of logs to keep in memory.
     def __init__(self, capacity: int = 1000):
-        with belief_scope("WebSocketLogHandler.__init__"):
         super().__init__()
         self.log_buffer: deque[LogEntry] = deque(maxlen=capacity)
         # In a real implementation, you'd have a way to manage active WebSocket connections
@@ -155,7 +154,6 @@ class WebSocketLogHandler(logging.Handler):
     # @POST: Log is added to the log_buffer.
     # @PARAM: record (logging.LogRecord) - The log record to emit.
     def emit(self, record: logging.LogRecord):
-        with belief_scope("WebSocketLogHandler.emit"):
         try:
             log_entry = LogEntry(
                 level=record.levelname,
@@ -183,7 +181,6 @@ class WebSocketLogHandler(logging.Handler):
     # @POST: Returns list of LogEntry objects.
     # @RETURN: List[LogEntry] - List of buffered log entries.
     def get_recent_logs(self) -> List[LogEntry]:
-        with belief_scope("WebSocketLogHandler.get_recent_logs"):
         """
         Returns a list of recent log entries from the buffer.
         """

backend/src/core/task_manager/persistence.py

@@ -11,8 +11,8 @@ from typing import List, Optional, Dict, Any
 import json
 
 from sqlalchemy.orm import Session
-from backend.src.models.task import TaskRecord
-from backend.src.core.database import TasksSessionLocal
+from ...models.task import TaskRecord
+from ..database import TasksSessionLocal
 from .models import Task, TaskStatus, LogEntry
 from ..logger import logger, belief_scope
 # [/SECTION]