kollabor 0.4.9__py3-none-any.whl → 0.4.15__py3-none-any.whl

plugins/query_enhancer_plugin.py

@@ -10,6 +10,7 @@ from typing import Any, Dict, List
 import aiohttp
 
 from core.events.models import Event, EventType, Hook, HookPriority
+from core.io.visual_effects import AgnosterSegment
 
 logger = logging.getLogger(__name__)
 
@@ -81,18 +82,16 @@ Enhanced query:""",
             ]
         }
 
-    def __init__(self, name: str, state_manager, event_bus, renderer, config) -> None:
+    def __init__(self, name: str, event_bus, renderer, config) -> None:
         """Initialize the query enhancer plugin.
-        
+
         Args:
             name: Plugin name.
-            state_manager: State management system.
             event_bus: Event bus for hook registration.
             renderer: Terminal renderer.
             config: Configuration manager.
         """
         self.name = name
-        self.state_manager = state_manager
         self.event_bus = event_bus
         self.renderer = renderer
         self.config = config
@@ -131,38 +130,75 @@ Enhanced query:""",
         
         logger.info(f"QueryEnhancer plugin '{name}' initialized with fast model: {self.fast_model_config.get('model', 'unknown')}")
     
-    def get_status_lines(self) -> Dict[str, List[str]]:
-        """Get status lines for the query enhancer plugin organized by area.
-        
-        Returns:
-            Dictionary with status lines organized by area A, B, C.
-        """
-        # Check if status display is enabled for this plugin
-        show_status = self.config.get('plugins.query_enhancer.show_status', True)
-        if not show_status or not self.enabled:
-            return {"A": [], "B": [], "C": []}
-        
-        success_rate = 0.0
-        if self.stats["total_queries"] > 0:
-            success_rate = (self.stats["enhanced_queries"] / self.stats["total_queries"]) * 100
-        
-        # Enhancement stats go in area B (performance-related)
-        return {
-            "A": [],
-            "B": [
-                f"Enhanced: {self.stats['enhanced_queries']}/{self.stats['total_queries']} ({success_rate:.1f}%)",
-                f"Avg Time: {self.stats['avg_enhancement_time']:.2f}s"
-            ],
-            "C": []
-        }
-    
     async def initialize(self) -> None:
         """Initialize the query enhancer plugin."""
         if self.enabled:
-            # Create HTTP session for fast model API calls
             timeout = aiohttp.ClientTimeout(total=self.fast_model_config.get("timeout", 5))
             self.session = aiohttp.ClientSession(timeout=timeout)
             logger.info("QueryEnhancer HTTP session initialized")
+
+        # Register status view
+        await self._register_status_view()
+
+    async def _register_status_view(self) -> None:
+        """Register query enhancer status view."""
+        try:
+            if (self.renderer and
+                hasattr(self.renderer, 'status_renderer') and
+                self.renderer.status_renderer and
+                hasattr(self.renderer.status_renderer, 'status_registry') and
+                self.renderer.status_renderer.status_registry):
+
+                from core.io.status_renderer import StatusViewConfig, BlockConfig
+
+                view = StatusViewConfig(
+                    name="Query Enhancer",
+                    plugin_source="query_enhancer",
+                    priority=350,
+                    blocks=[BlockConfig(
+                        width_fraction=1.0,
+                        content_provider=self._get_status_content,
+                        title="Query Enhancer",
+                        priority=100
+                    )],
+                )
+
+                registry = self.renderer.status_renderer.status_registry
+                registry.register_status_view("query_enhancer", view)
+                logger.info("Registered 'Query Enhancer' status view")
+
+        except Exception as e:
+            logger.error(f"Failed to register status view: {e}")
+
+    def _get_status_content(self) -> List[str]:
+        """Get query enhancer status (agnoster style)."""
+        try:
+            seg = AgnosterSegment()
+
+            show_status = self.config.get('plugins.query_enhancer.show_status', True)
+            if not show_status or not self.enabled:
+                seg.add_neutral("Enhancer: Off", "dark")
+                return [seg.render()]
+
+            success_rate = 0.0
+            if self.stats["total_queries"] > 0:
+                success_rate = (self.stats["enhanced_queries"] / self.stats["total_queries"]) * 100
+
+            model = self.fast_model_config.get('model', 'unknown')
+
+            seg.add_lime("Enhancer", "dark")
+            seg.add_cyan(model, "dark")
+            seg.add_lime(f"{self.stats['enhanced_queries']}/{self.stats['total_queries']}")
+            seg.add_cyan(f"{success_rate:.0f}%")
+            seg.add_neutral(f"{self.stats['avg_enhancement_time']:.2f}s", "mid")
+
+            return [seg.render()]
+
+        except Exception as e:
+            logger.error(f"Error getting status content: {e}")
+            seg = AgnosterSegment()
+            seg.add_neutral("Enhancer: Error", "dark")
+            return [seg.render()]
     
     async def register_hooks(self) -> None:
         """Register query enhancement hooks."""