Fix: update agents/ui_agent.py

agents/ui_agent.py

@@ -1,19 +1,24 @@
 """
-UIAgent — UI generation and component suggestions via LLMRouter
+UIAgent — UI component generation (React/Next.js/Tailwind)
 """
-import structlog
 from typing import Dict
+import structlog
 from .base_agent import BaseAgent
 
 log = structlog.get_logger()
 
-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)
+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
 - Dark mode support
-- Clean, readable code"""
+- Mobile-first responsive design
+- Accessibility (aria labels, semantic HTML)
+"""
 
 
 class UIAgent(BaseAgent):
@@ -21,11 +26,20 @@ 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", context.get("session_id", ""))
-        task_id    = kwargs.get("task_id",    context.get("task_id", ""))
-        messages   = [
+        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 = [
             {"role": "system", "content": UI_SYSTEM},
-            {"role": "user",   "content": f"Create a UI component for: {task}"},
+            {"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"
+            )},
         ]
-        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)
+        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