demo2-qwen/main.py

"""FastAPI 应用主文件"""

import logging

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel

from app.agents import orchestrate_agents
from app.config import get_settings

# 初始化日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# 初始化设置
settings = get_settings()

# 创建 FastAPI 应用
app = FastAPI(
    title=settings.project_name,
    version=settings.project_version,
    debug=settings.fastapi_debug,
)

# 添加 CORS 中间件
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


# 请求/响应模型
class RequirementRequest(BaseModel):
    """需求分析请求模型"""
    message: str

class FullWorkflowResponse(BaseModel):
    """完整工作流响应"""
    status: str = "success"

@app.post("/workflow/full", response_model=FullWorkflowResponse)
async def full_workflow(request: RequirementRequest):
    """
    完整工作流：PM Agent -> QA Agent -> Dev Agent

    这个端点接收一个简单的需求描述，通过三个Agent的协作，
    最终生成完整的需求分析、测试用例和Java代码。

    Args:
        request: RequirementRequest 包含简单的需求描述

    Returns:
        FullWorkflowResponse 包含：
        - requirement_analysis: PM Agent的需求分析结果
        - test_cases: QA Agent生成的测试用例
        - code_generation: Dev Agent生成的Java代码和单元测试

    Example:
        {
            "requirement": "创建一个用户管理系统，支持用户注册、登录、个人信息管理"
        }
    """
    try:
        logger.info(f"开始处理需求: {request.message}")

        # 调用编排函数，执行三个Agent的工作流
        await orchestrate_agents(request.message)

        # 构建响应
        response = FullWorkflowResponse(
            status="success"
        )

        return response

    except Exception as e:
        logger.error(f"处理需求时出错: {str(e)}", exc_info=True)
        return {
            "error": str(e),
            "status": "error"
        }



if __name__ == "__main__":
    import uvicorn

    uvicorn.run(
        app,
        host=settings.fastapi_host,
        port=settings.fastapi_port,
        log_level="info",
    )