Create core/agents/crewai_agent.py

core/agents/crewai_agent.py

@@ -0,0 +1,143 @@
+from typing import Dict, Any, List, Optional
+from .base import BaseAgent, TaskInput, AgentTool
+import crewai
+from crewai import Agent as CrewAgent, Task as CrewTask, Process
+from datetime import datetime
+import asyncio
+
+class CrewAIAgent(BaseAgent):
+    """CrewAI-based agent implementation"""
+    
+    async def initialize(self) -> None:
+        """Initialize CrewAI agent"""
+        await super().initialize()
+        
+        try:
+            # Convert tools to CrewAI format
+            crew_tools = [
+                self._convert_tool_to_crew(tool)
+                for tool in self.agent_config.tools
+            ]
+            
+            # Create CrewAI agent
+            self.crew_agent = CrewAgent(
+                role=self.agent_config.role.value,
+                goal=self.agent_config.description,
+                backstory=self.agent_config.system_message,
+                tools=crew_tools,
+                verbose=True
+            )
+            
+        except Exception as e:
+            self.logger.error(f"Failed to initialize CrewAI agent: {str(e)}")
+            raise
+    
+    def _convert_tool_to_crew(self, tool: AgentTool) -> crewai.tools.Tool:
+        """Convert Lattice tool to CrewAI tool"""
+        return crewai.tools.Tool(
+            name=tool.name,
+            description=tool.description,
+            func=tool.function
+        )
+    
+    async def _execute_implementation(self, task: TaskInput) -> Dict[str, Any]:
+        """Execute task using CrewAI"""
+        try:
+            # Create CrewAI task
+            crew_task = CrewTask(
+                description=task.description,
+                agent=self.crew_agent,
+                expected_output=task.inputs.get('expected_output'),
+                tools=task.tools if task.tools else None
+            )
+            
+            # Create crew with single agent
+            crew = crewai.Crew(
+                agents=[self.crew_agent],
+                tasks=[crew_task],
+                process=Process.sequential
+            )
+            
+            # Execute task
+            result = await asyncio.to_thread(crew.kickoff)
+            
+            return {
+                'output': result,
+                'metadata': {
+                    'framework': 'crewai',
+                    'process': 'sequential'
+                }
+            }
+            
+        except Exception as e:
+            self.logger.error(f"CrewAI task execution failed: {str(e)}")
+            raise
+
+class CrewAIWorkflow:
+    """CrewAI workflow implementation"""
+    
+    def __init__(self, config: Dict[str, Any]):
+        self.config = config
+        self.logger = logging.getLogger("lattice.agent.crewai.workflow")
+    
+    async def execute_workflow(
+        self,
+        agents: List[CrewAIAgent],
+        tasks: List[TaskInput]
+    ) -> Dict[str, Any]:
+        """Execute a workflow with multiple agents"""
+        try:
+            # Convert to CrewAI format
+            crew_agents = [agent.crew_agent for agent in agents]
+            crew_tasks = [
+                CrewTask(
+                    description=task.description,
+                    agent=agents[i].crew_agent,
+                    expected_output=task.inputs.get('expected_output'),
+                    tools=task.tools
+                )
+                for i, task in enumerate(tasks)
+            ]
+            
+            # Create crew
+            crew = crewai.Crew(
+                agents=crew_agents,
+                tasks=crew_tasks,
+                process=Process.sequential
+            )
+            
+            # Execute workflow
+            results = await asyncio.to_thread(crew.kickoff)
+            
+            return {
+                'results': results,
+                'metadata': {
+                    'framework': 'crewai',
+                    'agent_count': len(agents),
+                    'task_count': len(tasks)
+                },
+                'timestamp': datetime.now().isoformat()
+            }
+            
+        except Exception as e:
+            self.logger.error(f"CrewAI workflow execution failed: {str(e)}")
+            raise
+
+# Tool conversion utilities
+def convert_tool_to_crew(tool: AgentTool) -> crewai.tools.Tool:
+    """Convert a Lattice tool to CrewAI format"""
+    return crewai.tools.Tool(
+        name=tool.name,
+        description=tool.description,
+        func=tool.function
+    )
+
+def convert_crew_to_tool(crew_tool: crewai.tools.Tool) -> AgentTool:
+    """Convert a CrewAI tool to Lattice format"""
+    return AgentTool(
+        name=crew_tool.name,
+        description=crew_tool.description,
+        function=crew_tool.func,
+        required_permissions=[],  # Add appropriate permissions
+        metadata={'framework': 'crewai'}
+    )