diff --git a/main.py b/main.py index 041c589..28d5dc1 100644 --- a/main.py +++ b/main.py @@ -2,7 +2,7 @@ import logging -from fastapi import FastAPI +from fastapi import FastAPI, Request from fastapi.middleware.cors import CORSMiddleware from pydantic import BaseModel @@ -33,17 +33,12 @@ app.add_middleware( ) -# 请求/响应模型 -class RequirementRequest(BaseModel): - """需求分析请求模型""" - message: str - class FullWorkflowResponse(BaseModel): """完整工作流响应""" status: str = "success" @app.post("/workflow/full", response_model=FullWorkflowResponse) -async def full_workflow(request: RequirementRequest): +async def full_workflow(request: Request): """ 完整工作流:PM Agent -> QA Agent -> Dev Agent @@ -65,10 +60,13 @@ async def full_workflow(request: RequirementRequest): } """ try: - logger.info(f"开始处理需求: {request.message}") + body_bytes = await request.body() + message = body_bytes.decode('utf-8') + + logger.info(f"开始处理需求: {message}") # 调用编排函数,执行三个Agent的工作流 - await orchestrate_agents(request.message) + await orchestrate_agents(message) # 构建响应 response = FullWorkflowResponse(