remove unneeded codes

main.py

@@ -1,7 +1,6 @@
 """FastAPI 应用主文件"""
 
 import logging
-from typing import List
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
@@ -39,45 +38,8 @@ class RequirementRequest(BaseModel):
     """需求分析请求模型"""
     requirement: str
 
-
-class RequirementAnalysisResponse(BaseModel):
-    """需求分析响应"""
-    functional_requirements: List[str]
-    non_functional_requirements: List[str]
-    acceptance_criteria: List[str]
-    edge_cases: List[str]
-    summary: str
-
-
-class TestCase(BaseModel):
-    """测试用例"""
-    test_id: str
-    test_name: str
-    precondition: str
-    steps: List[str]
-    expected_result: str
-    test_type: str
-
-
-class TestCaseResponse(BaseModel):
-    """测试用例响应"""
-    test_cases: List[TestCase]
-    test_strategy: str
-    coverage_plan: str
-
-
-class CodeGenerationResponse(BaseModel):
-    """代码生成响应"""
-    java_code: str
-    unit_tests: str
-    implementation_notes: str
-
-
 class FullWorkflowResponse(BaseModel):
     """完整工作流响应"""
-    requirement_analysis: RequirementAnalysisResponse
-    test_cases: TestCaseResponse
-    code_generation: CodeGenerationResponse
     status: str = "success"
 
 @app.post("/workflow/full", response_model=FullWorkflowResponse)
@@ -106,19 +68,13 @@ async def full_workflow(request: RequirementRequest):
         logger.info(f"开始处理需求: {request.requirement}")
 
         # 调用编排函数，执行三个Agent的工作流
-        result = await orchestrate_agents(request.requirement)
-
-        logger.info("需求处理完成，正在组装响应")
+        await orchestrate_agents(request.requirement)
 
         # 构建响应
         response = FullWorkflowResponse(
-            requirement_analysis=result["requirement_analysis"],
-            test_cases=result["test_cases"],
-            code_generation=result["code_generation"],
             status="success"
         )
 
-        logger.info("响应组装完成")
         return response
 
     except Exception as e: