change the way get param
This commit is contained in:
16
main.py
16
main.py
@@ -2,7 +2,7 @@
|
|||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from fastapi import FastAPI
|
from fastapi import FastAPI, Request
|
||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
@@ -33,17 +33,12 @@ app.add_middleware(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# 请求/响应模型
|
|
||||||
class RequirementRequest(BaseModel):
|
|
||||||
"""需求分析请求模型"""
|
|
||||||
message: str
|
|
||||||
|
|
||||||
class FullWorkflowResponse(BaseModel):
|
class FullWorkflowResponse(BaseModel):
|
||||||
"""完整工作流响应"""
|
"""完整工作流响应"""
|
||||||
status: str = "success"
|
status: str = "success"
|
||||||
|
|
||||||
@app.post("/workflow/full", response_model=FullWorkflowResponse)
|
@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
|
完整工作流:PM Agent -> QA Agent -> Dev Agent
|
||||||
|
|
||||||
@@ -65,10 +60,13 @@ async def full_workflow(request: RequirementRequest):
|
|||||||
}
|
}
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
logger.info(f"开始处理需求: {request.message}")
|
body_bytes = await request.body()
|
||||||
|
message = body_bytes.decode('utf-8')
|
||||||
|
|
||||||
|
logger.info(f"开始处理需求: {message}")
|
||||||
|
|
||||||
# 调用编排函数,执行三个Agent的工作流
|
# 调用编排函数,执行三个Agent的工作流
|
||||||
await orchestrate_agents(request.message)
|
await orchestrate_agents(message)
|
||||||
|
|
||||||
# 构建响应
|
# 构建响应
|
||||||
response = FullWorkflowResponse(
|
response = FullWorkflowResponse(
|
||||||
|
|||||||
Reference in New Issue
Block a user