God Mode+ v3 fix: agents/ui_agent.py

agents/ui_agent.py

@@ -1,24 +1,19 @@
 """
-UIAgent — UI component generation (React/Next.js/Tailwind)
+UIAgent — UI generation and component suggestions via LLMRouter
 """
-from typing import Dict
 import structlog
+from typing import Dict
 from .base_agent import BaseAgent
 
 log = structlog.get_logger()
 
-UI_SYSTEM = """You are an expert React/Next.js/Tailwind CSS UI engineer.
-You generate beautiful, responsive, production-ready UI components.
-Always use:
-- TypeScript with proper types
-- Tailwind CSS for styling
-- shadcn/ui components when appropriate
-- Framer Motion for animations
-- Lucide React for icons
+UI_SYSTEM = """You are a senior frontend engineer specializing in React, Next.js, and Tailwind CSS.
+Generate production-ready UI components with:
+- TypeScript types
+- Responsive design
+- Accessibility (ARIA)
 - Dark mode support
-- Mobile-first responsive design
-- Accessibility (aria labels, semantic HTML)
-"""
+- Clean, readable code"""
 
 
 class UIAgent(BaseAgent):
@@ -26,20 +21,11 @@ class UIAgent(BaseAgent):
         super().__init__("UIAgent", ws_manager, ai_router)
 
     async def run(self, task: str, context: Dict = {}, **kwargs) -> str:
-        session_id = kwargs.get("session_id", "")
-        task_id = kwargs.get("task_id", "")
-
-        await self.emit(task_id, "agent_start", {"agent": "UIAgent", "task": task[:80]}, session_id)
-
-        messages = [
+        session_id = kwargs.get("session_id", context.get("session_id", ""))
+        task_id    = kwargs.get("task_id",    context.get("task_id", ""))
+        messages   = [
             {"role": "system", "content": UI_SYSTEM},
-            {"role": "user", "content": (
-                f"Generate a complete, production-ready UI component for: {task}\n\n"
-                f"Requirements:\n"
-                f"- TypeScript\n- Tailwind CSS\n- Dark mode\n- Mobile responsive\n"
-                f"- Include all imports\n- Export as default"
-            )},
+            {"role": "user",   "content": f"Create a UI component for: {task}"},
         ]
-        result = await self.llm(messages, task_id=task_id, session_id=session_id, temperature=0.3, max_tokens=6000)
-        await self.emit(task_id, "ui_generated", {"agent": "UIAgent"}, session_id)
-        return result
+        await self.emit(task_id, "ui_generation_started", {"task": task[:100]}, session_id)
+        return await self.ask_llm(messages=messages, task_id=task_id, session_id=session_id, temperature=0.5)