feat(新功能): 项目信息提取/生成接口

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试):
This commit is contained in:
zchengrong
2025-05-15 14:49:33 +08:00
parent 3095d2654e
commit 6cb32d11a8
4 changed files with 88 additions and 2 deletions

View File

@@ -0,0 +1,33 @@
import logging
from fastapi import APIRouter, HTTPException
from app.schemas.project_info_extraction import ProjectInfoExtractionModel
from app.schemas.response_template import ResponseModel
from app.service.project_info_extraction.service_generate_brand_info import ProjectInfoExtraction
router = APIRouter()
logger = logging.getLogger()
@router.post("/extraction_project_info")
def extraction_project_info(request_data: ProjectInfoExtractionModel):
"""
通过prompt 提取project_name,role ,gender ,style。
创建一个具有以下参数的请求体:
- **prompt**:
示例参数:
{
"prompt": "海边派对主题的系列设计"
}
"""
try:
logger.info(f"extraction_project_info request item is : @@@@@@:{request_data}")
service = ProjectInfoExtraction(request_data)
data = service.get_result()
logger.info(f"extraction_project_info response @@@@@@:{data}")
except Exception as e:
logger.warning(f"extraction_project_info Run Exception @@@@@@:{e}")
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data)