第二版

crews/sdlc_crew.py

@@ -0,0 +1,176 @@
+"""
+SDLC Crew - CrewAI 编排逻辑 (纯同步版本)
+负责协调 PM、QA、Dev 三个智能体按顺序执行
+"""
+
+from typing import Dict, Any, Optional, Generator
+from datetime import datetime
+from crewai import Crew
+from agents.pm_agent import create_pm_agent, create_pm_task
+from agents.qa_agent import create_qa_agent, create_qa_task
+from agents.dev_agent import create_dev_agent, create_dev_task
+
+
+class SDLCCrew:
+    """
+    SDLC 多智能体协同编排类
+    实现从需求分析→测试用例→代码实现的完整流程
+    """
+    
+    def __init__(self):
+        """初始化 SDLC Crew"""
+        self.pm_agent = create_pm_agent()
+        self.qa_agent = create_qa_agent()
+        self.dev_agent = create_dev_agent()
+        
+        # 存储各阶段结果
+        self.srs_document: Optional[str] = None
+        self.test_plan: Optional[str] = None
+        self.implementation: Optional[str] = None
+    
+    def execute(self, requirement: str) -> Generator[Dict[str, Any], None, None]:
+        """
+        执行完整的 SDLC 流程（纯同步生成器）
+        
+        Args:
+            requirement: 用户需求描述
+            
+        Yields:
+            Dict[str, Any]: 各阶段的 SSE事件
+        """
+        try:
+            # ========== 阶段 1: PM Agent - 需求分析 ==========
+            pm_task = create_pm_task(requirement)
+            pm_crew = Crew(
+                agents=[self.pm_agent],
+                tasks=[pm_task],
+                verbose=True
+            )
+            
+            # 发送开始事件
+            yield {
+                "event": "pm_start",
+                "data": {
+                    "stage": "需求分析",
+                    "status": "started",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            
+            # 执行 Crew 任务
+            result = pm_crew.kickoff()
+            content = result.raw if hasattr(result, 'raw') else str(result)
+            
+            # 发送完成事件
+            yield {
+                "event": "pm_complete",
+                "data": {
+                    "stage": "需求分析",
+                    "content": content,
+                    "status": "completed",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            self.srs_document = content
+            
+            # ========== 阶段 2: QA Agent - 测试用例设计 ==========
+            qa_task = create_qa_task(self.srs_document)
+            qa_crew = Crew(
+                agents=[self.qa_agent],
+                tasks=[qa_task],
+                verbose=True
+            )
+            
+            # 发送开始事件
+            yield {
+                "event": "qa_start",
+                "data": {
+                    "stage": "测试用例设计",
+                    "status": "started",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            
+            # 执行 Crew 任务
+            result = qa_crew.kickoff()
+            content = result.raw if hasattr(result, 'raw') else str(result)
+            
+            # 发送完成事件
+            yield {
+                "event": "qa_complete",
+                "data": {
+                    "stage": "测试用例设计",
+                    "content": content,
+                    "status": "completed",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            self.test_plan = content
+            
+            # ========== 阶段 3: Dev Agent - 代码实现 ==========
+            dev_task = create_dev_task(self.srs_document, self.test_plan)
+            dev_crew = Crew(
+                agents=[self.dev_agent],
+                tasks=[dev_task],
+                verbose=True
+            )
+            
+            # 发送开始事件
+            yield {
+                "event": "dev_start",
+                "data": {
+                    "stage": "代码实现",
+                    "status": "started",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            
+            # 执行 Crew 任务
+            result = dev_crew.kickoff()
+            content = result.raw if hasattr(result, 'raw') else str(result)
+            
+            # 发送完成事件
+            yield {
+                "event": "dev_complete",
+                "data": {
+                    "stage": "代码实现",
+                    "content": content,
+                    "status": "completed",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            self.implementation = content
+            
+            # ========== 最终结果汇总 ==========
+            yield {
+                "event": "final_result",
+                "data": {
+                    "stage": "交付完成",
+                    "status": "success",
+                    "timestamp": datetime.now().isoformat(),
+                    "summary": {
+                        "srs_length": len(self.srs_document) if self.srs_document else 0,
+                        "test_plan_length": len(self.test_plan) if self.test_plan else 0,
+                        "implementation_length": len(self.implementation) if self.implementation else 0
+                    },
+                    "deliverables": {
+                        "srs_document": self.srs_document,
+                        "test_plan": self.test_plan,
+                        "implementation": self.implementation
+                    }
+                }
+            }
+            
+        except Exception as e:
+            # 错误处理
+            yield {
+                "event": "error",
+                "data": {
+                    "stage": "系统错误",
+                    "error": str(e),
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+    
+    # 别名，方便调用
+    execute_sync = execute